样例步骤适用于以下产品。
从以下链接获取ResNet-50网络的权重文件(*.caffemodel)、模型文件(resnet50.prototxt),并以HwHiAiUser(运行用户)将获取的文件上传至开发环境的“vdec_resnet50_classification/caffe_model”目录下。
切换到“vdec_resnet50_classification”目录,执行如下命令。Ascendxxx为使用的昇腾AI处理器型号,请用户自行替换。
atc --model=caffe_model/resnet50.prototxt --weight=caffe_model/resnet50.caffemodel --framework=0 --output=model/resnet50_aipp --soc_version=Ascendxxx --insert_op_conf=caffe_model/aipp.cfg
通过链接vdec_h265_1frame_rabbit_1280x720.h265获取输入视频码流文件“vdec_h265_1frame_rabbit_1280x720.h265”,并以HwHiAiUser(运行用户)上传至开发环境的“vdec_resnet50_classification/data”目录下。
python3 ./src/acl_sample.py
init resource stage: init resource stage success [Model] class Model init resource stage: [Model] create model output dataset: [Model] create model output dataset success [Model] class Model init resource stage success [Dvpp] class Dvpp init resource stage: [Dvpp] class Dvpp init resource stage success [Vdec] class Vdec init resource stage: [Vdec] class Vdec init resource stage success [Vdec] forward index:0 [Vdec] create input stream desc success [Vdec] create output pic desc success [Vdec] vdec_send_frame stage success ...... [Vdec] [_callback] _callback exit success [Vdec] [_thread_func] _thread_func out [Vdec] vdec finish!!! [Dvpp] vpc resize stage: [Dvpp] vpc resize stage success [Model] create model input dataset: [Model] create model input dataset success [Model] execute stage: [Model] execute stage success ========= top5 inference results: ========= label:331 prob: 0.910156 label:330 prob: 0.078308 label:104 prob: 0.009209 label:332 prob: 0.003283 label:350 prob: 0.000005 result: class_label[331],top1[0.910156],top5[1.000000] ...... [Sample] release source stage: [Dvpp] class Dvpp release source success [Model] class Model release source success [Vdec] release resource: [Vdec] release resource success [Sample] release source stage success