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

数据准备

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

全是小狗的图片

另一边全是狼的图片

加载数据集

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

数据集可视化

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

训练模型

构建Resnet50网络

固定特征进行训练

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

总结

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

相关推荐
DevNo1 分钟前
我的旅拍素材整理和剪辑记录
人工智能
程序员cxuan2 分钟前
DeepSeek V4.1 Flash 正式发布!
人工智能·后端·程序员
海宇大数据3 分钟前
分布式网格实战:基于海宇柠檬查出险-登记证构建自动化车辆合规网关
运维·人工智能·分布式·自动化
欧特克_Glodon10 分钟前
OpenCV计算机视觉开发入门与实践<四十>:自动识别物体数量
c++·人工智能·opencv·计算机视觉
世岩清上12 分钟前
展厅不同分区播放差异化视频,怎样做到内容串联不割裂整体叙事?
运维·人工智能·音视频·展厅改造
Dawson Zhu13 分钟前
Agent 记忆调用的上下文感知:从“能召回“到“敢开口“的决策框架
人工智能·语言模型·架构·aigc·agi
来让爷抱一个13 分钟前
2026 姿态控制实战:八个姿势钉死动作,百智云精灵图把骨骼契约写进素材包
人工智能·机器学习
龙亘川21 分钟前
数字化建设案例|构建议案履职一体化闭环,解决议案建议办理多主体协同难题
大数据·人工智能·科技·信息可视化·智慧城市
CyberVein21 分钟前
Dell暴涨之后,更加需要关注算力服务了
人工智能
童园管理札记22 分钟前
CSDN 学前入门高质量指南:从零搭建编程学习体系
人工智能·经验分享·职场和发展·生活·学习方法