【MindStudio训练营第一期】课程笔记3-实例操作:基于Caffe ResNet-50网络实现图片分类
收藏回复举报
【MindStudio训练营第一期】课程笔记3-实例操作:基于Caffe ResNet-50网络实现图片分类
发表于2022-12-30 23:14:52
0 查看

前言

老师在进阶班课程中讲解了一个课程实例,讲的是基于Caffe ResNet-50网络实现图片分类,而且步骤比较基础和重要,在此记录下来。

架结构

这个可以直接在源文件中的readme中查看得到,对源文件的每一个文件夹都有着详细的讲解。

1.jpg

案例目标

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

在该样例中:

  • 先使用样例提供的脚本transferPic.py,将2张*.jpg图片都转换为*.bin格式,同时将图片从1024*683的分辨率缩放为224*224。
  • 加载离线模型om文件,对2张图片进行同步推理,分别得到推理结果,再对推理结果进行处理,输出top5置信度的类别标识。
  • 在加载离线模型前,提前将Caffe ResNet-50网络的模型文件转换为适配昇腾AI处理器的离线模型。

准备工作

        查看SSH配置

%E5%9B%BE1.jpg

 确认无误后进行下一步

        点击Tools-Start SSH session

%E5%9B%BE2.jpg

     控制远程远端的terminal

%E5%9B%BE3.jpg

 创建一个名为test1的文件夹并进入

mkdir test1 

cd test1
下载模型文件、权重文件

wget https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/resnet50/resnet50.prototxt --no-check-certificate 

wget https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/resnet50/resnet50.caffemodel --no-check-certificate

%E5%9B%BE4.jpg

 转换为离线om模型

按照图片进行操作

%E5%9B%BE6.jpg

选择刚下载好的文件

%E5%9B%BE9.jpg

%E5%9B%BE10.jpg

开始模型转换

%E5%9B%BE11.jpg

%E5%9B%BE12.jpg

转换成功,记住最终的文件路径

%E5%9B%BE13.jpg

 图片jpg转换为bin格式

在resnet50_imagenet_classification下新建model文件夹

%E5%9B%BE14.jpg

%E5%9B%BE15.jpg

在resnet50_imagenet_classification下新建model文件夹

%E5%9B%BE16.jpg

%E5%9B%BE17.jpg

下载两张图片放入data文件夹 

https://obs-9be7.obs.cn-east2.myhuaweicloud.com/models/aclsample/dog1\_1024\_683.jpg

https://obs-9be7.obs.cn-east2.myhuaweicloud.com/models/aclsample/dog2\_1024\_683.jpg

%E5%9B%BE18.jpg

 进入data文件夹

cd .\cplusplus\level2_simple_inference\1_classification\resnet50_imagenet_classification\data

进入build配置,到远端执行

%E5%9B%BE20.jpg

%E5%9B%BE21.jpg进行远程编译

%E5%9B%BE22.jpg
 File-Setting中找到远端工程地址%E5%9B%BE23.jpg

%E5%9B%BE24.jpg

进入该目录

/root/tmp/18d49515-4b15-4c11-b057-d61fd5cfe553

输入代码将图片从jpg转换为bin格式

cd data/ 

python3 ../script/transferPic.py 

%E5%9B%BE26.jpg

 远端服务器运行cmake

%E5%9B%BE26.jpg

%E5%9B%BE27.jpg
用命令行运行

./main

最终结果

%E5%9B%BE28.jpg

本帖最后由 匿名用户2025/04/01 20:16:46 编辑

我要发帖子