在go语言中使用ffmpeg-go库实现截取视频某一帧图片示例

Go  /  管理员 发布于 1年前   1188

go + ffmpeg-go实现截取视频的某一帧图片作为视频封面的功能。

FFmpeg是一个开源免费跨平台的视频和音频流方案，属于自由软件，采用LGPL或GPL许可证（依据你选择的组件）。

它提供了录制、转换以及流化音视频的完整解决方案。

它包含了非常先进的音频/视频编解码库libavcodec，为了保证高可移植性和编解码质量，libavcodec里很多codec都是从头开发的。

FFmpeg在Linux平台下开发，但它同样也可以在其它操作系统环境中编译运行，包括Windows、Mac OS X等。

FFmpeg项目由以下几部分组成:
FFMpeg 视频文件转换命令行工具,也支持经过实时电视卡抓取和编码成视频文件。
FFServer 基于HTTP(RTSP正在开发中)用于实时广播的多媒体服务器，也支持时间平移.
FFplay 用SDL和FFmpeg库开发的一个简单的媒体播放器.
libavcodec 一个包含了所有FFmpeg音视频编解码器的库.为了保证最优性能和高可复用性,大多数编解码器从头开发的.
libavformat 一个包含了所有的普通音视格式的解析器和产生器的库

进入步骤

1：安装ffmpeg

根据你的系统安装下载ffmpeg安装包，进行安装

https://ffbinaries.com/downloads 

2：安装ffmpeg的go绑定库

推荐使用这个项目

github.com/u2takey/ffmpeg-go 

它是一个绑定库，可以在go语言端使用ffmpeg命令行工具。

3：示例代码

package main

import (
    "bytes"
    "fmt"
    "github.com/disintegration/imaging"
    ffmpeg "github.com/u2takey/ffmpeg-go"
    "log"
    "os"
    "strings"
)

func GetSnapshot(videoPath, snapshotPath string, frameNum int) (snapshotName string, err error) {
    buf := bytes.NewBuffer(nil)
    err = ffmpeg.Input(videoPath).
        Filter("select", ffmpeg.Args{fmt.Sprintf("gte(n,%d)", frameNum)}).
        Output("pipe:", ffmpeg.KwArgs{"vframes": 1, "format": "image2", "vcodec": "mjpeg"}).
        WithOutput(buf, os.Stdout).
        Run()
    
    if err != nil {
        log.Fatal("生成缩略图失败：", err)
        return "", err
    }
    
    img, err := imaging.Decode(buf)
    if err != nil {
        log.Fatal("生成缩略图失败：", err)
        return "", err
    }
    
    err = imaging.Save(img, snapshotPath+".png")
    if err != nil {
        log.Fatal("生成缩略图失败：", err)
        return "", err
    }
    
    names := strings.Split(snapshotPath, "\\")
    snapshotName = names[len(names)-1] + ".png"
    return
}

func main() {
    _, err := GetSnapshot("./test.mp4", "test", 1)
    if err != nil {
        return
    }
}

函数里ffmpeg相关的代码主要是提取视频的某一帧以mjpeg编码为图片，然后保存到本地。

相关资料

https://ffmpeg.org/ffmpeg-filters.html