视频转为图片

项目中保存大量的视频,存在误识别的视频我们需要将其转为图片,从而重新做标签。

video2pic.py

import cv2
import argparse
import os


# 边里该文件夹下的文件名称
def read_directory(directory_name):
    file_list = []
    for filename in os.listdir(directory_name):
        str = directory_name + '/' + filename
        file_list.append(str)
    return file_list


# 设置处理的帧数与文件位置,并整合。
def parse_args(input, output):
    parser = argparse.ArgumentParser(description='Process pic')
    parser.add_argument('--input', help='video to process', dest='input', default=None, type=str)
    parser.add_argument('--output', help='pic to store', dest='output', default=None, type=str)
    # default为间隔多少帧截取一张图片
    parser.add_argument('--skip_frame', dest='skip_frame', help='skip number of video', default=75, type=int)
    # input为输入视频的路径 ,output为输出存放图片的路径
    args = parser.parse_args(['--input', input, '--output', output])
    return args


# 处理帧数函数
def process_video(i, i_video, o_video, num):
    cap = cv2.VideoCapture(i_video)
    num_frame = cap.get(cv2.CAP_PROP_FRAME_COUNT)
    print("该视频的总帧数为:", num_frame)
    expand_name = '.jpg'
    if not cap.isOpened():
        print("检查路径名")
    cnt = 0
    count = 0
    while 1:
        ret, frame = cap.read()
        cnt += 1
        if cnt % num == 0:
            count += 1
            cv2.imwrite(os.path.join(o_video, '59'+str(i) + '_' + str(count) + expand_name), frame)
        if not ret:
            break


if __name__ == '__main__':
    # 命名不要带中文,可能报错
    intput = read_directory("G:\\5-9\\59")
    output = 'G:\\5-9/pic'
    print("该目录下共有:", len(intput), "个视频")
    i = 0
    for input_i in intput:
        args = parse_args(input_i, output)
        print("开始输出第", i, "个视频:",args.input)
        process_video(i, args.input, args.output, args.skip_frame)
        print("第", i, "个视频处理完毕")
        print("-----------------------------------")
        i = i + 1
    print("汇总", len(intput), "个视频处理完毕")
  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值