updated configs to include valid_metric and weight params

AIWintermuteAI · Apr 21, 2020 · a0fdeb2 · a0fdeb2
1 parent 2bad2d3
commit a0fdeb2
Show file tree

Hide file tree

Showing 6 changed files with 25 additions and 13 deletions.
diff --git a/configs/classifier.json b/configs/classifier.json
@@ -7,15 +7,19 @@
         "labels":               [],
         "dropout" : 		0.5
     },
-    "pretrained" : {
-        "full":   				""
+     "weights" : {
+            "full":   				"",
+            "backend":   		    "imagenet",
+            "save_bottleneck":      false
+
     },
     "train" : {
         "actual_epoch":         1,
         "train_image_folder":   "/home/ubuntu/datasets/classifier/imgs",
         "train_times":          4,
         "valid_image_folder":   "/home/ubuntu/datasets/classifier/imgs_validation",
         "valid_times":          4,
+        "valid_metric":         "val_accuracy",
         "batch_size":           4,
         "learning_rate":        1e-4,
         "saved_folder":   		"classifier",

diff --git a/configs/detector.json b/configs/detector.json
@@ -10,8 +10,9 @@
         "object_scale" : 		5.0,
         "no_object_scale" : 	1.0
     },
-    "pretrained" : {
-        "full":   				""
+    "weights" : {
+        "full":   				"",
+        "backend":              "imagenet"
     },
     "train" : {
         "actual_epoch":         1,
@@ -21,6 +22,7 @@
         "valid_image_folder":   "/home/ubuntu/datasets/detector/imgs_validation",
         "valid_annot_folder":   "/home/ubuntu/datasets/detector/anns_validation",
         "valid_times":          4,
+        "valid_metric":         "mAP",
         "batch_size":           4,
         "learning_rate":        1e-4,
         "saved_folder":   		"detector",

diff --git a/configs/dogs_classifier.json b/configs/dogs_classifier.json
@@ -19,7 +19,7 @@
         "train_times":          1,
         "valid_image_folder":   "/home/ubuntu/datasets/dogs_classification/imgs_validation",
         "valid_times":          1,
-        "valid_metric":         "val_loss",
+        "valid_metric":         "val_accuracy",
         "batch_size":           4,
         "learning_rate":        1e-3,
         "saved_folder":   		"classifier",

diff --git a/configs/pascal_20_detector.json b/configs/pascal_20_detector.json
@@ -10,8 +10,9 @@
         "object_scale" : 		5.0,
         "no_object_scale" : 	1.0
     },
-    "pretrained" : {
-        "full":   				"detector/weights.h5"
+    "weights" : {
+        "full":   				"",
+        "backend":              "imagenet"
     },
     "train" : {
         "actual_epoch":         50,
@@ -21,6 +22,7 @@
         "valid_image_folder":   "/home/ubuntu/datasets/pascal_20_detection/imgs_validation",
         "valid_annot_folder":   "/home/ubuntu/datasets/pascal_20_detection/anns_validation",
         "valid_times":          1,
+        "valid_metric":         "mAP",
         "batch_size":           8,
         "learning_rate":        1e-4,
         "saved_folder":   		"detector",

diff --git a/configs/pascal_20_segnet.json b/configs/pascal_20_segnet.json
@@ -3,10 +3,11 @@
         "type":                 "SegNet",
         "architecture":         "MobileNet7_5",
         "input_size":           224,
-        "n_classes" : 		21
+        "n_classes" : 		20
     },
-    "pretrained" : {
-        "full":   				""
+    "weights" : {
+        "full":   				"",
+        "backend":              "imagenet"
     },
     "train" : {
         "actual_epoch":         50,
@@ -16,6 +17,7 @@
         "valid_image_folder":   "/home/ubuntu/datasets/pascal_20_segmentation/imgs_validation",
         "valid_annot_folder":   "/home/ubuntu/datasets/pascal_20_segmentation/anns_validation",
         "valid_times":          1,
+        "valid_metric":         "val_loss",
         "batch_size":           8,
         "learning_rate":        1e-3,
         "saved_folder":   		"pascal_20",

diff --git a/configs/segmentation.json b/configs/segmentation.json
@@ -3,10 +3,11 @@
         "type":                 "SegNet",
         "architecture":         "MobileNet",
         "input_size":           224,
-        "n_classes" : 		21
+        "n_classes" : 		20
     },
-    "pretrained" : {
-        "full":   				""
+    "weights" : {
+        "full":   				"",
+        "backend":              "imagenet"
     },
     "train" : {
         "actual_epoch":         1,
@@ -16,6 +17,7 @@
         "valid_image_folder":   "/home/ubuntu/datasets/segmentation/imgs_validation",
         "valid_annot_folder":   "/home/ubuntu/datasets/segmentation/anns_validation",
         "valid_times":          4,
+        "valid_metric":         "val_loss",
         "batch_size":           8,
         "learning_rate":        1e-4,
         "saved_folder":   		"segment",