图像分类:这是对图像进行分类的深度学习模型 源码

此仓库具有我的图像分类模型开发的历史。 模型代码开始于从Kaggle下载胸部X射线数据集。 然后要预处理数据集,请删除损坏的图像。 基本模型由4组卷积和maxpooling层组成,然后是平坦层和2个致密层。 模型版本2与基本模型相似,但是增加了增强功能。 为了比较这些模型,针对时期绘制了精度曲线和损耗曲线。 另外,输入一组测试图像以测试训练后的模型,并绘制混淆矩阵和ROC曲线。