- CPU: Intel(R) Xeon(R) Gold 6130 CPU @ 2.10GHz
- GPU: Tesla P100-SXM2-16GB
- CUDA: 11.7
- Pytorch 1.13.0
- data：data(https://pan.baidu.com/s/1NvgOmqIHd7kzcTMGWB08Zw) - r5va
| Dataset | mIoU | F1 score | Acc | AUC | Sen | MCC |
|---|---|---|---|---|---|---|
| DRIVE | 0.8277 | 0.8195 | 0.9639 | 0.9855 | 0.8392 | 0.8009 |
| STARE | 0.8472 | 0.8368 | 0.9722 | 0.9827 | 0.8348 | 0.8223 |
If you have installed 'WandB', login your ID in command line.
If not, fix 'wandb: false' in "configs/train.yml"
You can login through your command line or 'wandb.login()' inside "main.py"
For Train, edit the configs/train.yml and execute below command
bash bash_train.sh
For Inference, edit the configs/inference.yml and execute below command.
Please locate your model path via 'model_path' in "configs/inference.yml"
bash bash_inference.sh
- If you are using pretrained model, the result should be approximate to table's