上一篇用MindStudio转换完yolov8的om模型,准备在板卡里进行推理验证了。

我这里用的是我们自己的Tofu5m模块,40mm×40mm含外壳尺寸。可以输入网络RTSP视频流直接进行推理。这次用hi3516dv500版本的Tofu5m模块。

海思SD3403,SS928/926,hi3519dv500,hi3516dv500移植yolov7,yolov8(12)-LMLPHP

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 =
01-30 07:43