nn を深くする

Author: ieee0824

cmd/cmyk-train/train.go

@@ -105,8 +105,16 @@ func main() {
 	cmykDir := flag.String("cmyk", "", "cmyk img dir")
 	output := flag.String("f", "network.json", "network dump")
 	flag.Parse()
-	ff := &nn.FeedForward{}
-	ff.Init(4, 5, 3)
+
+	ff, err := nn.Load(*output)
+	if err != nil {
+		fmt.Println("new nn")
+		ff = &nn.FeedForward{}
+		ff.Init(4, 20, 3)
+	} else {
+		fmt.Println("update nn")
+	}
+
 
 	rgbFiles, err := getImgPaths(*rgbDir)
 	if err != nil {

cmd/convert/convert.go

@@ -44,7 +44,10 @@ func main() {
 	}
 
 	newImg := image.NewRGBA(img.Bounds())
-	converter := libcmyk.New(*networkFile)
+	converter, err := libcmyk.New(*networkFile)
+	if err != nil {
+		panic(err)
+	}
 
 	for y := 0; y < h; y++ {
 		for x := 0; x < w; x++ {