Skip to content

Pytorch Implementation for Skin Disease Classification

Notifications You must be signed in to change notification settings

FelixZhang7/Derm-neu

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

皮肤病检测分类程序Derm-neu

使用数据集Dermne,默认参数时测试准确率61.5%

数据集可从DropBox下载train test

运行代码前,需要将图片放在data目录下,目录树如下:

|--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

About

Pytorch Implementation for Skin Disease Classification

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages