.vscode/launch.json

@@ -14,16 +14,6 @@
             ],
             "justMyCode": true
         },
-        // {
-        //     "name": "Quick, Specific Debug",
-        //     "type": "python",
-        //     "request": "launch",
-        //     "module": "wyzely_detect",
-        //     "args": [
-        //         "--run-scale", "0.25", "--view-scale", "0.5", "--no-remove-representations", "--detect-object", "person", "--detect-object", "cell phone"
-        //     ],
-        //     "justMyCode": true
-        // },
         {
             // "name": "Python: Module",
             "name": "Full Debug",

wyzely_detect/__main__.py

@@ -1,5 +1,6 @@
 # import face_recognition
 from pathlib import Path
+import os
 import cv2
 
 # import hjson as json
@@ -15,7 +16,7 @@ args = None
 
 def main():
     global objects_and_peoples
-    global args 
+    global args
     # RUN_BY_COMPOSE = os.getenv("RUN_BY_COMPOSE") # Replace this with code to check for gpu
 
     args = argparser.parse_args()

wyzely_detect/utils/cli_args.py

@@ -154,7 +154,7 @@ def set_argparse():
     object_detection = argparser.add_argument_group("Object Detection options")
     object_detection.add_argument(
         "--detect-object",
-        action="append",
+        nargs="*",
         default=[],
         type=str,
         help="The object(s) to detect. Must be something the model is trained to detect",

wyzely_detect/utils/utils.py

@@ -100,7 +100,7 @@ def process_footage(
             # Also, make sure that the objects to detect are in the list of objects_and_peoples
             # If it isn't, print a warning
             for obj in detect_object:
-                if obj not in objects_and_peoples["objects"].keys():
+                if obj not in objects_and_peoples:
                     print(
                         f"Warning: {obj} is not in the list of objects the model can detect!"
                     )