diff options
author | Nick White <git@njw.name> | 2019-05-13 19:23:03 +0100 |
---|---|---|
committer | Nick White <git@njw.name> | 2019-05-13 19:23:03 +0100 |
commit | 02a6c66eb77a5b455bcf2d0547d2383074eb7e41 (patch) | |
tree | 00f2bde1dfa7a1b23e55478309f26ba1b54fdab9 /binarize/util.go | |
parent | d94fb9f7e74aaad335a587030ed2b4ce44c24cbf (diff) |
Reorganise image manipulation to separate integral image parts
Also unify everything else under preproc/
Note that the UsefulImg interface should be used by the main
functions, to simplify things, but this hasn't been done yet.
Diffstat (limited to 'binarize/util.go')
-rw-r--r-- | binarize/util.go | 87 |
1 files changed, 0 insertions, 87 deletions
diff --git a/binarize/util.go b/binarize/util.go deleted file mode 100644 index ad641c9..0000000 --- a/binarize/util.go +++ /dev/null @@ -1,87 +0,0 @@ -package binarize - -import ( - "errors" - "image" - "math" -) - -func mean(i []int) float64 { - sum := 0 - for _, n := range i { - sum += n - } - return float64(sum) / float64(len(i)) -} - -func stddev(i []int) float64 { - m := mean(i) - - var sum float64 - for _, n := range i { - sum += (float64(n) - m) * (float64(n) - m) - } - variance := sum / float64(len(i) - 1) - return math.Sqrt(variance) -} - -func meanstddev(i []int) (float64, float64) { - m := mean(i) - - var sum float64 - for _, n := range i { - sum += (float64(n) - m) * (float64(n) - m) - } - variance := float64(sum) / float64(len(i) - 1) - return m, math.Sqrt(variance) -} - -// gets the pixel values surrounding a point in the image -func surrounding(img *image.Gray, x int, y int, size int) []int { - b := img.Bounds() - step := size / 2 - - miny := y - step - if miny < b.Min.Y { - miny = b.Min.Y - } - minx := x - step - if minx < b.Min.X { - minx = b.Min.X - } - maxy := y + step - if maxy > b.Max.Y { - maxy = b.Max.Y - } - maxx := x + step - if maxx > b.Max.X { - maxx = b.Max.X - } - - var s []int - for yi := miny; yi <= maxy; yi++ { - for xi := minx; xi <= maxx; xi++ { - s = append(s, int(img.GrayAt(xi, yi).Y)) - } - } - return s -} - -func BinToZeroInv(bin *image.Gray, orig *image.RGBA) (*image.RGBA, error) { - b := bin.Bounds() - if ! b.Eq(orig.Bounds()) { - return orig, errors.New("bin and orig images need to be the same dimensions") - } - newimg := image.NewRGBA(image.Rect(0, 0, b.Dx(), b.Dy())) - for y := b.Min.Y; y < b.Max.Y; y++ { - for x := b.Min.X; x < b.Max.X; x++ { - if bin.GrayAt(x, y).Y == 255 { - newimg.Set(x, y, bin.GrayAt(x, y)) - } else { - newimg.Set(x, y, orig.At(x, y)) - } - } - } - - return newimg, nil -} |