python 获取MP4视频第一帧 | Python工具类

目录

前言

环境部署

代码

总结 

---

前言

获取MP4视频第一帧图片，mp4的视频路径支持url链接。

环境部署

需要部署ffmpeg环境，具体的部署步骤可以参考我的另一篇文章：windows ffmpeg安装部署_阿良的博客-CSDN博客

代码

不废话，上代码。

#!/user/bin/env python
# coding=utf-8
"""
@project : csdn
@author  : huyi
@file   : extract_video_first_frame.py
@ide    : PyCharm
@time   : 2021-11-11 21:33:38
"""
import os
import subprocess
import uuid
import urllib.request


def extract(video_path: str, tmp_dir: str):
    if video_path.startswith("http"):
        mp4_path = _download_mp4(video_path, tmp_dir)
    else:
        mp4_path = video_path
    pic_path = os.path.join(tmp_dir, '{}.jpg'.format(uuid.uuid4()))
    ffmpeg_cmd = 'ffmpeg -i {} -f image2 -ss 1 -frames:v 1 {}'.format(
        mp4_path, pic_path)
    print(ffmpeg_cmd)
    ffmpeg_pipe = subprocess.Popen(ffmpeg_cmd, shell=True)
    ffmpeg_pipe.wait()
    return pic_path


def _download_mp4(video_url: str, tmp_dir: str):
    new_video_path = os.path.join(tmp_dir, '{}.mp4'.format(uuid.uuid4()))
    urllib.request.urlretrieve(video_url, new_video_path)
    return new_video_path

验证一下，验证代码如下

if __name__ == '__main__':
    print(
        extract(
            'http://xxxxxxx/test1.mp4',
            'C:/Users/huyi/Desktop'))

总结 

没啥好总结的，java我也做了个工具类，给自己做个记忆点。

分享：

        遇见你之前，不知道什么叫喜欢。错过你之后，不知道什么叫喜欢。——《雪中悍刀行》

如果本文对你有帮助的话，请不要吝啬你的赞，谢谢！