You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
9 months ago | |
|---|---|---|
| modules | 9 months ago | |
| .gitignore | 9 months ago | |
| ImageTest.py | 9 months ago | |
| README.md | 9 months ago | |
| test.py | 9 months ago | |
| train.py | 9 months ago | |
README.md
所有自定义配置皆以常量形式定义在.py开头中,可以在项目启动前进行修改(主要是数据集路径与模型路径)。 其中,ImageTest.py为测试图片变换的代码(与模型训练无关),其中有CLAHE变换的函数,可输出变换前后的图片。 数据集以kaggle的Chest X-Ray Images (Pneumonia)数据集为基础,对训练集和测试集中的normal文件夹做了修改:1. 删去测试集中的normal文件夹中的图片,2. 将训练集中的normal文件夹中的部分图片剪切到测试集的normal文件夹中。(原因:实验时发现,训练时,模型在训练集和验证集的normal标签上准确率都很高,但是测试集上准确率却只有50%,怀疑测试集的normal文件夹中的图片标记有误)3.弃用数据集原本的验证集(样本过少),改为从训练集以9:1的比例划分成训练集和验证集。