模型后处理部分

[omaiyiwa]

该怎么转换后的yolov5模型，调用摄像头，检测模型中的某个类别，在屏幕上只画这个类别的框，保存检测图片 目前我是这样做，能只输出人信息，但是屏幕会画其它框。 import time from ax import pipeline

pipeline.load([ 'libsample_vin_ivps_joint_vo_sipy.so', '-p', '/home/config/yolov5s.json', '-c', '2', ])

while pipeline.work(): time.sleep(0.001) tmp = pipeline.result()

print(tmp)

if tmp and tmp['nObjSize']:
    for i in tmp['mObjects']:
        if i['label'] == 0:
            print(i)
        else:
            pass