Atlas 200I DK A2视频保存

Luzichang

已于 2023-07-12 15:28:57 修改

阅读量1.2k

点赞数

文章标签：音视频计算机视觉 python

于 2023-07-12 11:34:57 首次发布

本文链接：https://blog.csdn.net/Luzichang/article/details/131677978

版权

文章讲述了在使用Atlas200IDK开发者套件进行目标检测时遇到的视频保存格式问题，原代码使用FFmpeg生成的MP4文件无法在默认播放器中打开，而VLC可以。作者参考了相关博客，通过修改代码，使用cv2的VideoWriter模块以AVI格式保存视频，解决了播放问题，并能在Jupyter环境中正常展示。

摘要由CSDN通过智能技术生成

Atlas 200I DK A2开发者套件内置案例第一个目标检测，视频保存下来只能用VLC打开，电脑自带的播放器打不开，应该是ffmpeg生成格式问题（参考：解决ffmpeg生成mp4文件不能正常预览的问题_mp4reader__无相_的博客-CSDN博客），修改为cv2保存可用自带播放器正常在本地展示。

原代码部分

def infer_video(video_path, model, labels_dict, cfg, output_path='output.mp4'):
    """视频推理"""
    image_widget = widgets.Image(format='jpeg', width=800, height=600)
    display(image_widget)

    # 读入视频
    cap = vreader(video_path)
    video_writer = None
    for img_frame in cap:
        # 对视频帧进行推理
        image_pred = infer_frame_with_vis(img_frame, model, labels_dict, cfg, bgr2rgb=False)
        image_widget.value = img2bytes(image_pred[:, :, ::-1])

        if video_writer is None:
            # 初始化视频写入对象
            video_writer = FFmpegWriter(output_path)
        # 将每一帧的预测结果写入视频对象
        video_writer.writeFrame(image_pred)
    video_writer.close()

视频打不开

VLC能打开

修改后（参考：Pyhon cv2.VideoWriter 保存视频_cv2保存视频_南洲.的博客-CSDN博客）

def infer_video(video_path, model, labels_dict, cfg, output_path='output.mp4'):
    """视频推理"""
    # 读入视频
    cap = cv2.VideoCapture(video_path)
    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
    fps = cap.get(cv2.CAP_PROP_FPS)
    fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V')
    outVideo = cv2.VideoWriter('save_test_video.avi', fourcc, fps, (width,height))
    
    image_widget = widgets.Image(format='jpeg', width=800, height=600)
    display(image_widget)
    
    while True:
        # 对视频帧进行推理
        ret, frame = cap.read()
        if not ret:
            print("... end of video file reached")
            break
        image_pred = infer_frame_with_vis(frame, model, labels_dict, cfg, bgr2rgb=False)
        image_widget.value = img2bytes(image_pred)
        outVideo.write(image_pred)

保存为avi格式，也能在jupyter中正常展示：