Pytorch版 YOLO v3，中的视频保存

最新推荐文章于 2025-07-03 08:26:01 发布

Sunrisher

最新推荐文章于 2025-07-03 08:26:01 发布

阅读量2.2k

点赞数 1

CC 4.0 BY-SA版权

文章标签： YOLOv3 Pytorch 视频保存

本文链接：https://blog.youkuaiyun.com/weixin_40373414/article/details/82415272

本文介绍如何使用YOLOv3进行视频的目标检测，并针对运行视频文件后无法保存处理结果的问题提供了解决方案。通过修改video.py文件，利用OpenCV库成功实现了视频的保存。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近在学YOLOv3

YOLOv3，于是看 “从零开始PyTorch项目：YOLO v3目标检测实现”

英文原版：https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/

网上也有许多翻译

原版教程中的github项目地址 https://github.com/ayooshkathuria/YOLO_v3_tutorial_from_scratch

遇到的问题：用YOLOv3 运行视频文件，处理后的视频没有保存

解决办法：修改video.py 文件，用opencv，保存视频文件

详细步骤：

1. 在while循环前添加

fourcc = cv2.VideoWriter_fourcc('M','J','P','G')
fps = 24
savedPath = 'saveVideo.avi'  # 保存地址
ret, frame = cap.read()
videoWriter = cv2.VideoWriter(savedPath, fourcc, fps,(frame.shape[1], frame.shape[0])) # 最后为视频图片的形状

while cap.isOpened():
    ret, frame = cap.read()
    
    if ret:

## 中间省略，之后修改部分都有注释

    list(map(lambda x: write(x, frame), output))
    
    cv2.imshow("frame", frame)
    videoWriter.write(frame)           # 每次循环，写入该帧

    key = cv2.waitKey(1)
    if key & 0xFF == ord('q'):
        break
    frames += 1
    print(time.time() - start)
    print("FPS of the video is {:5.2f}".format( frames / (time.time() - start)))
else:
    videoWriter.release()              # 结束循环的时候释放
    break

不过这么写，相当于扔了视频第一frame，强迫症患者，可以优化一下。