Golang实现图片的背景替换和通道分离

package main

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

func main() {
    // 读取原始图像
    file1, err := os.Open("image1.jpg")
    if err != nil {
        fmt.Println(err)
        return
    }
    defer file1.Close()

    img1, _, err := image.Decode(file1)
    if err != nil {
        fmt.Println(err)
        return
    }

    // 读取背景图像
    file2, err := os.Open("image2.jpg")
    if err != nil {
        fmt.Println(err)
        return
    }
    defer file2.Close()

    img2, _, err := image.Decode(file2)
    if err != nil {
        fmt.Println(err)
        return
    }

    // 创建新图像
    bounds := img1.Bounds()
    newImg := image.NewRGBA(bounds)

    // 遍历像素，进行背景替换
    for x := bounds.Min.X; x < bounds.Max.X; x++ {
        for y := bounds.Min.Y; y < bounds.Max.Y; y++ {
            r1, g1, b1, _ := img1.At(x, y).RGBA()
            r2, g2, b2, _ := img2.At(x, y).RGBA()

            // 判断是否为背景像素，并替换
            if r1 == 0 && g1 == 0 && b1 == 0 {
                newImg.Set(x, y, color.RGBA{uint8(r2), uint8(g2), uint8(b2), 255})
            } else {
                newImg.Set(x, y, color.RGBA{uint8(r1), uint8(g1), uint8(b1), 255})
            }
        }
    }

    // 保存结果图像
    resultFile, err := os.Create("result.jpg")
    if err != nil {
        fmt.Println(err)
        return
    }
    defer resultFile.Close()

    jpeg.Encode(resultFile, newImg, &jpeg.Options{100})
}

package main

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

func main() {
    // 读取原始图像
    file, err := os.Open("image.jpg")
    if err != nil {
        fmt.Println(err)
        return
    }
    defer file.Close()

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

    // 创建新图像
    bounds := img.Bounds()
    rImg := image.NewRGBA(bounds)
    gImg := image.NewRGBA(bounds)
    bImg := image.NewRGBA(bounds)

    // 遍历像素，进行通道分离
    for x := bounds.Min.X; x < bounds.Max.X; x++ {
        for y := bounds.Min.Y; y < bounds.Max.Y; y++ {
            r, g, b, _ := img.At(x, y).RGBA()

            // 向对应通道中设置像素值
            rImg.Set(x, y, color.RGBA{uint8(r >> 8), 0, 0, 255})
            gImg.Set(x, y, color.RGBA{0, uint8(g >> 8), 0, 255})
            bImg.Set(x, y, color.RGBA{0, 0, uint8(b >> 8), 255})
        }
    }

    // 保存结果图像
    rFile, err := os.Create("r.jpg")
    if err != nil {
        fmt.Println(err)
        return
    }
    defer rFile.Close()

    gFile, err := os.Create("g.jpg")
    if err != nil {
        fmt.Println(err)
        return
    }
    defer gFile.Close()

    bFile, err := os.Create("b.jpg")
    if err != nil {
        fmt.Println(err)
        return
    }
    defer bFile.Close()

    jpeg.Encode(rFile, rImg, &jpeg.Options{100})
    jpeg.Encode(gFile, gImg, &jpeg.Options{100})
    jpeg.Encode(bFile, bImg, &jpeg.Options{100})
}