add more options

Author: LittleLittleCloud

src/Microsoft.ML.AutoML/CodeGen/object_detection_search_space.json

@@ -12,10 +12,15 @@
       "type": "string",
       "default": "PredictedLabel"
     },
+    {
+      "name": "PredictedBoundingBoxColumnName",
+      "type": "string",
+      "default": "PredictedBoundingBoxes"
+    },
     {
       "name": "BoundingBoxColumnName",
       "type": "string",
-      "default": "BoundingBox"
+      "default": "BoundingBoxes"
     },
     {
       "name": "ImageColumnName",
@@ -34,13 +39,23 @@
     },
     {
       "name": "InitLearningRate",
-      "type": "float",
+      "type": "double",
       "default": 1.0
     },
     {
       "name": "WeightDecay",
-      "type": "float",
+      "type": "double",
       "default": 0.0
+    },
+    {
+      "name": "IOUThreshold",
+      "type": "double",
+      "default": 0.5
+    },
+    {
+      "name": "ScoreThreshold",
+      "type": "double",
+      "default": 0.5
     }
   ]
 }

src/Microsoft.ML.AutoML/CodeGen/search-space-schema.json

@@ -201,6 +201,7 @@
         "AddKeyValueAnnotationsAsText",
         "Arch",
         "PredictedLabelColumnName",
+        "PredictedBoundingBoxColumnName",
         "BoundingBoxColumnName",
         "ImageColumnName",
         "IOUThreshold",

src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/ObjectDetection.cs

@@ -24,6 +24,10 @@ public override IEstimator<ITransformer> BuildFromOption(MLContext context, Obje
                 MaxEpoch = param.MaxEpoch,
                 InitLearningRate = param.InitLearningRate,
                 WeightDecay = param.WeightDecay,
+                PredictedBoundingBoxColumnName = param.PredictedBoundingBoxColumnName,
+                ScoreThreshold = param.ScoreThreshold,
+                Steps = param.Steps,
+                IOUThreshold = param.IOUThreshold,
             };
 
             return context.MulticlassClassification.Trainers.ObjectDetection(option);