Go中RGB到YUV 420P的图像转换

package main

import (
    "image"
    "image/color"
    "image/jpeg"
    "os"
)

func main() {
    // 读取RGB图像
    file, err := os.Open("input.jpg")
    if err != nil {
        panic(err)
    }
    defer file.Close()

    img, _, err := image.Decode(file)
    if err != nil {
        panic(err)
    }

    // 创建YUV 420P图像
    yuvImg := image.NewYCbCr(image.Rect(0, 0, img.Bounds().Dx(), img.Bounds().Dy()), image.YCbCrSubsampleRatio420)

    // 遍历每个像素，进行RGB到YUV的转换
    for y := 0; y < img.Bounds().Dy(); y++ {
        for x := 0; x < img.Bounds().Dx(); x++ {
            r, g, b, _ := img.At(x, y).RGBA()
            yuv := color.RGBToYCbCr(uint8(r>>8), uint8(g>>8), uint8(b>>8))
            yuvImg.YCbCrAt(x, y).Y = yuv.Y
            yuvImg.YCbCrAt(x, y).Cb = yuv.Cb
            yuvImg.YCbCrAt(x, y).Cr = yuv.Cr
        }
    }

    // 保存YUV图像
    yuvFile, err := os.Create("output.yuv")
    if err != nil {
        panic(err)
    }
    defer yuvFile.Close()

    err = jpeg.Encode(yuvFile, yuvImg, nil)
    if err != nil {
        panic(err)
    }
}