Golang と FFmpeg を使用してビデオのちらつきを除去する実践-Golang-php.cn

Golang と FFmpeg を使用してビデオのちらつきを除去する実践

WBOY

リリース： 2023-09-27 16:46:46

オリジナル

1113 人が閲覧しました

Golang と FFmpeg を使用してビデオのちらつきを除去する実践

Golang と FFmpeg を使用してビデオのちらつきを解消する方法

概要:
ビデオのちらつきの問題は、ビデオ処理でよく遭遇する課題です。プロセス。録画した映像のフレームレートと点灯周波数が一致していない場合、映像がちらつく場合があります。この記事では、Golang ライブラリと FFmpeg ライブラリを使用してビデオのちらつきを除去する方法を紹介し、具体的なコード例を示します。

手順:

FFmpeg ライブラリをインストールします:
まず、Golang 開発環境に FFmpeg ライブラリをインストールする必要があります。次のコマンドでインストールできます:
go get github.com/giorgisio/goav/avcodec
```
    github.com/giorgisio/goav/avfilter 
    github.com/giorgisio/goav/avutil 
    github.com/giorgisio/goav/swscale
```
ログイン後にコピー

ビデオファイルを開きます:
Use avformat.OpenInput( in FFmpeg ライブラリ ) 関数は、処理する必要があるビデオファイルを開きます。ビデオファイルのパスをパラメータとして渡して、ビデオファイルの関連情報を取得します。

サンプルコードは次のとおりです。

package main

import (
    "fmt"
    "github.com/giorgisio/goav/avformat"
)

