Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
suhas-srinath authored Sep 15, 2024
1 parent 7bb220e commit b179bdb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ This is the official repository for the implementation of [GenSelfDiff-HIS](http
The public datasets can be downloaded from [MoNuSeg](https://monuseg.grand-challenge.org/Data/) and [GlaS](https://www.kaggle.com/datasets/sani84/glasmiccai2015-gland-segmentation). These datasets should be placed into the appropriate folders within the pre_process directory for data pre-processing. Our proposed HN dataset will be made available for research purposes in the future.

# Pre-training
One can run the following script using multi-gpus for generative self-supervision using diffusion. Specify the path to the directory of the unlabeled images on which pre-training needs to be done in the script "pretrain/utils.py".
One can run the following script using multi-gpus for generative self-supervision using diffusion. Specify the path to the directory of the unlabeled images on which pre-training needs to be done in the script "GenSelfDiff/pretrain/utils.py".
<pre> CUDA_VISIBLE_DEVICES=0,1,2,3 python3 ./GenSelfDiff/pretrain/SS_diff.py </pre>

# Downstream Training for Segmentation
Expand Down

0 comments on commit b179bdb

Please sign in to comment.