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

数据准备

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

全是小狗的图片

另一边全是狼的图片

加载数据集

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

数据集可视化

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

训练模型

构建Resnet50网络

固定特征进行训练

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

总结

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

相关推荐
yongui478341 天前
MATLAB 二维方腔自然对流 SIMPLE 算法
人工智能·算法·matlab
翔云 OCR API1 天前
API让文档信息“活”起来:通用文档识别接口-开发者文字识别API
前端·数据库·人工智能·mysql·ocr
540_5401 天前
ADVANCE Day22_复习日
人工智能·python·机器学习
良策金宝AI1 天前
2025年度回顾:工程AI从“能用“走向“可信“的五大里程碑
人工智能
@鱼香肉丝没有鱼1 天前
Transformer原理—注意力机制
人工智能·深度学习·transformer·注意力机制
AAD555888991 天前
轴体分类识别:基于Decoupled-Solo-Light模型的中心轴、铁质轴和尼龙轴自动检测与分类系统
人工智能·分类·数据挖掘
新智元1 天前
全网破防,AI「手指难题」翻车逼疯人类!6 根手指,暴露 Transformer 致命缺陷
人工智能·openai
wa的一声哭了1 天前
拉格朗日插值
人工智能·线性代数·算法·机器学习·计算机视觉·自然语言处理·矩阵
小程故事多_801 天前
深度解析WeKnora,腾讯开源RAG框架如何重塑复杂文档的智能处理生态
人工智能·开源·aigc
七月丶1 天前
Cloudflare 🌏 中国大陆网络访问优化 - 0元成本
人工智能·react.js·设计模式