这是一个皮肤病8分类任务。首先展示类别名称列表:
classes_name_list=['AK','BCC','BKL','DF','MEL','NV','SCC','VASC']
以上代码表明,分类任务为8分类,标签分别为AK,BCC,BKL等等。
通过对DenseNet121进行迁移学习,将顶层修改为自己设计的8分类器,并在训练集上训练,(训练图像全部归一化为0-1范围内),获得训练好的模型:model.h5
加载该预训练模型:
import tensorflow as tf
model=tf.keras.models.load_model('model.h5')
该模型结构如下图:
1.首先我们使用skimage模块来预处理图像:
加载