人工智能-项目实践-图像识别-keras使用迁移学习实现医学图像二分类(AK、SK)
问题描述
要解决的是一个医学图像的二分类问题,有AK和SK两种病症,根据一定量数据,进行训练,对图像进行预测。
解决思路
整体上采用迁移学习来训练神经网络,使用InceptionV3结构,框架采用keras.
具体思路:
读取图片数据,保存成.npy格式,方便后续加载
标签采用one-hot形式,由于标签隐藏在文件夹命名中,所以需要自行添加标签,并保存到.npy文件中,方便后续加载
将数据分为训练集、验证集、测试集
使用keras建立InceptionV3基本模型,不包括顶层,使用预训练权重,在基本模型的基础上自定义几层神经网络,得到最后的模型,对模型进行训练
优化模型,调整超参数,提高准确率
在测试集上对模型进行评估,使用精确率、召回率
对单张图片进行预测,并输出每种类别的概率