样例介绍
功能描述
该样例主要是基于Caffe ResNet-50网络(单输入、单Batch)实现图片分类的功能。
根据运行应用的入参,该样例可实现以下功能:
- 将*.jpg格式的图片解码成YUV420SP NV12格式的输入图片,缩放,再进行模型推理,得到推理结果后,处理推理结果,输出top5置信度。
- 将YUV420SP NV12格式的输入图片,抠图,再进行模型推理,得到推理结果后,处理推理结果,输出top5置信度。
- 将YUV420SP NV12格式的输入图片,抠图贴图,再进行模型推理,得到推理结果后,处理推理结果,输出top5置信度。
- 将YUV420SP NV12格式的输入图片,编码为jpg格式的图片。
- 将YUV420SP NV12格式的输入图片(分辨率8192*8192)缩放,得到4000*4000的子图。
- 将YUV420SP NV12格式的输入图片,进行批量抠图。
- 将YUV420SP NV12格式的输入图片,进行批量抠图粘贴。
主要接口
主要接口如表1所示。
目录结构
目录结构如下所示。
vpc_jpeg_resnet50_imagenet_classification ├──scripts │ ├── host_version.conf //版本号配置文件。 │ └── testcase_300.sh //运行脚本。 ├──src │ ├── acl_dvpp_process.py //媒体数据处理。 │ ├── acl_model_process.py //模型运行文件。 │ ├── constant.py //常量定义。 │ └── main.py //运行文件,流程逻辑控制。 ├── data │ ├── fusion_result.json //atc转化后生成,记录融合算子信息。 │ └── vdec_h265_1frame_rabbit_1280x720.h265 //用户待处理的视频文件,由用户自行获取。 ├── caffe_model //需要用户部署。 │ ├── aipp.cfg //模型的配置数据。 │ ├── resnet50.caffemodel //ResNet-50模型。 │ └── resnet50.prototxt // ResNet-50模型的网络文件。 └── model //推理模型转换后生成的目录。 └── resnet50_aipp.om //转换后的模型文件。