卷积神经网络之ResNet50迁移学习

数据准备

下载狗与狼分类数据集,数据来自ImageNet,每个分类有大约120张训练图像与30张验证图像。使用download接口下载数据集,并自动解压到当前目录。

全是小狗的图片

另一边全是狼的图片

加载数据集

狼狗数据集提取自ImageNet分类数据集,使用mindspore.dataset.ImageFolderDataset接口来加载数据集,并进行相关图像增强操作。

数据集可视化

训练数据集通过MindSpore的ImageFolderDataset接口加载,返回值为字典。用户可以通过create_dict_iterator接口创建数据迭代器,使用next迭代访问数据集。在本章中,每次使用next可获取18个图像及标签数据。

训练模型

构建Resnet50网络

固定特征进行训练

训练和评估
可视化模型预测

总结

使用迁移学习方法对ImageNet数据集中的狼和狗图像进行分类的案例。首先介绍了数据集的下载和预处理操作,然后使用ResNet50模型进行训练和验证,最后保存了精度最高的模型参数。同时也展示了预测结果的可视化以及固定特征进行训练的方法。

相关推荐
点云侠1 分钟前
三维PCA原理及计算步骤
人工智能·机器学习
Kun Li2 分钟前
OmniText: A Training-Free Generalist for Controllable Text-Image Manipulation
人工智能·图片生成
PNP Robotics3 分钟前
连接AI产业·链动全球|PNP机器人亮相2026杭州全球人工智能大会
人工智能·python·学习·开源
Dev7z5 分钟前
面向健身与康复训练的基于深度学习的人体姿态检测与动作纠正系统
人工智能·深度学习·健身·康复训练·人体姿态检测·动作纠正系统
咚咚王者6 分钟前
人工智能之语言领域 自然语言处理 第七章 命名实体识别
人工智能·自然语言处理
我材不敲代码7 分钟前
计算机视觉基础——opencv的基础操作
人工智能·opencv·计算机视觉
搬砖者(视觉算法工程师)7 分钟前
用直白语言讲透 Transformer
人工智能
VALENIAN瓦伦尼安教学设备7 分钟前
便携式蒸汽阀门漏气检测仪作用
人工智能·嵌入式硬件·算法
成都它思科技有限公司8 分钟前
语音识别错误率是多少,我们认为错误率是多少?
人工智能·语音识别
Cx330❀8 分钟前
Linux ELF格式与可执行程序加载全解析:从磁盘文件到运行进程
linux·运维·服务器·人工智能·科技