Add CI test

Author: Ian-I-Chien

.github/workflows/ci.yaml

@@ -0,0 +1,39 @@
+name: Code Quality and Tests
+
+on:
+  push:
+    branches:
+      - main
+      - '**'
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  pr_checks:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v2
+
+      - name: Set up Python
+        uses: actions/setup-python@v2
+        with:
+          python-version: '3.12'
+
+      - name: Install dependencies
+        run: |
+          pip install -r requirements.txt
+
+      - name: Download YOLOv7 pose model
+        run: |
+          curl -L -o yolov7-w6-pose.pt https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-w6-pose.pt
+
+      - name: Check Python script for errors
+        run: |
+          CI_MODE=1 python video.py
+
+      - name: Run Black to check code style
+        run: |
+          black --check .

.gitignore

@@ -28,6 +28,8 @@ data/images/*
 !data/images/zidane.jpg
 !data/images/bus.jpg
 !data/*.sh
+!fall_dataset/ci_videos/**
+!.github/workflows/ci.yaml
 
 results*.csv
 

fall_dataset/ci_videos/video_1.mp4

@@ -41,4 +41,6 @@ thop~=0.1.1.post2209072238
 # roboflow
 
 telepot~=12.7
-torchmetrics~=0.10.2
+torchmetrics~=0.10.2
+
+black==24.8.0

requirements.txt

@@ -53,7 +53,11 @@ def process_video_file(video_path, output_dir):
 
 
 if __name__ == "__main__":
-    videos_path = "fall_dataset/videos"
+    if os.environ.get("CI_MODE") == "1":
+        videos_path = "fall_dataset/ci_videos"
+        print("[CI MODE] Only running on CI test videos...")
+    else:
+        videos_path = "fall_dataset/videos"
     output_dir = "output_videos"
     os.makedirs(output_dir, exist_ok=True)