Как узнать продолжительность видео на Python?

Мне нужно получить продолжительность видео в Python. Мне нужно получить следующие форматы видео: MP4, Flash-видео, AVI и MOV... У меня есть решение для общего хостинга, поэтому у меня нет поддержки FFmpeg.

16 ответов

Решение

Вам, вероятно, потребуется вызвать внешнюю программу. ffprobe может предоставить вам эту информацию:

import subprocess

def getLength(filename):
  result = subprocess.Popen(["ffprobe", filename],
    stdout = subprocess.PIPE, stderr = subprocess.STDOUT)
  return [x for x in result.stdout.readlines() if "Duration" in x]

(ответ 2020 года)

Решения:

  1. opencv 0,0065 с ✔
  2. ffprobe 0,0998 с
  3. moviepy 2.8239 с

✔ Метод OpenCV:

def with_opencv(filename):
    import cv2
    video = cv2.VideoCapture(filename)

    duration = video.get(cv2.CAP_PROP_POS_MSEC)
    frame_count = video.get(cv2.CAP_PROP_FRAME_COUNT)

    return duration, frame_count

Применение: print(with_opencv('my_video.webm'))


Другие:

ffprobe метод:

def with_ffprobe(filename):
    import subprocess, json

    result = subprocess.check_output(
            f'ffprobe -v quiet -show_streams -select_streams v:0 -of json "{filename}"',
            shell=True).decode()
    fields = json.loads(result)['streams'][0]

    duration = fields['tags']['DURATION']
    fps      = eval(fields['r_frame_rate'])
    return duration, fps

moviepy метод:

def with_moviepy(filename):
    from moviepy.editor import VideoFileClip
    clip = VideoFileClip(filename)
    duration       = clip.duration
    fps            = clip.fps
    width, height  = clip.size
    return duration, fps, (width, height)

Как сообщается здесь https://www.reddit.com/r/moviepy/comments/2bsnrq/is_it_possible_to_get_the_length_of_a_video/

Вы могли бы использовать модуль moviepy

from moviepy.editor import VideoFileClip
clip = VideoFileClip("my_video.mp4")
print( clip.duration )

Найдите эту новую библиотеку Python: https://github.com/sbraz/pymediainfo

Чтобы получить продолжительность:

from pymediainfo import MediaInfo
media_info = MediaInfo.parse('my_video_file.mov')
#duration in milliseconds
duration_in_ms = media_info.tracks[0].duration

Приведенный выше код проверен на наличие действительного файла mp4 и работает, но вы должны сделать больше проверок, потому что он сильно зависит от вывода MediaInfo.

Чтобы сделать вещи немного проще, следующие коды помещают вывод в JSON.

Вы можете использовать его с помощью probe(filename)или получить продолжительность с помощью duration(filename):

json_info     = probe(filename)
secondes_dot_ = duration(filename) # float number of seconds

Работает на Ubuntu 14.04 где конечно ffprobe установлены. Код не оптимизирован для скорости или красивых целей, но он работает на моей машине, надеюсь, это поможет.

#
# Command line use of 'ffprobe':
#
# ffprobe -loglevel quiet -print_format json \
#         -show_format    -show_streams \
#         video-file-name.mp4
#
# man ffprobe # for more information about ffprobe
#

import subprocess32 as sp
import json


def probe(vid_file_path):
    ''' Give a json from ffprobe command line

    @vid_file_path : The absolute (full) path of the video file, string.
    '''
    if type(vid_file_path) != str:
        raise Exception('Gvie ffprobe a full file path of the video')
        return

    command = ["ffprobe",
            "-loglevel",  "quiet",
            "-print_format", "json",
             "-show_format",
             "-show_streams",
             vid_file_path
             ]

    pipe = sp.Popen(command, stdout=sp.PIPE, stderr=sp.STDOUT)
    out, err = pipe.communicate()
    return json.loads(out)


def duration(vid_file_path):
    ''' Video's duration in seconds, return a float number
    '''
    _json = probe(vid_file_path)

    if 'format' in _json:
        if 'duration' in _json['format']:
            return float(_json['format']['duration'])

    if 'streams' in _json:
        # commonly stream 0 is the video
        for s in _json['streams']:
            if 'duration' in s:
                return float(s['duration'])

    # if everything didn't happen,
    # we got here because no single 'return' in the above happen.
    raise Exception('I found no duration')
    #return None


if __name__ == "__main__":
    video_file_path = "/tmp/tt1.mp4"
    duration(video_file_path) # 10.008

