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

数据准备

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

全是小狗的图片

另一边全是狼的图片

加载数据集

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

数据集可视化

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

训练模型

构建Resnet50网络

固定特征进行训练

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

总结

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

相关推荐
AI精钢17 小时前
Hermes Agent 整合 OpenCode CLI 的实战经验
人工智能·云原生·aigc
多年小白17 小时前
AI 日报 - 2026年4月25日(周六)
网络·人工智能·科技·深度学习·ai
互联科技报17 小时前
从形似到神离:解码超级编导、超级智剪、筷子科技的架构分野与云混剪2.0的范式革命
人工智能·线性代数·矩阵
唐兴通个人17 小时前
国机集团央企出海样本:苏美达如何用AI重构TOB销售链路与全球品牌能见度,AI出海营销培训师专家
人工智能
周末也要写八哥17 小时前
大语言模型的“自我迭代”
人工智能·语言模型·自然语言处理
生成论实验室17 小时前
生命降U:从分子共鸣到觉知涌现
人工智能·科技·架构·生活·信息与通信
晓山清17 小时前
CCF评级AI方向整理
人工智能·人机交互·aaai·普适计算
zhenxin012217 小时前
GPU算力梯队:从入门到超算的AI任务匹配指南
人工智能·gpu算力
MediaTea17 小时前
AI 术语通俗词典:F1 值(分类)
人工智能·算法·机器学习·分类·数据挖掘
深度学习lover18 小时前
<数据集>yolo 垃圾识别<目标检测>
人工智能·深度学习·yolo·目标检测·计算机视觉·垃圾识别