使用python将视频按照帧转为图片

本文介绍了如何使用Python的OpenCV库将视频转换为批量图片,以供深度学习数据集制作使用。通过设置帧间隔,可以控制图片的输出频率。此外,作者还预告了后续将分享如何使用Labelme创建数据集,并将.json转换为.txt,适用于YOLO训练。
摘要由CSDN通过智能技术生成
  1. 前言:做深度学习,常常需要自己制作数据集,因此如何将视频转化为图片,将图片制作成相应格式的数据集,是一个基础。
    本文旨在解决将视频转成批量的图片,废话不多说,亲测完成的。
import cv2

video_path = 'F:/BaiduNetdiskDownload/video.mp4'  # 视频地址
output_path = 'F:/BaiduNetdiskDownload/'  # 输出文件夹
interval = 10  # 每间隔10帧取一张图片

if __name__ == '__main__':
    num = 1
    video = cv2.VideoCapture(video_path)
    fps = video.get(cv2.CAP_PROP_FPS)#帧率
    frames = video.get(cv2.CAP_PROP_FRAME_COUNT)#总的帧数
    print("fps=",int(fps),"frames=",int(frames))#打印一下(以上三行可以不要,写在这里是为了大概知道视频会输出多少张图片来合理的取帧的间隔)
    while video.isOpened():
        is_read, frame = video.read()
        if is_read:
            if num % interval == 1:
                file_name = '%d' % num
                cv2.imwrite(output_path + str(file_name) + '.jpg', frame)#输出图片
                cv2.waitKey(1)
            num += 1
        else:
            break

如果觉得本文帮助到你,就点个赞吧。关注我之后会更新如何用labelme制作数据集,然后将.json转成.txt,可以用于yolo训练。

  • 6
    点赞
  • 27
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值