Используйте современный метод с https://github.com/kkroening/ffmpeg-python ( pip install ffmpeg-python --user). Не забудьте установить ffmpegслишком.

Получить информацию о видео:

      import ffmpeg

info=ffmpeg.probe(filename)

print(f"duration={info['format']['duration']}")
print(f"framerate={info['streams'][0]['avg_frame_rate']}")

Использовать ffmpeg-pythonпакет, чтобы также легко создавать, редактировать и применять фильтры к видео.

from subprocess import check_output

file_name = "movie.mp4"

#For Windows
a = str(check_output('ffprobe -i  "'+file_name+'" 2>&1 |findstr "Duration"',shell=True)) 

#For Linux
#a = str(check_output('ffprobe -i  "'+file_name+'" 2>&1 |grep "Duration"',shell=True)) 

a = a.split(",")[0].split("Duration:")[1].strip()

h, m, s = a.split(':')
duration = int(h) * 3600 + int(m) * 60 + float(s)

print(duration)

Как сообщается здесь https://www.reddit.com/r/moviepy/comments/2bsnrq/is_it_possible_to_get_the_length_of_a_video/

вы можете использовать модуль Moviepy

from moviepy.editor import VideoFileClip 
clip = VideoFileClip("my_video.mp4") 
print( clip.duration )

Если вы пытаетесь получить продолжительность многих видео в папке, произойдет сбой с сообщением об ошибке:AttributeError: объект 'AudioFileClip' не имеет атрибута 'reader'

Итак, чтобы избежать этого, вам нужно добавить

clip.close()

Исходя из этого: https://zulko.github.io/moviepy/_modules/moviepy/video/io/VideoFileClip.html

Итак, код будет выглядеть так:

from moviepy.editor import VideoFileClip
clip = VideoFileClip("my_video.mp4")
print( clip.duration )
clip.close()

Ура!:)

Функция, которую я придумал. Это в основном использует только ffprobe аргументы

from subprocess import  check_output, CalledProcessError, STDOUT 


def getDuration(filename):

    command = [
        'ffprobe', 
        '-v', 
        'error', 
        '-show_entries', 
        'format=duration', 
        '-of', 
        'default=noprint_wrappers=1:nokey=1', 
        filename
      ]

    try:
        output = check_output( command, stderr=STDOUT ).decode()
    except CalledProcessError as e:
        output = e.output.decode()

    return output


fn = '/app/648c89e8-d31f-4164-a1af-034g0191348b.mp4'
print( getDuration(  fn ) )

Продолжительность выходов, как это:

7.338000

Ссылаясь на ответ @Nikolay user7673116 с использованием opencv-python (cv2):

У меня его метод не сработал (Python 3.8.10, opencv-python==4.5.5.64) и в комментариях сказано, что opencv в данном случае использовать нельзя, что тоже неверно.

CAP_PROP_POS_MSEC дает вам миллисекунду текущего кадра, в котором находится VideoCapture, а не общее количество миллисекунд видео, поэтому при простой загрузке видео это, очевидно, 0.

Но на самом деле мы можем получить частоту кадров и общее количество кадров, чтобы рассчитать общее количество миллисекунд видео:

      import cv2

video = cv2.VideoCapture("video.mp4")

# the frame rate or frames per second
frame_rate = video.get(cv2.CAP_PROP_FPS)

# the total number of frames
total_num_frames = video.get(cv2.CAP_PROP_FRAME_COUNT)

# the duration in seconds
duration = total_num_frames / frame_rate

Приведенный выше ответ pymediainfo мне действительно помог. Спасибо.

Новичку действительно потребовалось время, чтобы выяснить, чего не хватает (sudo apt install mediainfo) и как адресовать атрибуты другими способами (см. Ниже).

Отсюда этот дополнительный пример:

# sudo apt install mediainfo
# pip3 install pymediainfo
from pymediainfo import MediaInfo
media_info = MediaInfo.parse('/home/pi/Desktop/a.mp4')
for track in media_info.tracks:
    #for k in track.to_data().keys():
    #    print("{}.{}={}".format(track.track_type,k,track.to_data()[k]))
    if track.track_type == 'Video':
        print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
        print("{} width                 {}".format(track.track_type,track.to_data()["width"]))
        print("{} height                {}".format(track.track_type,track.to_data()["height"]))
        print("{} duration              {}s".format(track.track_type,track.to_data()["duration"]/1000.0))
        print("{} duration              {}".format(track.track_type,track.to_data()["other_duration"][3][0:8]))
        print("{} other_format          {}".format(track.track_type,track.to_data()["other_format"][0]))
        print("{} codec_id              {}".format(track.track_type,track.to_data()["codec_id"]))
        print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
    elif track.track_type == 'Audio':
        print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
        print("{} format                {}".format(track.track_type,track.to_data()["format"]))
        print("{} codec_id              {}".format(track.track_type,track.to_data()["codec_id"]))
        print("{} channel_s             {}".format(track.track_type,track.to_data()["channel_s"]))
        print("{} other_channel_s       {}".format(track.track_type,track.to_data()["other_channel_s"][0]))
        print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
