Rumah > Artikel > pembangunan bahagian belakang > Amalan menggunakan Golang dan FFmpeg untuk mencapai penyahkelipan video

Amalan menggunakan Golang dan FFmpeg untuk mencapai penyahkelipan video

WBOYasal: 2023-09-27 16:46:461083semak imbas

Ikhtisar:
Masalah kelipan video adalah cabaran yang sering dihadapi dalam proses pemprosesan video. Apabila kadar bingkai video yang dirakam tidak sepadan dengan kekerapan pencahayaan, ia boleh menyebabkan kelipan dalam video. Artikel ini akan memperkenalkan cara menggunakan perpustakaan Golang dan FFmpeg untuk mencapai penyahkelipan video dan memberikan contoh kod khusus.

Langkah:

Pasang perpustakaan FFmpeg:
Pertama, kita perlu memasang perpustakaan FFmpeg dalam persekitaran pembangunan Golang. Ia boleh dipasang melalui arahan berikut:
pergi dapatkan github.com/giorgisio/goav/avcodec
```
    github.com/giorgisio/goav/avfilter 
    github.com/giorgisio/goav/avutil 
    github.com/giorgisio/goav/swscale
```

Buka fail video:
Gunakan fungsi avformat.OpenInput() dalam perpustakaan FFmpeg untuk membuka fail video yang perlu diproses. Dapatkan maklumat tentang fail video dengan menghantar laluan fail video sebagai parameter.

Kod sampel adalah seperti berikut:

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))
    }
}

Memproses bingkai video:
Gunakan fungsi avcodec.AvcodecDecodeVideo2() dalam perpustakaan FFmpeg untuk menyahkod bingkai video. Proses setiap bingkai dengan menggelung melalui bingkai video. Semasa pemprosesan, anda boleh menggunakan pustaka pemprosesan imej Golang (seperti GoCV) untuk melaksanakan operasi pemprosesan imej, seperti mengurangkan kecerahan, meningkatkan kontras, dsb.

Kod contoh adalah seperti berikut:

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)
    }
}

Tulis video yang diproses:
Gunakan fungsi avcodec.AvcodecEncodeVideo2() dalam perpustakaan FFmpeg untuk mengekod bingkai video yang diproses, dan kemudian gunakan fungsi avcodec.AvcodecEncodeVideo2() dalam perpustakaan FFmpeg untuk mengekod bingkai video yang diproses, dan kemudian gunakan fungsi avformat.AvWrite toFrame() mengekod bingkai yang dikodkan Tulis ke fail video sasaran.

Kod sampel adalah seperti berikut:

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)
}

Ringkasan:
Artikel ini memperkenalkan cara menggunakan perpustakaan Golang dan FFmpeg untuk melaksanakan penyahklik video dan menyediakan contoh kod terperinci. Dengan menggunakan fungsi dalam pustaka FFmpeg, kami boleh membuka fail video, memproses bingkai video dan mengekod semula bingkai yang diproses dan menulisnya ke fail video sasaran. Dalam amalan, operasi pemprosesan imej boleh dilakukan mengikut keperluan khusus untuk menyelesaikan masalah kelipan video. Dengan memanfaatkan kuasa Golang dan FFmpeg, kami boleh menangani masalah kelipan video dengan lebih fleksibel dan cekap.

Atas ialah kandungan terperinci Amalan menggunakan Golang dan FFmpeg untuk mencapai penyahkelipan video. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

golang 循环 github ffmpeg

Kenyataan：

Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn

Artikel sebelumnya：Corak Fasad Golang dan pengalaman praktikal dalam reka bentuk berorientasikan objekArtikel seterusnya：Corak Fasad Golang dan pengalaman praktikal dalam reka bentuk berorientasikan objek

Artikel berkaitan

Lihat lagi