模型后处理部分

[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