DARL

KO | EN

gitlite — search

#python #react #typescript #llm #golang #deep-learning #java #windows #awesome #cli #kubernetes #awesome-list #pytorch #openai

(★ 68)

Official repository of "Diffusion Adversarial Representation Learning for Self-supervised Vessel Segmentation", ICLR 2023.

DARL Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Diffusion Adversarial Representation Learning for Self-supervised Vessel Segmentation

This repository is the official implementation of "Diffusion Adversarial Representation Learning for Self-supervised Vessel Segmentation".

[[ICLR 2023](https://openreview.net/forum?id=H0gdPxSwkPb)]
[[arXiv](https://arxiv.org/abs/2209.14566)]

![Image of The Proposed method](https://raw.githubusercontent.com/boahK/DARL/main/figs/method.png)

## Requirements
  * OS : Ubuntu
  * Python >= 3.6
  * PyTorch >= 1.4.0

## Data
In our experiments, we used the publicly available XCAD dataset. Please refer to our main paper.

## Training

To train our model, run this command:

```train
python3 main.py -p train -c config/train.json
```

## Test

To test the trained our model, run:

```eval
python3 main.py -p test -c config/test.json
```

## Pre-trained Models

You can download our pretrained model of XCAD dataset [here](https://drive.google.com/file/d/1Kuh-YEhRaR4LEsltnXflnJgxSoTx06j5/view?usp=sharing).
Then, you can test the model by saving the pretrained weights in the directory ./pretrained_model.
To brifely test our method given the pretrained model, we provided the toy example in the directory './data/'.

## Citations

```
@inproceedings{
kim2023diffusion,
title={Diffusion Adversarial Representation Learning for Self-supervised Vessel Segmentation},
author={Boah Kim and Yujin Oh and Jong Chul Ye},
booktitle={The Eleventh International Conference on Learning Representations },
year={2023},
url={https://openreview.net/forum?id=H0gdPxSwkPb}
}
```

DARL

DARL Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

DARL

DARL Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README