print("********************************************************************")
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Video width                 1920
Video height                1080
Video duration              383.84s
Video duration              00:06:23
Video other_format          AVC
Video codec_id              avc1
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Audio format                AAC
Audio codec_id              mp4a-40-2
Audio channel_s             2
Audio other_channel_s       2 channels
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

Откройте терминал cmd и установите пакет python: mutagen используя эту команду

python -m pip install mutagen

затем используйте этот код, чтобы получить продолжительность видео и его размер:

import os
from mutagen.mp4 import MP4

audio = MP4("filePath")

print(audio.info.length)
print(os.path.getsize("filePath"))

Вот что я использую в prod сегодня, используя способ cv2, который хорошо работает для mp4, wmv и flv, что мне и нужно:

      try:
    import cv2  # opencv-python - optional if using ffprobe
except ImportError:
    cv2 = None

import subprocess

def get_playback_duration(video_filepath, method='cv2'):  # pragma: no cover
    """
    Get video playback duration in seconds and fps
    "This epic classic car collection centres on co.webm"
    :param video_filepath: str, path to video file
    :param method: str, method cv2 or default ffprobe
    """
    if method == 'cv2':  # Use opencv-python
        video = cv2.VideoCapture(video_filepath)
        fps = video.get(cv2.CAP_PROP_FPS)
        frame_count = video.get(cv2.CAP_PROP_FRAME_COUNT)
        duration_seconds = frame_count / fps if fps else 0
    else:  # ffprobe
        result = subprocess.check_output(
            f'ffprobe -v quiet -show_streams -select_streams v:0 -of json "{video_filepath}"', shell=True).decode()
        fields = json.loads(result)['streams'][0]
        duration_seconds = fields['tags'].get('DURATION')
        fps = eval(fields.get('r_frame_rate'))
    return duration_seconds, fps

ffprobeне работает для flvи я не мог заставить ничего работать webm. В противном случае это прекрасно работает и сегодня используется в производстве.

Для тех, кто любит использовать программу mediainfo:

import json
import subprocess

#===============================
def getMediaInfo(mediafile):
    cmd = "mediainfo --Output=JSON %s"%(mediafile)
    proc = subprocess.Popen(cmd, shell=True,
        stderr=subprocess.PIPE, stdout=subprocess.PIPE)
    stdout, stderr = proc.communicate()
    data = json.loads(stdout)
    return data

#===============================
def getDuration(mediafile):
    data = getMediaInfo(mediafile)
    duration = float(data['media']['track'][0]['Duration'])
    return duration

Вот метод, который я получил. Способ cv2 хорошо работает для mp4, wmv и flv, что мне и нужно:

      try:
    import cv2  # opencv-python - optional if using ffprobe
except ImportError:
    cv2 = None

import subprocess

def get_playback_duration(video_filepath, method='cv2'):  # pragma: no cover
    """
    Get video playback duration in seconds and fps
    """
    if method == 'cv2':  # Use opencv-python
        video = cv2.VideoCapture(video_filepath)
        fps = video.get(cv2.CAP_PROP_FPS)
        duration_seconds = video.get(cv2.CAP_PROP_FRAME_COUNT) / fps
    else:
        result = subprocess.check_output(f'ffprobe -v quiet -show_streams -select_streams v:0 -of json '
                                         f'"{video_filepath}"', shell=True).decode()
        fields = json.loads(result)['streams'][0]
        duration_seconds = fields['tags']['DURATION']
        fps = eval(fields['r_frame_rate'])
    return duration_seconds, fps

ffprobeне работает для flv.

Используя ffprobe в функции, он возвращает продолжительность видео в секундах.

def video_duration(filename):
    import subprocess
    secs = subprocess.check_output(f'ffprobe -v error -select_streams v:0 -show_entries stream=duration -of default=noprint_wrappers=1:nokey=1 "{filename}"', shell=True).decode()
    return secs
Другие вопросы по тегам