keras加载h5模型文件,进行图像分类任务的结果预测。

本文介绍了一个皮肤病8分类任务,基于DenseNet121进行迁移学习并训练得到预训练模型model.h5。讨论了如何加载此模型以及使用skimage和keras的图像预处理模块进行预测,强调了两者在图像范围转换上的差异。
摘要由CSDN通过智能技术生成

这是一个皮肤病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模块来预处理图像:

加载

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值