【CANN训练营】【新手班】直播课中基于Caffe ResNet-50网络实现图片分类（仅推理）任务实操笔记

最新推荐文章于 2022-12-17 22:24:25 发布

清谈い

最新推荐文章于 2022-12-17 22:24:25 发布

阅读量447

点赞数

文章标签： caffe 深度学习人工智能

本文链接：https://blog.csdn.net/qingtan610/article/details/126388389

版权

一、任务描述

该样例主要是基于Caffe ResNet-50网络（单输入、单Batch）实现图片分类的功能。

在该样例中：

二、具体实操

1、申请华为云服务器

2、下载任务资源仓中的样例代码

下载完成

定位到样例目录

3、准备ResNet-50模型。

获取ResNet-50原始模型。

您可以从以下链接中获取ResNet-50网络的模型文件（*.prototxt）、预训练模型文件（*.caffemodel），并以运行用户将获取的文件上传至开发环境的“样例目录/caffe_model“目录下。如果目录不存在，需要自行创建。
- ResNet-50网络的模型文件（*.prototxt）：https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/resnet50/resnet50.prototxt。
- ResNet-50网络的预训练模型文件（*.caffemodel）：https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/resnet50/resnet50.caffemodel。
- 使用wget命令
将ResNet-50原始模型转换为适配昇腾AI处理器的离线模型（*.om文件）。

切换到样例目录，执行如下命令(以昇腾310 AI处理器为例)：

模型转换成功

4、准备测试图片

请从以下链接获取该样例的输入图片，并以运行用户将获取的文件上传至开发环境的“样例目录/data“目录下。如果目录不存在，需自行创建。

https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/models/aclsample/dog1_1024_683.jpg

https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/models/aclsample/dog2_1024_683.jpg
切换到“样例目录/data“目录下，执行transferPic.py脚本，将*.jpg转换为*.bin，同时将图片从1024*683的分辨率缩放为224*224。在“样例目录/data“目录下生成2个*.bin文件。