visualize.py 文件源码

def get_video_writer(args):
    # Define the codec and create VideoWriter object
    #fourcc = cv2.VideoWriter_fourcc(*'XVID')
    fourcc = cv2.VideoWriter_fourcc('P','I','M','1')
    return cv2.VideoWriter(args.video_path, fourcc, args.fps, (args.width, args.height))