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

数据准备

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

全是小狗的图片

另一边全是狼的图片

加载数据集

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

数据集可视化

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

训练模型

构建Resnet50网络

固定特征进行训练

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

总结

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

相关推荐
Profile排查笔记几秒前
有哪些开源指纹浏览器项目?先核对客户端、内核与许可证
前端·人工智能·后端·自动化
AI浩1 分钟前
BEVNeXt:复兴用于3D目标检测的稠密BEV框架
人工智能·目标检测·3d
镭封9 分钟前
AI文字转语音能制作多人对白吗?角色分配操作指南
人工智能·小程序·音视频·语音识别·媒体
Zero Lu12 分钟前
国内如何用 Opus 5.5 生成视频?Cyberbara Motion 案例与提示词分享
人工智能·aigc·音视频·提示词
文声图深圳科技有限公司13 分钟前
企业文档翻译,为什么翻完还得重排?(一套保住排版的处理流程)
人工智能
释迦呼呼14 分钟前
同一个函数套 96 遍:大模型的全部秘密,藏在一道初中函数题里
人工智能
loulanyue_18 分钟前
10000P国产算力破壁上线,算电协同与超智融合打破算力孤岛:高文院士图灵奖得主纵论世界算力网新纪元——读2026世界算力网大会实录
人工智能·大模型·架构设计·高性能计算·算力网络
Thneonl18 分钟前
0.18 也被 DENY:治理必须是台确定性机器
人工智能·架构
老猫聊AI20 分钟前
Qwen-image-2.1 本地部署保姆级教程 | 首页
人工智能·llm
FridaySunny23 分钟前
Agent 工作交接:继续、压缩还是新开?120 次运行的会话策略实验
人工智能