diff options
| -rw-r--r-- | binarize/main.go | 35 |
1 files changed, 25 insertions, 10 deletions
diff --git a/binarize/main.go b/binarize/main.go index 610effc..aa8d4ee 100644 --- a/binarize/main.go +++ b/binarize/main.go @@ -6,10 +6,12 @@ package main import ( "flag" "fmt" + "image" + "image/draw" + _ "image/jpeg" + "image/png" "log" "os" - - "github.com/Ernyoke/Imger/imgio" // TODO: get rid of this and do things myself ) func main() { @@ -29,16 +31,29 @@ func main() { *wsize++ } - img, err := imgio.ImreadGray(flag.Arg(0)) + f, err := os.Open(flag.Arg(0)) + defer f.Close() if err != nil { - log.Fatalf("Could not read image %s\n", flag.Arg(0)) + log.Fatalf("Could not open file %s: %v\n", flag.Arg(0), err) } - - // TODO: estimate an appropriate window size based on resolution - thresh := IntegralSauvola(img, *ksize, *wsize) - - err = imgio.Imwrite(thresh, flag.Arg(1)) + img, _, err := image.Decode(f) if err != nil { - log.Fatalf("Could not write image %s\n", flag.Arg(1)) + log.Fatalf("Could not decode image: %v\n", err) } + b := img.Bounds() + gray := image.NewGray(image.Rect(0, 0, b.Dx(), b.Dy())) + draw.Draw(gray, b, img, b.Min, draw.Src) + + // TODO: estimate an appropriate window size based on resolution + thresh := IntegralSauvola(gray, *ksize, *wsize) + + f, err = os.Create(flag.Arg(1)) + if err != nil { + log.Fatalf("Could not create file %s: %v\n", flag.Arg(1), err) + } + defer f.Close() + err = png.Encode(f, thresh) + if err != nil { + log.Fatalf("Could not encode image: %v\n", err) + } } |