Added integration with Object Detector effect

Author: BrennoCaldato

src/classes/effect_init.py

@@ -150,5 +150,42 @@
             "value": 30,
             "type": "spinner"
         }
+    ],
+
+    "Object Detector": [
+        {
+            "value": "../yolo/yolov3.weights",
+            "title": "Model Weights",
+            "type": "text",
+            "setting": "model-weights"
+        },
+        {
+            "value": "../yolo/yolov3.cfg",
+            "title": "Model Config",
+            "type": "text",
+            "setting": "model-config"
+        },
+        {
+            "value": "../yolo/obj.names",
+            "title": "Class names",
+            "type": "text",
+            "setting": "class-names"
+        },
+        {
+            "title": "Processing Device",
+            "setting": "processing-device",
+            "value": "GPU",
+            "values": [
+                {
+                    "value": "GPU",
+                    "name": "GPU"
+                },
+                {
+                    "value": "CPU",
+                    "name": "CPU"
+                }
+            ],
+            "type": "dropdown",
+        }
     ]
 }

src/effects/icons/objectdetector.png

@@ -372,7 +372,21 @@ def generateJson(self, protobufPath):
             # Get processing start frame
             print(self.context["region"])
             jsonString +=',"first_frame": %d' % (self.context["region"]["first-frame"])
-        
+
+        # Special case where more info is needed for the JSON string
+        if self.effect_name == "Object Detector":
+            modelweightsPath = self.context["model-weights"]
+            jsonString += ', "model_weights": "%s"' % modelweightsPath
+
+            modelConfigPath = self.context["model-config"]
+            jsonString += ', "model_configuration": "%s"' % modelConfigPath
+                
+            classNamesPath = self.context["class-names"]
+            jsonString += ', "classes_file": "%s"' % classNamesPath
+
+            processingDevice  = self.context["processing-device"]
+            jsonString += ', "processing_device": "%s"' % processingDevice
+
         # Finish JSON string
         jsonString+='}'
         return jsonString

src/windows/process_effect.py

@@ -2958,10 +2958,11 @@ def addEffect(self, effect_names, position):
             if js_position == 0 or (clip.data["position"] <= js_position <= clip.data["position"] + (clip.data["end"] - clip.data["start"])):
                 log.info("Applying effect to clip")
                 log.info(clip)
-
+                print(name)
+                print(effect_options)
                 # Handle custom effect dialogs
                 if name in effect_options:
-
+                    
                     # Get effect options
                     effect_params = effect_options.get(name)