Merge pull request #26 from freemocap/philip/patch_entrypoint

Patch: entrypoint arguments

.gitignore

@@ -158,6 +158,9 @@ cython_debug/
 # ML models
 *.pt
 
+# Data
+*.npy
+
 #image outputs
 *.jpg
 *.jpeg

README.md

@@ -8,3 +8,5 @@ Installation: `pip install skellytracker``
 Then it can be run with `skellytracker`.
 
 Running the basic `skellytracker` will open the first webcam port on your computer and run pose estimaiton in realtime with mediapipe holistic as a tracker. You can specify the tracker with `skellytracker TRACKER_NAME`, where `TRACKER_NAME` is the name of an available tracker. To view the names of all available trackers, see `RUN_ME.py`.
+
+It will take some time to initialize the tracker the first time you run it, as it will likely need to download the model.

pyproject.toml

@@ -83,7 +83,7 @@ push            = true
 "skellytracker/__init__.py" = ["{version}"]
 
 [project.scripts]
-skellytracker = "skellytracker.__main__:main"
+skellytracker = "skellytracker.__main__:cli_main"
 
 [tool.setuptools]
 py-modules = ["skellytracker"]

RUN_ME.py → skellytracker/RUN_ME.py

@@ -38,7 +38,7 @@ def main(demo_tracker: str = "mediapipe_holistic_tracker"):
         ).demo()
 
     elif demo_tracker == "yolo_tracker":
-        YOLOPoseTracker(model_size="high_res").demo()
+        YOLOPoseTracker(model_size="nano").demo()
     elif demo_tracker == "SAM_tracker":
         SAMTracker().demo()
     elif demo_tracker == "yolo_object_tracker":

skellytracker/__main__.py

@@ -10,13 +10,16 @@
 
 logger = logging.getLogger(__name__)
 
-from RUN_ME import main
+from skellytracker.RUN_ME import main
 
 
-if __name__ == "__main__":
+def cli_main():
     logger.info(f"Running as a script")
     if len(sys.argv) > 1:
-        demo_tracker = sys.argv[1]
+        demo_tracker = str(sys.argv[1])
     else:
         demo_tracker = "mediapipe_holistic_tracker"
     main(demo_tracker=demo_tracker)
+
+if __name__ == "__main__":
+    cli_main()