文件夹下所有视频按指定秒提取

16713720339

之前的帖子发的是按照指定帧提取图片，但是使用下来发现太慢了，因为要解析所有的帧。
之前的帖子 https://www.52pojie.cn/forum.php?mod=viewthread&tid=1726138&page=1#pid45082722
新特点：
1、这次用的指定秒提取，免得解析每一帧，减少时间
2、可以输入的是文件夹地址，也可以是单个视频文件的地址

[Python] 纯文本查看 复制代码

import cv2
import os
#图片截取
def get_frame_from_video(video_name, frame_time, img_dir, img_name):
    """
    get a specific frame of a video by time in milliseconds
    :param video_name: video name
    :param frame_time: time of the desired frame
    :param img_dir: path which use to store output image
    :param img_name: name of output image
    :return: None
    """
    vidcap = cv2.VideoCapture(video_name)
    # Current position of the video file in milliseconds.
    vidcap.set(cv2.CAP_PROP_POS_MSEC, frame_time - 1)
    # read(): Grabs, decodes and returns the next video frame
    success, image = vidcap.read()

    if not os.path.exists(img_dir):
        os.makedirs(img_dir)

    if success:
        # save frame as JPEG file
        #cv2.imwrite(img_dir + sep + img_name, image)  
        cv2.imencode('.jpg', image)[1].tofile(img_dir +'/'+ str(img_name)+'.jpg') ##正确方法
        # cv2.imshow("frame%s" % frame_time, image)
        # cv2.waitKey()



#得到相应的秒值列表
def qvmiaozhi(zongmiao,jiangemiao):
    a=[]
    miao = 1
    while miao<=zongmiao:
        a.append(miao)
        miao = int(miao) + int(jiangemiao)
    return a

#获取视频的秒时长
def get_video_miaoshu(filename):
  cap = cv2.VideoCapture(filename)
  if cap.isOpened():
    rate = cap.get(5)
    frame_num =cap.get(7)
    duration = frame_num/rate
    return duration
  return -1





if __name__ == "__main__":

    path_lujing = input('输入文件夹路径')

    im_file = path_lujing 
    jiangemiao = input('间隔时间')

    wenjian_houzhui = os.path.splitext(path_lujing)[-1] #获取输入的后缀
    #判断输入的是一个文件夹路径  还是  一个文件
    if wenjian_houzhui == '':

        for im_name in os.listdir(im_file):              #im_name 完整名字带后缀
            wenjian_ming = os.path.splitext(im_name)[0]        #
            wenjian_houzhui = os.path.splitext(im_name)[-1]
            #if wenjian_houzhui == 'mp4' or 'avi':
            #print(im_name)
        
            path = path_lujing +'/'+ im_name
            #print(path)
            
            img_dir = path_lujing + wenjian_ming + '/'
            #img_name = '666'
            #jiangemiao = 10


            zongmiao = get_video_miaoshu(path)
            maio_lie = qvmiaozhi(zongmiao,jiangemiao)



            #get_frame_from_video(video_name, frame_time, img_dir, img_name)

            print(im_name+'有'+str(zongmiao)+'秒')
            #print(maio_lie)
            print('保存'+str(len(maio_lie))+'张图片')
            n = 1
            for i in maio_lie:
                get_frame_from_video(path, i*1000, img_dir, i)
                print('已经保存完成'+str(n)+'张')
                n = n + 1

    else:   
            print('单文件')
            path = path_lujing
            #print(path)
            wenjian_ming = os.path.splitext(path)[0]
            img_dir = wenjian_ming + '/'
            #img_name = '666'
            #jiangemiao = 10


            zongmiao = get_video_miaoshu(path)
            maio_lie = qvmiaozhi(zongmiao,jiangemiao)



            #get_frame_from_video(video_name, frame_time, img_dir, img_name)

            print(path+'有'+str(zongmiao)+'秒')
            #print(maio_lie)
            print('保存'+str(len(maio_lie))+'张图片')
            n = 1
            for i in maio_lie:
                get_frame_from_video(path, i*1000, img_dir, i)
                print('已经保存完成'+str(n)+'张')
                n = n + 1

拣尽寒枝不肯栖

本人小白，很想用，就是不会用。代码弄进去，也出错

Enovelist

哈哈哈太牛感谢分享

柑桔

学习学习

hugicegrue

多谢楼主分享，码下细细学习

wuming4

收藏持续进行

sxtyys

学习学习

Wapj_Wolf

收藏为敬，感谢分享。

shouhui

不是软件不会用，纯支持了