Golang을 사용하여 사진의 색상 복원 및 화이트 밸런스 처리를 수행하는 방법

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

func loadImage(filepath string) (image.Image, error) {
    imgFile, err := os.Open(filepath)
    if err != nil {
        return nil, err
    }
    defer imgFile.Close()

    img, err := jpeg.Decode(imgFile)
    if err != nil {
        return nil, err
    }

    return img, nil
}

func colorCorrection(img image.Image) image.Image {
    bounds := img.Bounds()
    newImg := image.NewRGBA(bounds)

    for y := bounds.Min.Y; y < bounds.Max.Y; y++ {
        for x := bounds.Min.X; x < bounds.Max.X; x++ {
            oldPixel := img.At(x, y)
            r, g, b, a := oldPixel.RGBA()

            // 色调修复
            r += 100

            newPixel := color.RGBA{
                R: uint8(r),
                G: uint8(g),
                B: uint8(b),
                A: uint8(a),
            }

            newImg.Set(x, y, newPixel)
        }
    }

    return newImg
}

func whiteBalance(img image.Image) image.Image {
    bounds := img.Bounds()
    newImg := image.NewRGBA(bounds)

    for y := bounds.Min.Y; y < bounds.Max.Y; y++ {
        for x := bounds.Min.X; x < bounds.Max.X; x++ {
            oldPixel := img.At(x, y)
            r, g, b, a := oldPixel.RGBA()

            // 白平衡处理
            newPixel := color.RGBA{
                R: uint8(b),
                G: uint8(g),
                B: uint8(r),
                A: uint8(a),
            }

            newImg.Set(x, y, newPixel)
        }
    }

    return newImg
}

func saveImage(filepath string, img image.Image) error {
    outFile, err := os.Create(filepath)
    if err != nil {
        return err
    }
    defer outFile.Close()

    err = jpeg.Encode(outFile, img, nil)
    if err != nil {
        return err
    }

    return nil
}

func main() {
    // 加载图像
    img, err := loadImage("input.jpg")
    if err != nil {
        panic(err)
    }

    // 进行色彩修复
    correctedImg := colorCorrection(img)

    // 进行白平衡处理
    finalImg := whiteBalance(correctedImg)

    // 保存处理后的图像
    err = saveImage("output.jpg", finalImg)
    if err != nil {
        panic(err)
    }
}