上一篇用MindStudio转换完yolov8的om模型,准备在板卡里进行推理验证了。
我这里用的是我们自己的Tofu5m模块,40mm×40mm含外壳尺寸。可以输入网络RTSP视频流直接进行推理。这次用hi3516dv500版本的Tofu5m模块。
SDK里的demo部分是H.264视频输入,为了保持一致来验证,先改成H.264的视频流文件。
这里采用FFMPEG方式把本地文件转成H.264流出来。
ffmpeg -i video.avi -vcodec copy video.h264
首先在参数配置中选择对应的模型。
/* function : show the sample of yolo */
td_void sample_svp_npu_acl_yolo(td_u32 index) //yolo 推理
配置采用的模型类型。
sample_svp_npu_acl_set_detection_info(&detection_info, index);//detection_info包含目标类型数量,类型名称
static td_void sample_svp_npu_acl_set_detection_info(sample_svp_npu_detection_info *detection_info, td_u32 index)
{
detection_info->num_name =