Go截取视频某一帧图片 ｜ 青训营笔记

这是我参与「第五届青训营 」伴学笔记创作活动的第 8 天

前言

最近遇到一个需求，需要截取视频的某一帧图片作为视频封面。我搜寻了相关资料，在 go 语言端常见的有两种做法

1. 使用 opencv 的 go 绑定库
2. 使用 ffmpeg 的 go 绑定库

这里我打算使用第二种方法，使用 ffmpeg 的 go 绑定库。

ffmpeg 介绍

FFmpeg是一个开源免费跨平台的视频和音频流方案，属于自由软件，采用LGPL或GPL许可证（依据你选择的组件）。它提供了录制、转换以及流化音视频的完整解决方案。它包含了非常先进的音频/视频编解码库libavcodec，为了保证高可移植性和编解码质量，libavcodec里很多codec都是从头开发的。

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

FFmpeg项目由以下几部分组成:

FFMpeg 视频文件转换命令行工具,也支持经过实时电视卡抓取和编码成视频文件。

FFServer 基于HTTP(RTSP正在开发中)用于实时广播的多媒体服务器，也支持时间平移.

FFplay 用SDL和FFmpeg库开发的一个简单的媒体播放器.

libavcodec 一个包含了所有FFmpeg音视频编解码器的库.为了保证最优性能和高可复用性,大多数编解码器从头开发的.

libavformat 一个包含了所有的普通音视格式的解析器和产生器的库

解决

安装 ffmpeg

浏览器访问 ffbinaries.com/downloads 根据你的系统安装 ffmpeg

安装 ffmpeg 的 go 绑定库

这里我使用了 github.com/u2takey/ffmpeg-go 这个项目，它是一个绑定库，可以在 go 语言端使用 ffmpeg 命令行工具。

编写代码

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编码为图片，然后保存到本地。