/IQA

Retinal Image IQA ~

Primary LanguagePython

Just a simple transfer learning for Retinal images IQA

Dataset is based on DRIMDB, model is Inception pre_trained on ImageNet

  • Dataset can be downloaded Here

  • Put dataset in the ./data/DRIMDB

  • Download the pre_trained model Here

  • Put the pre_trained model into ./model/pre_train/inception_v3_2016_08_28

  • go to ./src run

python3 main.py --prepare --GPU 0 --epoch 30

The final result on test_set is 100%

Unfinished on own data