使用opencv获取视频时长

import os
import cv2
import re
import logging

#定义函数获取视频路径及名称
def getVideoNames(path=os.path.realpath(__file__)):
    name = []    #存储视频绝对路径及名称
    count=0
    ''':param   root当前目录的路径,dirs为该目录下的子目录,files为非目录文件'''
    for root, dirs, files in os.walk(path):
        for file in files:
            if re.match(".+\.mp4",file):            #匹配mp4文件
                print(file)
                name.append(os.path.join(root, file))
                count+=1
            else:
                logging.warning(file+"不是mp4文件")
    print(root+"下共"+str(count)+"个mp4文件")
    return name

def getTotalTime(files):
    total=0
    for file_path in files:
        cap=cv2.VideoCapture(file_path)
        if cap.isOpened():
            rate = cap.get(5)       #获取帧率
            fraNum=cap.get(7)       #获取帧数
            duration=fraNum/rate/60
            total+=duration
    return total

files=getVideoNames('D:\CloudMusic\MV')
total=getTotalTime(files)
print("D:\CloudMusic\MV下视频总时长为:"+str(total))

 

你可能感兴趣的:(python学习笔记)