使用数据集Dermne,默认参数时测试准确率61.5%
|--data
|--train
|--AcneandRosaceaPhotos
...
...
|--test
|--gen_label_csv.py
|--model_v4.py
|--main_inception_v4.py
python gen_label_csv.py
python main_inception_v4.py
框架:Pytorch 0.4
使用的模型为ImageNet预训练的inception v4,整体思路:数据增强之后直接将网络全部层一起训练,做简单分类。
输入尺寸为384×384