File tree Expand file tree Collapse file tree 1 file changed +18
-2
lines changed Expand file tree Collapse file tree 1 file changed +18
-2
lines changed Original file line number Diff line number Diff line change @@ -38,8 +38,10 @@ Modern human action classification using **MediaPipe** pose estimation and **PyT
3838### Installation
3939
4040``` bash
41- # Creat the conda environments
41+ # Creat the conda env
4242conda env create -f environment.yml
43+
44+ # Activate the conda env
4345conda activate human-action-classification
4446
4547# Basic installation
@@ -78,7 +80,7 @@ print(f"Action: {result['action']['top_class']}")
7880print (f " Confidence: { result[' action' ][' top_confidence' ]:.2f } " )
7981```
8082
81- ### Command Line
83+ ### Command Line: Quickstart
8284
8385``` bash
8486# Single image
@@ -90,6 +92,9 @@ hac-infer --video video.mp4 --model weights/best.pth
9092# Real-time webcam
9193hac-infer --webcam --model weights/best.pth
9294
95+ # Real-time webcam (uses pretrained ImageNet backbone)
96+ python scripts/simple_demo.py --webcam
97+
9398# Launch web demo
9499hac-demo --model weights/best.pth --share
95100```
@@ -160,6 +165,17 @@ trainer = Trainer(
160165trainer.train()
161166```
162167
168+ ### 4. Train on Your Data
169+
170+ ``` bash
171+ # Organize data as: data/train/class1/, data/val/class1/, ...
172+ python -m hac.training.train \
173+ --data_dir data/ \
174+ --model_name mobilenetv3_small_100 \
175+ --num_classes 40 \
176+ --epochs 50
177+ ```
178+
163179## Architecture
164180
165181```
You can’t perform that action at this time.
0 commit comments