mirror of
https://github.com/esimov/caire.git
synced 2024-05-31 14:55:55 +08:00
35 lines
652 B
Go
35 lines
652 B
Go
package caire
|
|
|
|
import (
|
|
"image"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func Benchmark_Carver(b *testing.B) {
|
|
sampleImg := filepath.Join("./testdata", "sample.jpg")
|
|
f, err := os.Open(sampleImg)
|
|
if err != nil {
|
|
b.Fatalf("could not load sample image: %v", err)
|
|
}
|
|
defer f.Close()
|
|
|
|
src, _, err := image.Decode(f)
|
|
if err != nil {
|
|
b.Fatalf("error decoding image: %v", err)
|
|
}
|
|
b.ResetTimer()
|
|
|
|
img := p.imgToNRGBA(src)
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
width, height := img.Bounds().Max.X, img.Bounds().Max.Y
|
|
c := NewCarver(width, height)
|
|
c.ComputeSeams(p, img)
|
|
seams := c.FindLowestEnergySeams(p)
|
|
img = c.RemoveSeam(img, seams, p.Debug)
|
|
}
|
|
|
|
}
|