From a0fdeb2bd6b4608879864d1f499720e8888c0ad0 Mon Sep 17 00:00:00 2001
From: AIWintermuteAI <dmitrywat@gmail.com>
Date: Tue, 21 Apr 2020 22:56:35 +0800
Subject: [PATCH] updated configs to include valid_metric and weight params

---
 configs/classifier.json         | 8 ++++++--
 configs/detector.json           | 6 ++++--
 configs/dogs_classifier.json    | 2 +-
 configs/pascal_20_detector.json | 6 ++++--
 configs/pascal_20_segnet.json   | 8 +++++---
 configs/segmentation.json       | 8 +++++---
 6 files changed, 25 insertions(+), 13 deletions(-)

diff --git a/configs/classifier.json b/configs/classifier.json
index 07b93ca..447e6b8 100644
--- a/configs/classifier.json
+++ b/configs/classifier.json
@@ -7,8 +7,11 @@
         "labels":               [],
         "dropout" : 		0.5
     },
-    "pretrained" : {
-        "full":   				""
+     "weights" : {
+            "full":   				"",
+            "backend":   		    "imagenet",
+            "save_bottleneck":      false
+        
     },
     "train" : {
         "actual_epoch":         1,
@@ -16,6 +19,7 @@
         "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
index 025a3e7..04c6541 100644
--- 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
index e80827e..71319ee 100644
--- 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
index ff8be22..d5695a7 100644
--- 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
index f2f438b..2315e85 100644
--- 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
index 46171a8..4e99db1 100644
--- 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",