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

数据准备

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

全是小狗的图片

另一边全是狼的图片

加载数据集

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

数据集可视化

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

训练模型

构建Resnet50网络

固定特征进行训练

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

总结

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

相关推荐
尹玉玺几秒前
从原子残片到天道公理:以德谟克利特思想全貌,重新定界西方唯物认知的历史层级与边界
人工智能·算法·架构
Raas1002 分钟前
2026年AI网关推荐?MAI Gateway(魔芋企业级AI网关)等主流方案横评
大数据·人工智能·ai·gateway·ai网关·mai gateway
狂师3 分钟前
AI Agent 评测:到底要测什么、新手怎么测?(附完整流程)
人工智能·llm·测试
sugar15697 分钟前
给企业系统装上 MCP:陀螺匠接入 AI 的落地实践与三个坑
人工智能
AI技术新视界8 分钟前
ds4 (DwarfStar )是什么?
人工智能·推理引擎·本地ai
IT_陈寒9 分钟前
深入理解Python:核心原理与最佳实践
前端·人工智能·后端
Data-Miner11 分钟前
DeepSeek 对百度搜索的冲击有多大?当搜索被 AI 重写,数据分析行业正在被悄悄改写
人工智能·百度·数据分析
电科_银尘12 分钟前
【电科_银尘】从电子基础到人工智能,分享能用得上的知识
人工智能
创世虚拟世界13 分钟前
把 3D 虚拟世界拉到本地:vibe coding 的手感,是从看见自己那一行改动开始的
javascript·数据库·人工智能·3d·gitee·虚拟现实
水如烟14 分钟前
孤能子视角:SEP《人工智能的认识论》EIS读法·证成——异质观察符校验:不是更严,是不同判据层
人工智能