Golang集成FFmpeg的音視頻處理的實現(xiàn)

更新時間：2025年02月17日 10:55:18 作者：Yy_Yyyyy_zz

FFmpeg是一個開源的音視頻處理工具,廣泛用于視頻轉換、截圖、處理和流媒體推送等操作,本文主要介紹了Golang集成FFmpeg的音視頻處理的實現(xiàn),具有一定的參考價值,感興趣的可以了解一下

概述

FFmpeg是一個開源的音視頻處理工具，廣泛用于視頻轉換、截圖、處理和流媒體推送等操作。在Go語言項目中，F(xiàn)Fmpeg可以通過調用系統(tǒng)命令來實現(xiàn)相關操作。本文將介紹如何在Go語言中集成并使用FFmpeg，特別是在視頻處理中截取關鍵幀并生成圖片的應用。

使用步驟

安裝FFmpeg首先確保FFmpeg已安裝并配置在系統(tǒng)的環(huán)境變量中?？梢酝ㄟ^運行以下命令來檢查FFmpeg是否已安裝：
```
ffmpeg -version
```
如果沒有安裝，可以根據(jù)操作系統(tǒng)的不同，選擇相應的安裝方式。
安裝Go語言依賴庫在Go語言中，F(xiàn)Fmpeg的調用通常通過執(zhí)行exec.Command來實現(xiàn)，因此只需要使用Go標準庫os/exec即可執(zhí)行FFmpeg命令。無須額外安裝第三方庫。
調用FFmpeg命令使用exec.Command來執(zhí)行FFmpeg命令，并捕獲命令執(zhí)行結果。

FFmpeg常用指令

以下是一些FFmpeg的常用命令行參數(shù)，可以幫助你在Go項目中更好地利用FFmpeg：

視頻轉碼轉換視頻格式（例如：將MP4轉換為AVI）：
```
ffmpeg -i input.mp4 output.avi
```

截取視頻截圖截取視頻的關鍵幀：

ffmpeg -i input.mp4 -vf "select='eq(pict_type,I)'" -vsync vfr output_%03d.jpg

設置截圖的間隔時間每隔一定時間截取一次圖片：
```
ffmpeg -i input.mp4 -vf "fps=1/5" output_%03d.jpg
```
提取視頻中的音頻提取視頻中的音頻并保存為MP3格式：
```
ffmpeg -i input.mp4 -vn output.mp3
```

關鍵幀截取邏輯代碼實現(xiàn)

接下來，我們將詳細解釋一個基于Go語言和FFmpeg的示例代碼，這個示例代碼會從視頻中每隔一定時間截取關鍵幀，并返回相應的圖片。

代碼實現(xiàn)

以下是實現(xiàn)“根據(jù)指定時間間隔截取視頻關鍵幀”的Go語言代碼。

// GetMp4ImagesIntervalSeconds
// @Description: 指定時長截取圖片
// @param videoBase64 base64視頻
// @param intervalSeconds 截圖間隔秒數(shù)
// @return imageSlice 截圖切片
func GetMp4ImagesIntervalSeconds(videoBase64 string, intervalSeconds int) (imageSlice []ImageInfo, err error) {
    // 解碼base64字符串
    videoData, err := base64.StdEncoding.DecodeString(videoBase64)
    if err != nil {
        Log.LogError("GetMp4ImagesIntervalSeconds DecodeString err：%v", err.Error())
        return nil, err
    }

    //判斷目錄是否存在
    if _, err = os.Stat("./temp"); os.IsNotExist(err) {
        err = os.Mkdir("./temp", 0755)
        if err != nil {
            Log.LogError("GetMp4ImagesIntervalSeconds os.Mkdir err：%v", err.Error())
            return nil, err
        }
    }

    // 創(chuàng)建臨時視頻文件
    tmpVideoFile, err := os.CreateTemp("./temp", "tempvideo_*.mp4")
    if err != nil {
        Log.LogError("GetMp4ImagesIntervalSeconds os.CreateTemp err：%v", err.Error())
        return nil, err
    }
    defer os.Remove(tmpVideoFile.Name())
    defer tmpVideoFile.Close()

    // 將視頻數(shù)據(jù)寫入臨時文件
    _, err = tmpVideoFile.Write(videoData)
    if err != nil {
        Log.LogError("GetMp4ImagesIntervalSeconds tmpVideoFile.Write err：%v", err.Error())
        return nil, err
    }

    // 創(chuàng)建臨時圖片目錄
    tempImageDir, err := os.MkdirTemp("./temp", "tempimage_")
    if err != nil {
        Log.LogError("GetMp4ImagesIntervalSeconds os.MkdirTemp err：%v", err.Error())
        return nil, err
    }
    defer os.RemoveAll(tempImageDir)

    // 使用ffmpeg截取關鍵幀并將圖片放入指定目錄
    /*
        -i tmpVideoFile.Name(): 指定輸入文件為臨時視頻文件的名稱。
        -vf select='eq(pict_type,I)',fps=1/intervalSeconds: 使用select濾鏡選擇關鍵幀并指定幀率為1/intervalSeconds，即每間隔intervalSeconds秒取一幀。
        filepath.Join(tempImageDir, "temp_%03d.jpg"): 指定輸出的圖片文件名的格式，temp_%03d.jpg表示輸出文件名的格式為temp_001.jpg、temp_002.jpg等，%03d將會被替換成遞增的數(shù)字序列。這里的 tempImageDir 是保存圖片的臨時目錄。
    */
    cmd := exec.Command("ffmpeg", "-i", tmpVideoFile.Name(), "-vf", fmt.Sprintf("select='eq(pict_type,I)',fps=1/%d", intervalSeconds), filepath.Join(tempImageDir, "temp_%03d.jpg"))
    err = cmd.Run()
    if err != nil {
        Log.LogError("GetMp4ImagesIntervalSeconds exec.Command err：%v", err.Error())
        return nil, err
    }

    // 遍歷臨時圖片目錄，將圖片文件名添加到切片中
    files, _ := filepath.Glob(filepath.Join(tempImageDir, "temp_*.jpg"))
    for _, fileName := range files {
        imageBase64, dhash, err := fileToBase64(fileName)
        if err != nil {
            Log.LogError("GetMp4ImagesIntervalSeconds fileToBase64(fileName) err：%v", err.Error())
            return nil, err
        }
        imageSlice = append(imageSlice, ImageInfo{
            ImageBase64: imageBase64,
            DHash:       dhash,
        })
        fmt.Println("dhash：", dhash)
    }
    Log.LogInfo("GetMp4ImagesIntervalSeconds success number:%v", len(imageSlice))
    return imageSlice, nil
}

代碼說明

視頻數(shù)據(jù)處理通過base64.StdEncoding.DecodeString解碼視頻的Base64數(shù)據(jù)，得到視頻的二進制流，并將其寫入臨時的MP4文件。
創(chuàng)建臨時目錄創(chuàng)建兩個臨時目錄：一個用于存儲視頻文件，另一個用于存儲從視頻中截取的圖片文件。
使用FFmpeg命令行截取視頻關鍵幀通過exec.Command調用FFmpeg命令，參數(shù)-vf select='eq(pict_type,I)',fps=1/{intervalSeconds}表示選擇I幀（關鍵幀）并按指定時間間隔截圖。圖片文件保存到臨時圖片目錄中。
返回截圖結果截取的圖片被保存為JPG格式，程序遍歷圖片文件夾并將每張圖片轉為Base64字符串，并生成ImageInfo結構體返回。