Traitement d'images Golang : apprenez à étirer et à transformer géométriquement des images

Traitement d'images Golang : Apprenez à étirer et transformer géométriquement des images

En tant que langage de programmation puissant, Golang possède également de nombreuses fonctions riches en traitement d'image. Dans cet article, nous apprendrons comment utiliser Golang pour étirer et transformer géométriquement des images.

Tout d'abord, nous devons importer les packages associés tels que image et gonum.org/v1/plot. Vous trouverez ci-dessous un exemple de programme simple pour lire, redimensionner et enregistrer des images. image和gonum.org/v1/plot等相关包。以下是一个简单的示例程序，用于读取、调整和保存图像。

package main

import (
    "image"
    "image/color"
    "image/jpeg"
    "image/png"
    "log"
    "math"
    "os"

    "github.com/gonum/matrix/mat64"
)

// 读取图像文件
func loadImage(filename string) (image.Image, error) {
    file, err := os.Open(filename)
    if err != nil {
        return nil, err
    }
    defer file.Close()
    img, _, err := image.Decode(file)
    return img, err
}

// 保存图像文件
func saveImage(img image.Image, format, filename string) error {
    file, err := os.Create(filename)
    if err != nil {
        return err
    }
    defer file.Close()
    switch format {
    case "jpeg":
        return jpeg.Encode(file, img, &jpeg.Options{Quality: 100})
    case "png":
        return png.Encode(file, img)
    default:
        return nil
    }
}

// 拉伸图像
func stretchImage(img image.Image, width, height int) image.Image {
    newImg := image.NewRGBA(image.Rect(0, 0, width, height))
    for y := 0; y < height; y++ {
        for x := 0; x < width; x++ {
            srcX := int(float64(x) / float64(width) * float64(img.Bounds().Dx()))
            srcY := int(float64(y) / float64(height) * float64(img.Bounds().Dy()))
            newImg.Set(x, y, img.At(srcX, srcY))
        }
    }
    return newImg
}

// 几何变换
func geometricTransformation(img image.Image, theta float64) image.Image {
    width, height := img.Bounds().Dx(), img.Bounds().Dy()
    cosTheta := math.Cos(theta)
    sinTheta := math.Sin(theta)
    M := mat64.NewDense(3, 3, nil)
    M.Set(0, 0, cosTheta)
    M.Set(0, 1, sinTheta)
    M.Set(0, 2, 0)
    M.Set(1, 0, -sinTheta)
    M.Set(1, 1, cosTheta)
    M.Set(1, 2, 0)
    M.Set(2, 0, 0)
    M.Set(2, 1, 0)
    M.Set(2, 2, 1)
    newImg := image.NewRGBA(image.Rect(0, 0, width, height))
    for y := 0; y < height; y++ {
        for x := 0; x < width; x++ {
            v := mat64.NewVector(3, []float64{float64(x), float64(y), 1})
            res := mat64.NewVector(3, nil)
            res.MulVec(M, v)
            srcX := int(res.At(0, 0) / res.At(2, 0))
            srcY := int(res.At(1, 0) / res.At(2, 0))
            if srcX >= 0 && srcX < width && srcY >= 0 && srcY < height {
                newImg.Set(x, y, img.At(srcX, srcY))
            } else {
                newImg.Set(x, y, color.White)
            }
        }
    }
    return newImg
}

func main() {
    // 读取图像
    img, err := loadImage("input.jpg")
    if err != nil {
        log.Fatal(err)
    }

    // 拉伸图像
    stretchedImg := stretchImage(img, 500, 500)
    err = saveImage(stretchedImg, "jpeg", "stretched.jpg")
    if err != nil {
        log.Fatal(err)
    }

    // 几何变换
    geometricImg := geometricTransformation(img, 0.5)
    err = saveImage(geometricImg, "jpeg", "geometric.jpg")
    if err != nil {
        log.Fatal(err)
    }
}

在上面的代码中，我们定义了loadImage函数用于读取图片文件，saveImage函数用于保存图片文件。同时，我们还定义了stretchImage函数和geometricTransformation函数分别用于图像的拉伸和几何变换。

在main函数中，我们首先读取了一张名为input.jpg的图像。接下来，我们调用stretchImage函数将图像拉伸到500x500的大小，并保存为stretched.jpg文件。然后，我们调用geometricTransformation函数进行几何变换，并将结果保存为geometric.jpgrrreee

Dans le code ci-dessus, nous avons défini la fonction loadImage pour lire les fichiers image et la fonction saveImage pour enregistrer les fichiers image. Dans le même temps, nous avons également défini la fonction stretchImage et la fonction geometricTransformation pour l'étirement de l'image et la transformation géométrique respectivement.

Dans la fonction main, on lit d'abord une image nommée input.jpg. Ensuite, nous appelons la fonction stretchImage pour étirer l'image à une taille de 500x500 et l'enregistrer sous un fichier stretched.jpg. Ensuite, nous appelons la fonction geometricTransformation pour effectuer une transformation géométrique et enregistrer le résultat sous forme de fichier geometric.jpg. 🎜🎜Vous pouvez ajuster les paramètres et les fonctions du code en fonction de vos besoins pour obtenir un traitement d'image plus détaillé. J'espère que cet article pourra vous aider à apprendre à utiliser Golang pour étirer et transformer géométriquement des images. 🎜

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!