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

数据准备

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

全是小狗的图片

另一边全是狼的图片

加载数据集

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

数据集可视化

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

训练模型

构建Resnet50网络

固定特征进行训练

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

总结

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

相关推荐
AC赳赳老秦几秒前
医疗数据安全处理:DeepSeek实现敏感信息脱敏与结构化提取
大数据·服务器·数据库·人工智能·信息可视化·数据库架构·deepseek
木头程序员4 分钟前
机器学习模型成员推断攻击与防御:敏感数据保护实战指南
人工智能·机器学习
咋吃都不胖lyh4 分钟前
归因分析(Attribution Analysis)详解
大数据·人工智能
AI科技星6 分钟前
能量绝对性与几何本源:统一场论能量方程的第一性原理推导、验证与范式革命
服务器·人工智能·科技·线性代数·算法·机器学习·生活
浔川python社10 分钟前
浔川 AI 翻译 v6.0 版本合规优化公告:强化违规内容治理,明确恶意使用处置规则
人工智能
2401_8612775510 分钟前
中国电信星辰AI大模型有哪些主要功能
人工智能·云计算·软件工程·语音识别
夫唯不争,故无尤也14 分钟前
智能旅行助手agent:从零构建AI旅游推荐
人工智能·python·大模型开发
仙魁XAN14 分钟前
如何用豆包、即梦 AI ,快速实现“AI森林治愈系风格视频”的效果
人工智能·ai·视频生成·豆包·即梦·森林治愈系
Cigaretter715 分钟前
Day 42 简单CNN
python·深度学习·cnn
春日见16 分钟前
控制算法:PID算法
linux·运维·服务器·人工智能·驱动开发·算法·机器人