44 lines
1.0 KiB
Go
44 lines
1.0 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"github.com/corona10/goimagehash"
|
||
|
|
"image/jpeg"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
file1, _ := os.Open("sample1.jpg")
|
||
|
|
file2, _ := os.Open("sample2.jpg")
|
||
|
|
defer file1.Close()
|
||
|
|
defer file2.Close()
|
||
|
|
var b bytes.Buffer
|
||
|
|
foo := bufio.NewWriter(&b)
|
||
|
|
img1, _ := jpeg.Decode(file1)
|
||
|
|
img2, _ := jpeg.Decode(file2)
|
||
|
|
hash1, _ := goimagehash.AverageHashExtend(img1, 15)
|
||
|
|
hash2, _ := goimagehash.AverageHashExtend(img2, 15)
|
||
|
|
distance, _ := hash1.Distance(hash2)
|
||
|
|
fmt.Printf("Distance between images: %v\n", distance)
|
||
|
|
|
||
|
|
err := hash1.Dump(foo)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Println(err)
|
||
|
|
}
|
||
|
|
foo.Flush()
|
||
|
|
bar := bufio.NewReader(&b)
|
||
|
|
hash3, err := goimagehash.LoadImageHashExtend(bar)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Println(err)
|
||
|
|
}
|
||
|
|
distance, _ = hash1.Distance(hash3)
|
||
|
|
fmt.Printf("Distance between hash1 and hash3: %v\n", distance)
|
||
|
|
distance, _ = hash2.Distance(hash3)
|
||
|
|
fmt.Printf("Distance between hash2 and hash3: %v\n", distance)
|
||
|
|
fmt.Println(hash1.ToString())
|
||
|
|
fmt.Println(hash2.ToString())
|
||
|
|
fmt.Println(hash3.ToString())
|
||
|
|
}
|