├── README.md ├── csv ├── readme.txt ├── test.csv ├── test_balance_700.csv ├── test_gan_350.csv ├── test_gan_350_out_epoch_121.csv ├── test_out.csv ├── test_out_epoch_121.csv ├── test_out_epoch_132.csv ├── test_out_epoch_75.csv ├── train.csv ├── train_balance_50000.csv ├── train_gan.csv ├── val.py └── validation2.py ├── dr_dataloader.py ├── evaluate ├── .ipynb_checkpoints │ └── evaluate-checkpoint.ipynb ├── evaluate.ipynb └── output.csv ├── gcam ├── 13502_left.jpg └── 2901_right.jpg ├── main.sh ├── models ├── __init__.py ├── alexnet.py ├── densenet.py ├── inception.py ├── resnet.py ├── retinet.py ├── squeezenet.py └── vgg.py ├── train_densenet121.py ├── train_multi_gpu.py ├── train_multi_gpu_resnet50.py └── train_test_split.py /README.md: -------------------------------------------------------------------------------- 1 | # Diabetic_Reinopathy_Dectection 2 | 糖网眼底图像分类_pytorch 3 | -------------------------------------------------------------------------------- /csv/readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/readme.txt -------------------------------------------------------------------------------- /csv/test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/test.csv -------------------------------------------------------------------------------- /csv/test_balance_700.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/test_balance_700.csv -------------------------------------------------------------------------------- /csv/test_gan_350.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/test_gan_350.csv -------------------------------------------------------------------------------- /csv/test_gan_350_out_epoch_121.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/test_gan_350_out_epoch_121.csv -------------------------------------------------------------------------------- /csv/test_out.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/test_out.csv -------------------------------------------------------------------------------- /csv/test_out_epoch_121.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/test_out_epoch_121.csv -------------------------------------------------------------------------------- /csv/test_out_epoch_132.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/test_out_epoch_132.csv -------------------------------------------------------------------------------- /csv/test_out_epoch_75.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/test_out_epoch_75.csv -------------------------------------------------------------------------------- /csv/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/train.csv -------------------------------------------------------------------------------- /csv/train_balance_50000.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/train_balance_50000.csv -------------------------------------------------------------------------------- /csv/train_gan.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/train_gan.csv -------------------------------------------------------------------------------- /csv/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/val.py -------------------------------------------------------------------------------- /csv/validation2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/csv/validation2.py -------------------------------------------------------------------------------- /dr_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/dr_dataloader.py -------------------------------------------------------------------------------- /evaluate/.ipynb_checkpoints/evaluate-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/evaluate/.ipynb_checkpoints/evaluate-checkpoint.ipynb -------------------------------------------------------------------------------- /evaluate/evaluate.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/evaluate/evaluate.ipynb -------------------------------------------------------------------------------- /evaluate/output.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/evaluate/output.csv -------------------------------------------------------------------------------- /gcam/13502_left.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/gcam/13502_left.jpg -------------------------------------------------------------------------------- /gcam/2901_right.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/gcam/2901_right.jpg -------------------------------------------------------------------------------- /main.sh: -------------------------------------------------------------------------------- 1 | python train_densenet121.py > out.txt & 2 | 3 | -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/models/alexnet.py -------------------------------------------------------------------------------- /models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/models/densenet.py -------------------------------------------------------------------------------- /models/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/models/inception.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/retinet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/models/retinet.py -------------------------------------------------------------------------------- /models/squeezenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/models/squeezenet.py -------------------------------------------------------------------------------- /models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/models/vgg.py -------------------------------------------------------------------------------- /train_densenet121.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/train_densenet121.py -------------------------------------------------------------------------------- /train_multi_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/train_multi_gpu.py -------------------------------------------------------------------------------- /train_multi_gpu_resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/train_multi_gpu_resnet50.py -------------------------------------------------------------------------------- /train_test_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lg920810/Diabetic_Reinopathy_Dectection/HEAD/train_test_split.py --------------------------------------------------------------------------------