Skip to content

Arctic-Xiangjian/HC701-PROJECT

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HC701-PROJECT

Experiment notification

please fix the random seed by this, and if you have time please test from 42-46.

torch.cuda.manual_seed(42)
torch.cuda.manual_seed_all(42)
random.seed(42)
np.random.seed(42)
os.environ['PYTHONHASHSEED'] = str(42)
torch.manual_seed(42)

Train

python train.py --backbone densenet121 --batch_size 1 --epochs 500 --dataset messidor --seed 42

Load data

The data you can find in my /l/users/xiangjian.hou/

And demo in HC701-PROJECT/notebook/Load_Data_and_train_Demo.ipynb

ideal workspace dir should be

[Your WORKSPACE FOLDER]
├── HC701-PROJECT
│   ├── hc701fed
│   ├── result
│   ├── .gitignore
│   ├── README.md
│   ├── test.py
│   ├── train.py
└── preprocessed
    ├── APTOS
    ├── EyePACS
    ├── messidor
    └── messidor2

Test data

The messidor and messidor2 do not give offical test set. So We can split 0.5/0.25/0.25 as train/val/test split.

EyePACS and APTOS we do not have the test set labels. We need upload to the website the get the result.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published