这次我们使用基于ResNet50的模型进行图片分类,这个模型是2015年由微软实验室的何恺明提出
ResNet网络提出了残差网络结构(Residual Network)来减轻退化问题,使用ResNet网络可以实现搭建较深的网络结构(突破1000层)ResNet网络层数越深,其训练误差和测试误差越小。

首先下载数据集,本次数据集一共有50000张训练图片和10000张评估图片,这里使用官方提供的`mindspore.dataset.Cifar10Dataset`接口来加载数据集,并进行相关图像增强操作。

 这里使用matplotlib库来显示图片,并用了numpy来处理数据,最后对数据集合进行可视化,最后训练完的预测准确率大概70%左右,这是5epoch的准确率
我调整了训练轮数.30轮的训练之下,准确率有提升,Accuracy提升到了0.8

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