dustoair
3 years ago
commit
d1818b16b2
@ -0,0 +1,3 @@
|
|||||||
|
.idea
|
||||||
|
go.sum
|
||||||
|
*.jpg
|
||||||
@ -0,0 +1,5 @@
|
|||||||
|
module IdSense
|
||||||
|
|
||||||
|
go 1.18
|
||||||
|
|
||||||
|
require github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646
|
||||||
@ -0,0 +1,259 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/nfnt/resize"
|
||||||
|
"image"
|
||||||
|
"image/color"
|
||||||
|
"image/draw"
|
||||||
|
"image/jpeg"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
var Signs = map[string]string{
|
||||||
|
"0": "0011111001111111011100110110000101100001011000010011111100000110",
|
||||||
|
"1": "0010000001100000011000000110000011111110111111100111111000000000",
|
||||||
|
"2": "1000011010001110100011101001101010011010111100101111001001100010",
|
||||||
|
"3": "0100001101000011010100110111000101111011011111110100111000000110",
|
||||||
|
"4": "0000110000111100011111001110110011001110000111100001110000001000",
|
||||||
|
"5": "0001000001110001011100010111000101111011010111110100111000001100",
|
||||||
|
"7": "0100000001000000010001110100111101011100011110000111000001100000",
|
||||||
|
"9": "0111000001111000110110111001111011011110111111001111100001110000",
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// 打开文件
|
||||||
|
// 配合bilibili视频 https://www.bilibili.com/video/BV19L4y1T7sE
|
||||||
|
f, _ := os.Open("id.jpg")
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
// 解析图片
|
||||||
|
img, err := jpeg.Decode(f)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalln(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
img = catImg(img)
|
||||||
|
|
||||||
|
img = bin01(img)
|
||||||
|
|
||||||
|
img = minImg(img)
|
||||||
|
imgs := splitImg(img)
|
||||||
|
showImgs(imgs)
|
||||||
|
id := analyse(imgs)
|
||||||
|
log.Println(id)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// 数字识别
|
||||||
|
func analyse(srcs []image.Image) string {
|
||||||
|
id := ""
|
||||||
|
|
||||||
|
for i := 0; i < len(srcs); i++ {
|
||||||
|
// 获取图片的指纹
|
||||||
|
|
||||||
|
sign := ""
|
||||||
|
for x := 0; x < srcs[i].Bounds().Dx(); x++ {
|
||||||
|
for y := 0; y < srcs[i].Bounds().Dy(); y++ {
|
||||||
|
r, _, _, _ := srcs[i].At(x, y).RGBA()
|
||||||
|
//黑白灰阈值
|
||||||
|
if r > 0x7777 {
|
||||||
|
sign += "1"
|
||||||
|
} else {
|
||||||
|
sign += "0"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 对比指纹
|
||||||
|
mb := ""
|
||||||
|
percent := 0.0
|
||||||
|
for k, v := range Signs {
|
||||||
|
sum := 0
|
||||||
|
for i := 0; i < 64; i++ {
|
||||||
|
if v[i:i+1] == sign[i:i+1] {
|
||||||
|
sum++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if float64(sum)/64 > percent {
|
||||||
|
mb = k
|
||||||
|
percent = float64(sum) / 64
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//log.Println("output-"+strconv.Itoa(i)+".jpg", sign, mb, percent)
|
||||||
|
fmt.Println("output-"+strconv.Itoa(i)+".jpg", sign, mb, percent)
|
||||||
|
id += mb
|
||||||
|
}
|
||||||
|
|
||||||
|
return id
|
||||||
|
}
|
||||||
|
|
||||||
|
// 切割图片
|
||||||
|
func splitImg(src image.Image) []image.Image {
|
||||||
|
var dsts []image.Image
|
||||||
|
|
||||||
|
posx := 0
|
||||||
|
for x := 0; x < src.Bounds().Dx(); x++ {
|
||||||
|
|
||||||
|
found := false
|
||||||
|
for y := 0; y < src.Bounds().Dy(); y++ {
|
||||||
|
r, _, _, _ := src.At(x, y).RGBA()
|
||||||
|
if r == 0xFFFF {
|
||||||
|
found = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if found {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
dst := image.NewGray16(image.Rect(0, 0, x-posx, src.Bounds().Dy()))
|
||||||
|
draw.Draw(dst, dst.Bounds(), src, image.Point{X: posx, Y: 0}, draw.Src)
|
||||||
|
|
||||||
|
// 下一个起点
|
||||||
|
for xx := x + 1; xx < src.Bounds().Dx(); xx++ {
|
||||||
|
found := false
|
||||||
|
for y := 0; y < src.Bounds().Dy(); y++ {
|
||||||
|
r, _, _, _ := src.At(xx, y).RGBA()
|
||||||
|
if r == 0xFFFF {
|
||||||
|
found = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if found {
|
||||||
|
posx = xx
|
||||||
|
x = xx
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//使用第三方库将图片重置为8**
|
||||||
|
img := resize.Resize(8, 8, dst, resize.Lanczos3)
|
||||||
|
dsts = append(dsts, img)
|
||||||
|
}
|
||||||
|
|
||||||
|
return dsts
|
||||||
|
}
|
||||||
|
|
||||||
|
// 去边
|
||||||
|
func minImg(src image.Image) image.Image {
|
||||||
|
// 第一个白点 和 最后一个白点的坐标
|
||||||
|
rect := src.Bounds()
|
||||||
|
|
||||||
|
var (
|
||||||
|
lt, rb image.Point
|
||||||
|
)
|
||||||
|
|
||||||
|
// 左上角x
|
||||||
|
for x := 0; x < rect.Dx(); x++ {
|
||||||
|
for y := 0; y < rect.Dy(); y++ {
|
||||||
|
r, _, _, _ := src.At(x, y).RGBA()
|
||||||
|
if r == 0xFFFF {
|
||||||
|
lt.X = x
|
||||||
|
x = rect.Dx()
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 左上角y
|
||||||
|
for y := 0; y < rect.Dy(); y++ {
|
||||||
|
for x := 0; x < rect.Dx(); x++ {
|
||||||
|
r, _, _, _ := src.At(x, y).RGBA()
|
||||||
|
if r == 0xFFFF {
|
||||||
|
lt.Y = y
|
||||||
|
y = rect.Dy()
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 右下角x
|
||||||
|
for x := rect.Dx() - 1; x > 0; x-- {
|
||||||
|
for y := rect.Dy() - 1; y > 0; y-- {
|
||||||
|
r, _, _, _ := src.At(x, y).RGBA()
|
||||||
|
if r == 0xFFFF {
|
||||||
|
rb.X = x + 1
|
||||||
|
x = 0
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 右下角y
|
||||||
|
for y := rect.Dy() - 1; y > 0; y-- {
|
||||||
|
for x := rect.Dx() - 1; x > 0; x-- {
|
||||||
|
r, _, _, _ := src.At(x, y).RGBA()
|
||||||
|
if r == 0xFFFF {
|
||||||
|
rb.Y = y
|
||||||
|
y = 0
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
newRect := image.Rect(0, 0, rb.X-lt.X+1, rb.Y-lt.Y)
|
||||||
|
log.Println(lt, rb)
|
||||||
|
log.Println(src.Bounds(), newRect)
|
||||||
|
dst := image.NewRGBA(newRect)
|
||||||
|
draw.Draw(dst, dst.Bounds(), src, lt, draw.Over)
|
||||||
|
return dst
|
||||||
|
}
|
||||||
|
|
||||||
|
// 二值化
|
||||||
|
func bin01(src image.Image) image.Image {
|
||||||
|
dst := image.NewGray16(src.Bounds())
|
||||||
|
draw.Draw(dst, dst.Bounds(), src, src.Bounds().Min, draw.Over)
|
||||||
|
|
||||||
|
rect := src.Bounds()
|
||||||
|
for x := 0; x < rect.Dx(); x++ {
|
||||||
|
for y := 0; y < rect.Dy(); y++ {
|
||||||
|
r, _, _, _ := dst.At(x, y).RGBA()
|
||||||
|
if r < 0x5555 {
|
||||||
|
dst.Set(x, y, color.White)
|
||||||
|
} else {
|
||||||
|
dst.Set(x, y, color.Black)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return dst
|
||||||
|
}
|
||||||
|
|
||||||
|
// 预览图片
|
||||||
|
func showImg(src image.Image) {
|
||||||
|
dst, err := os.Create("output.jpg")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalln(err)
|
||||||
|
}
|
||||||
|
jpeg.Encode(dst, src, nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 预览图片组
|
||||||
|
func showImgs(srcs []image.Image) {
|
||||||
|
for i := 0; i < len(srcs); i++ {
|
||||||
|
dst, err := os.Create(fmt.Sprintf("output-%d.jpg", i))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalln(err)
|
||||||
|
}
|
||||||
|
jpeg.Encode(dst, srcs[i], nil)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 号码定位
|
||||||
|
func catImg(src image.Image) image.Image {
|
||||||
|
rect := src.Bounds()
|
||||||
|
// 左上角 w*100/290, h*150/180
|
||||||
|
lt := image.Point{X: rect.Dx() * 100 / 290, Y: rect.Dy() * 150 / 180}
|
||||||
|
// 右下角 w*260/290, h*170/180
|
||||||
|
rb := image.Point{X: rect.Dx() * 260 / 290, Y: rect.Dy() * 165 / 180}
|
||||||
|
newRect := image.Rectangle{
|
||||||
|
Min: image.Point{X: 0, Y: 0},
|
||||||
|
Max: image.Point{X: rb.X - lt.X, Y: rb.Y - lt.Y},
|
||||||
|
}
|
||||||
|
dst := image.NewRGBA(newRect)
|
||||||
|
draw.Draw(dst, dst.Bounds(), src, lt, draw.Over)
|
||||||
|
return dst
|
||||||
|
}
|
||||||
Loading…
Reference in new issue