CNN 卷积神经网络对染色血液细胞分类(blood-cells)

目录

1. 介绍

2. 加载数据

3. 可视化

3.1 显示单幅图像

3.2 显示多幅图像

4. 定义CNN卷积神经网络

5. 网络训练

5.1 实例化网络 + 损失函数 + 优化器

5.2 网络训练

5.3 验证在测试集上的精度

5.4 打印结果 + 返回

6.  if __name__=='__main__'

7. 训练结果可视化

8. 网络在测试集的预测

8.1 导入的文件

8.2 预处理

8.3 加载模型 + class

8.4 加载测试集

8.5 预测图像

9. 代码

---

1. 介绍

项目目录如下：

demo.py 是主程序，包括了CNN网络的搭建和train过程

predict 是网络对 testSet 的验证

•  EOSINOPHIL   嗜酸性粒细胞
• LYMPHOCYTE   淋巴细胞
• MONOCYTE      单核细胞
• NEUTROPHIL    中性粒细胞

demo.py 需要的库文件如下：

 

2. 加载数据

对数据的预处理如下：

 

读取train和test文件的目录

这里通过Image