func main() {
    filepath := "path_to_video_file.mp4"

    avformat.AvRegisterAll()

    // 打开视频文件
    ctx := avformat.AvformatAllocContext()
    if err := avformat.AvformatOpenInput(&ctx, filepath, nil, nil); err != 0 {
        fmt.Printf("无法打开文件 %s: %s
", filepath, avutil.AvStrerror(err))
    }
    defer avformat.AvformatCloseInput(&ctx)
    
    // 获取视频文件信息
    if err := avformat.AvformatFindStreamInfo(ctx, nil); err < 0 {
        fmt.Printf("无法获取文件信息: %s
", avutil.AvStrerror(err))
    }
}

ログイン後にコピー

ビデオフレームの処理:
FFmpeg ライブラリの avcodec.AvcodecDecodeVideo2() 関数を使用して、ビデオフレームをデコードします。ビデオフレームをループして各フレームを処理します。処理中に、Golang の画像処理ライブラリ (GoCV など) を使用して、明るさの低減、コントラストの増加などの画像処理操作を実行できます。

サンプルコードは次のとおりです:

package main

import (
    "fmt"
    "github.com/giorgisio/goav/avcodec"
    "github.com/giorgisio/goav/avformat"
    "github.com/giorgisio/goav/avutil"
    "github.com/giorgisio/goav/swscale"
    "gocv.io/x/gocv"
)

func main() {
    filepath := "path_to_video_file.mp4"

    avformat.AvRegisterAll()

    // 打开视频文件
    ctx := avformat.AvformatAllocContext()
    if err := avformat.AvformatOpenInput(&ctx, filepath, nil, nil); err != 0 {
        fmt.Printf("无法打开文件 %s: %s
", filepath, avutil.AvStrerror(err))
    }
    defer avformat.AvformatCloseInput(&ctx)

    // 获取视频文件信息
    if err := avformat.AvformatFindStreamInfo(ctx, nil); err < 0 {
        fmt.Printf("无法获取文件信息: %s
", avutil.AvStrerror(err))
    }

    // 查找视频流索引
    streamIndex := avutil.AvFindBestStream(ctx, avutil.AvmediaType(avformat.AvmTypeVideo), -1, -1, nil, 0)

    codecParams := ctx.Streams()[streamIndex].CodecParameters()

    // 获取解码器
    codec := avcodec.AvcodecFindDecoder(codecParams.CodecId())
    if codec == nil {
        fmt.Println("无法获取解码器")
    }

    // 打开解码器
    codecCtx := avcodec.AvcodecAllocContext3(codec)
    if err := avcodec.AvcodecParametersToContext(codecCtx, codecParams); err < 0 {
        fmt.Printf("无法打开解码器: %s
", avutil.AvStrerror(err))
    }
    defer avcodec.AvcodecFreeContext(&codecCtx)

    if err := avcodec.AvcodecOpen2(codecCtx, codec, nil); err < 0 {
        fmt.Printf("无法打开解码器: %s
", avutil.AvStrerror(err))
    }

    // 初始化帧
    frame := avutil.AvFrameAlloc()
    defer avutil.AvFrameFree(&frame)

    // 初始化解码器上下文
    packet := avcodec.AvPacketAlloc()
    defer avcodec.AvPacketFree(&packet)

    swsCtx := swscale.SwsGetContext(codecParams.Width(), codecParams.Height(), codecCtx.PixFmt(),
        codecParams.Width(), codecParams.Height(), avutil.AV_PIX_FMT_BGR24,
        swscale.SWS_BICUBIC, nil, nil, nil)
    defer swscale.SwsFreeContext(&swsCtx)

    for {
        // 读取帧
        if err := avformat.AvReadFrame(ctx, packet); err != 0 {
            fmt.Printf("无法读取帧: %s
", avutil.AvStrerror(err))
            break
        }

        if packet.StreamIndex() == streamIndex {
            if err := avcodec.AvcodecSendPacket(codecCtx, packet); err < 0 {
                fmt.Printf("无法发送数据包到解码器: %s
", avutil.AvStrerror(err))
            }

            if err := avcodec.AvcodecReceiveFrame(codecCtx, frame); err < 0 {
                fmt.Printf("无法接收解码帧: %s
", avutil.AvStrerror(err))
            }

            // 进行图像处理操作
            img := gocv.NewMatFromBytes(codecParams.Width(), codecParams.Height(), gocv.MatType(gocv.MatTypeCV8UC3), frame.Data(0))
            imgDst := gocv.NewMat()

            // 图像处理操作，以减少亮度为例
            gocv.ConvertScaleAbs(img, &imgDst, 0.5, 0)

            // 输出图像
            fmt.Printf("输出图像: %v
", imgDst)

            img.Close()
            imgDst.Close()
        }

        avcodec.AvPacketUnref(packet)
    }
}

ログイン後にコピー

処理されたビデオを書き込みます:
FFmpeg ライブラリの avcodec.AvcodecEncodeVideo2() 関数を使用して、処理されたビデオをエンコードしますvideo Frame を作成し、 avformat.AvWriteFrame() 関数を使用して、エンコードされたフレームをターゲットビデオファイルに書き込みます。

サンプルコードは次のとおりです:

package main

import (
    "fmt"
    "github.com/giorgisio/goav/avcodec"
    "github.com/giorgisio/goav/avformat"
    "github.com/giorgisio/goav/avutil"
    "github.com/giorgisio/goav/swscale"
    "gocv.io/x/gocv"
)

func main() {
    filepath := "path_to_video_file.mp4"
    outputpath := "path_to_output_file.mp4"

    avformat.AvRegisterAll()

    // 打开视频文件
    ctx := avformat.AvformatAllocContext()
    if err := avformat.AvformatOpenInput(&ctx, filepath, nil, nil); err != 0 {
        fmt.Printf("无法打开文件 %s: %s
", filepath, avutil.AvStrerror(err))
    }
    defer avformat.AvformatCloseInput(&ctx)

    // 获取视频文件信息
    if err := avformat.AvformatFindStreamInfo(ctx, nil); err < 0 {
        fmt.Printf("无法获取文件信息: %s
", avutil.AvStrerror(err))
    }

    // 查找视频流索引
    streamIndex := avutil.AvFindBestStream(ctx, avutil.AvmediaType(avformat.AvmTypeVideo), -1, -1, nil, 0)

    codecParams := ctx.Streams()[streamIndex].CodecParameters()

    // 获取解码器
    codec := avcodec.AvcodecFindDecoder(codecParams.CodecId())
    if codec == nil {
        fmt.Println("无法获取解码器")
    }

    // 打开解码器
    codecCtx := avcodec.AvcodecAllocContext3(codec)
    if err := avcodec.AvcodecParametersToContext(codecCtx, codecParams); err < 0 {
        fmt.Printf("无法打开解码器: %s
", avutil.AvStrerror(err))
    }
    defer avcodec.AvcodecFreeContext(&codecCtx)

    if err := avcodec.AvcodecOpen2(codecCtx, codec, nil); err < 0 {
        fmt.Printf("无法打开解码器: %s
", avutil.AvStrerror(err))
    }

    // 初始化帧
    frame := avutil.AvFrameAlloc()
    defer avutil.AvFrameFree(&frame)

    // 初始化解码器上下文
    packet := avcodec.AvPacketAlloc()
    defer avcodec.AvPacketFree(&packet)

    swsCtx := swscale.SwsGetContext(codecParams.Width(), codecParams.Height(), codecCtx.PixFmt(),
        codecParams.Width(), codecParams.Height(), avutil.AV_PIX_FMT_BGR24,
        swscale.SWS_BICUBIC, nil, nil, nil)
    defer swscale.SwsFreeContext(&swsCtx)

    // 创建输出格式上下文
    fmtCtx := avformat.AvformatAllocContext()
    defer avformat.AvformatFreeContext(fmtCtx)

    // 设置输出文件的格式
    fmtCtx.SetOutputFormat(avformat.AvGuessFormat("", outputpath, ""))

    // 创建输出文件
    if avformat.AvioOpen(&fmtCtx.Pb, outputpath, avformat.AVIO_FLAG_WRITE) < 0 {
        fmt.Println("无法打开输出文件")
    }

    // 写入文件头部
    if avformat.AvformatWriteHeader(fmtCtx, nil) < 0 {
        fmt.Println("无法写入文件头部")
    }

    for {
        // 读取帧
        if err := avformat.AvReadFrame(ctx, packet); err != 0 {
            fmt.Printf("无法读取帧: %s
", avutil.AvStrerror(err))
            break
        }

        if packet.StreamIndex() == streamIndex {
            if err := avcodec.AvcodecSendPacket(codecCtx, packet); err < 0 {
                fmt.Printf("无法发送数据包到解码器: %s
", avutil.AvStrerror(err))
            }

            if err := avcodec.AvcodecReceiveFrame(codecCtx, frame); err < 0 {
                fmt.Printf("无法接收解码帧: %s
", avutil.AvStrerror(err))
            }

            // 进行图像处理操作
            img := gocv.NewMatFromBytes(codecParams.Width(), codecParams.Height(), gocv.MatType(gocv.MatTypeCV8UC3), frame.Data(0))
            imgDst := gocv.NewMat()

            // 图像处理操作，以减少亮度为例
            gocv.ConvertScaleAbs(img, &imgDst, 0.5, 0)

            // 将处理后的图像数据转换为原始数据
            dstData := imgDst.ToBytes()

            // 创建输出帧
            outputFrame := avutil.AvFrameAlloc()
            defer avutil.AvFrameFree(&outputFrame)

            outputFrame.SetData(dstData)

            // 编码输出帧
            if err := avcodec.AvcodecSendFrame(codecCtx, outputFrame); err < 0 {
                fmt.Printf("无法发送输出帧到编码器: %s
", avutil.AvStrerror(err))
            }

            for err := avcodec.AvcodecReceivePacket(codecCtx, packet); err >= 0; err = avcodec.AvcodecReceivePacket(codecCtx, packet) {
                packet.SetStreamIndex(0)
                packet.RescaleTs(codecCtx.TimeBase(), ctx.Streams()[streamIndex].TimeBase())

                if err := avformat.AvWriteFrame(fmtCtx, packet); err < 0 {
                    fmt.Printf("无法写入帧: %s
", avutil.AvStrerror(err))
                }
                avcodec.AvPacketUnref(packet)
            }

            img.Close()
            imgDst.Close()
        }

        avcodec.AvPacketUnref(packet)
    }

    // 写入文件尾部
    avformat.AvWriteTrailer(fmtCtx)
}

ログイン後にコピー

概要:
この記事では、Golang ライブラリと FFmpeg ライブラリを使用してビデオのちらつきを除去する方法を紹介し、以下を提供します。詳細なコード例。 FFmpeg ライブラリの関数を使用すると、ビデオファイルを開いてビデオフレームを処理し、処理されたフレームを再エンコードしてターゲットビデオファイルに書き込むことができます。実際には、ビデオのちらつきの問題を解決するために、特定のニーズに応じて画像処理操作を実行できます。 Golang と FFmpeg の機能を活用することで、ビデオのちらつきの問題をより柔軟かつ効率的に処理できます。

以上がGolang と FFmpeg を使用してビデオのちらつきを除去する実践の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。