yolov3用python接口检测并保存视频

本文直接修改darknet下python目录内的darknet.py来完成视频的检测与保存。在看本文章之前，请按照如下链接完成相关文件修改。
https://blog.csdn.net/phinoo/article/details/83009061
修改完之后确保可以运行没问题，修改darknet.py文件的main函数，下面把代码贴上。注意在抬头加上import cv2，里面关于路径的都修改为自己文件所在的路径。我用的是opencv3.4.6，因为保存视频的接口是opencv3的，注意下。

if __name__ == "__main__":
    net = load_net("../cfg/yolov3.cfg", "../backup/yolov3.weights", 0)
    meta = load_meta("../cfg/coco.data")
    vid = cv2.VideoCapture('../video/input2.avi')
    fourcc = cv2.VideoWriter_fourcc('M','P','4','2') #opencv3.0
    videoWriter = cv2.VideoWriter('../video/output2.avi', fourcc, 25, (1920,1080))
    while True:
        return_value,arr=vid.read()
        if not return_value:
            break 
        im=nparray_to_image(arr)
        boxes= detect(net, meta, im)
       
        for i in range(len(boxes)):
            score=boxes[i][1]
            label=boxes[i][0]
            xmin=boxes[i][2][0]-boxes[i][2][2]/2
            ymin=boxes[i][2][1]-boxes[i][2][3]/2
            xmax=boxes[i][2][0]+boxes[i][2][2]/2
            ymax=boxes[i][2][1]+boxes[i][2][3]/2
            cv2.rectangle(arr,(int(xmin),int(ymin)),(int(xmax),int(ymax)),(0,255,0),2)
            cv2.putText(arr,str(label),(int(xmin),int(ymin)),fontFace=cv2.FONT_HERSHEY_SIMPLEX,fontScale=0.8,color=(0,255,255),thickness=2)    
        cv2.imshow("Canvas", arr)
        videoWriter.write(arr) 
        cv2.waitKey(1) 
    cv2.destroyAllWindows()

用自带的darknet.py检测视频会比直接运行源程序速度慢一点，但是可以接受。