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

数据准备

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

全是小狗的图片

另一边全是狼的图片

加载数据集

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

数据集可视化

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

训练模型

构建Resnet50网络

固定特征进行训练

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

总结

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

相关推荐
Elastic 中国社区官方博客1 分钟前
使用 LocalAI 和 Elasticsearch 构建本地 RAG 应用
大数据·数据库·人工智能·elasticsearch·搜索引擎·ai·全文检索
magic_ll2 分钟前
【大模型】使用llamafactory 训练 qwen2.5-VL 的目标检测任务
人工智能·目标检测·计算机视觉
Franklin3 分钟前
AI Coding 基础实践04:Pycharm 项目移植AI 工具 TRAE的history
人工智能
得贤招聘官9 分钟前
破局传统招聘:AI面试智能体构建精准高效新生态
大数据·人工智能·面试
胡萝卜3.09 分钟前
Linux包管理器:高效安装软件的秘诀
linux·运维·服务器·人工智能·linux包管理·yum教程·apt入门
~央千澈~9 分钟前
人工智能AI算法推荐之番茄算法推荐证实其算法推荐规则技术解析·卓伊凡
人工智能·算法·机器学习
执笔论英雄12 分钟前
【RL】ROLL中loss 计算compute_approx_kl
人工智能
sealaugh3215 分钟前
AI(学习笔记第十七课)langchain v1.0(SQL Agent)
人工智能·笔记·学习
zbguolei16 分钟前
使用VBA将EXCEL生成PPT
人工智能·opencv·计算机视觉
易百纳16 分钟前
易百纳携多模态AI桌面机器人——Kubee Robot亮相2025火山引擎冬季FORCE大会
人工智能·火山引擎