Cara menggunakan Golang untuk melakukan pemulihan warna dan pemprosesan imbangan putih pada gambar

Cara menggunakan Golang untuk melakukan pemulihan warna dan pemprosesan imbangan putih pada gambar

Abstrak:
Artikel ini akan memperkenalkan cara menggunakan bahasa pengaturcaraan Golang untuk melakukan pemulihan warna dan pemprosesan imbangan putih pada gambar. Dengan menggunakan perpustakaan pemprosesan imej Golang, kami boleh melakukan pelbagai operasi pada gambar dengan mudah, termasuk pemulihan warna dan imbangan putih. Artikel ini membimbing anda langkah demi langkah melalui proses dan menyediakan contoh kod yang sepadan.

Pengenalan:
Dalam tugas pemprosesan imej digital, pemulihan warna dan imbangan putih adalah dua operasi yang sangat biasa. Matlamat pemulihan warna adalah untuk melaraskan warna dan kontras imej supaya ia kelihatan lebih realistik dan semula jadi. Imbangan putih ialah pelarasan suhu warna dalam imej untuk menjadikannya kelihatan lebih seimbang dan semula jadi. Kedua-dua operasi ini sangat penting untuk meningkatkan kesan visual imej, dan kami boleh melaksanakannya menggunakan Golang.

Langkah 1: Import perpustakaan dan alatan yang diperlukan
Pertama, kita perlu mengimport pustaka pemprosesan imej dan pustaka operasi fail Golang.

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

Langkah 2: Muatkan Fail Imej
Seterusnya, kita perlu memuatkan fail imej yang ingin kita proses. Kita boleh mencapai ini menggunakan perpustakaan manipulasi fail Golang.

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
}

Langkah 3: Pembetulan Warna
Sekarang kita telah memuatkan fail imej, kita boleh mula menetapkan warna. Pemulihan warna boleh dicapai dengan melaraskan rona, ketepuan dan kecerahan imej. Berikut ialah contoh mudah yang hanya meningkatkan nilai saluran merah bagi setiap piksel imej sebanyak 100.

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
}

Langkah 4: Lakukan pemprosesan imbangan putih
Seterusnya kita boleh melakukan pemprosesan imbangan putih untuk memastikan suhu warna yang seimbang dalam imej. Di bawah ialah contoh mudah yang hanya menukar nilai saluran merah dan biru untuk setiap piksel imej.

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
}

Langkah 5: Simpan imej yang diproses
Akhir sekali, kita perlu menyimpan imej yang diproses ke dalam fail.

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
}

Mengintegrasikan langkah di atas:

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

Kesimpulan:
Dalam artikel ini, kami memperkenalkan cara untuk mewarnakan imej tetapan dan imbangan putih menggunakan bahasa pengaturcaraan Golang. Dengan menggunakan perpustakaan pemprosesan imej Golang, anda boleh melaksanakan dua operasi pemprosesan imej biasa ini dengan mudah. Sama ada digunakan untuk projek peribadi atau aplikasi komersial, operasi ini boleh membantu anda meningkatkan kesan visual imej anda. Semoga artikel ini dapat membantu anda memahami cara melakukan operasi ini di Golang dan memberikan bantuan untuk tugas pemprosesan imej anda.

Atas ialah kandungan terperinci Cara menggunakan Golang untuk melakukan pemulihan warna dan pemprosesan imbangan putih pada gambar. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!