Remove pyside code

README.md

@@ -1,20 +1,10 @@
-# skelly_tracker
+# skellytracker
 
-Currently, the bare bones of a tracking backend for freemocap.
+The tracking backend for freemocap. Collects different pose estimation tools and aggregates them using a consistent API. Can run pose estimation on images, webcams, and videos.
 
 ## Run skelly_tracker
 
-Currently, it is not pip installable. Download it by running `git clone https://github.com/freemocap/skelly_tracker`. 
+Installation: `pip install skellytracker``
+Then it can be run with `skellytracker`.
 
-Change directories to skelly_tracker with `cd skelly_tracker`, and install the requirements with `pip install .`.
-
-Then it can be run with `skelly_tracker`.
-
-## Using skelly_tracker
-
-A GUI should pop up that looks like this: 
-
-<img width="591" alt="Image of skelly_tracker GUI" src="https://github.com/freemocap/skelly_tracker/assets/24758117/8767a7fb-2535-407c-99fb-40295ee5af93">
-
-
-Choose your image in the file dialog and click run to see the image annotated with the tracking information.
+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`.

RUN_ME.py

@@ -1,35 +1,48 @@
 import cv2
 
-from skellytracker.trackers.bright_point_tracker.brightest_point_tracker import BrightestPointTracker
+from skellytracker.trackers.bright_point_tracker.brightest_point_tracker import (
+    BrightestPointTracker,
+)
 from skellytracker.trackers.charuco_tracker.charuco_tracker import CharucoTracker
-from skellytracker.trackers.mediapipe_tracker.mediapipe_holistic_tracker import MediapipeHolisticTracker
-from skellytracker.trackers.segment_anything_tracker.segment_anything_tracker import SAMTracker
+from skellytracker.trackers.mediapipe_tracker.mediapipe_holistic_tracker import (
+    MediapipeHolisticTracker,
+)
+from skellytracker.trackers.segment_anything_tracker.segment_anything_tracker import (
+    SAMTracker,
+)
 from skellytracker.trackers.yolo_tracker.yolo_tracker import YOLOPoseTracker
-from skellytracker.trackers.yolo_object_tracker.yolo_object_tracker import YOLOObjectTracker
+from skellytracker.trackers.yolo_object_tracker.yolo_object_tracker import (
+    YOLOObjectTracker,
+)
 
-if __name__ == "__main__":
-
-    demo_tracker = "yolo_object_tracker"
 
+def main(demo_tracker: str = "mediapipe_holistic_tracker"):
 
     if demo_tracker == "brightest_point_tracker":
         BrightestPointTracker().demo()
 
     elif demo_tracker == "charuco_tracker":
-        CharucoTracker(squaresX=7,
-                       squaresY=5,
-                       dictionary=cv2.aruco.getPredefinedDictionary(cv2.aruco.DICT_4X4_250)).demo()
+        CharucoTracker(
+            squaresX=7,
+            squaresY=5,
+            dictionary=cv2.aruco.getPredefinedDictionary(cv2.aruco.DICT_4X4_250),
+        ).demo()
 
     elif demo_tracker == "mediapipe_holistic_tracker":
-        MediapipeHolisticTracker(model_complexity=2,
-                                 min_detection_confidence=0.5,
-                                 min_tracking_confidence=0.5,
-                                 static_image_mode=False,
-                                 smooth_landmarks=True).demo()
+        MediapipeHolisticTracker(
+            model_complexity=2,
+            min_detection_confidence=0.5,
+            min_tracking_confidence=0.5,
+            static_image_mode=False,
+            smooth_landmarks=True,
+        ).demo()
 
     elif demo_tracker == "yolo_tracker":
         YOLOPoseTracker(model_size="high_res").demo()
     elif demo_tracker == "SAM_tracker":
         SAMTracker().demo()
     elif demo_tracker == "yolo_object_tracker":
         YOLOObjectTracker(model_size="medium").demo()
+
+if __name__ == "__main__":
+    main(demo_tracker="mediapipe_holistic_tracker")

pyproject.toml

@@ -56,9 +56,8 @@ keywords = [
 #dynamic = ["dependencies"]
 dependencies = [
     "opencv-contrib-python==4.8.*",
-    "pyside6~=6.6.0",
     "pydantic==1.*",
-    "mediapipe~=0.10.0",
+    "mediapipe==0.10.9",
     "ultralytics~=8.0.202",
 ]
 requires-python = ">=3.9,<3.12"

skellytracker/__main__.py

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