update rit object detection formula

Signed-off-by: GuillaumeFalourd <[email protected]>
GuillaumeFalourd · Feb 28, 2021 · 1d5a413 · 1d5a413
1 parent d644992
commit 1d5a413
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/object/detection/src/formula/formula.py b/object/detection/src/formula/formula.py
@@ -8,8 +8,12 @@ def run(image_path):
 
     detector = ObjectDetection()
     detector.setModelTypeAsRetinaNet()
-    #detector.setModelPath( os.path.join(execution_path , "images/keanu_reeves.jpg"))
+
+    # Test
     detector.setModelPath("images/keanu_reeves.jpg")
+    #detector.setModelPath( os.path.join(execution_path , "images/keanu_reeves.jpg"))
+
+    # Load model on an image.jpg file
     detector.loadModel()
     detections = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image.jpg"), output_image_path=os.path.join(execution_path , "imagenew.jpg"))