Skip to content

Commit ed7e3b6

Browse files
author
Saumya Saksena
committed
Update README*
1 parent ad1f89e commit ed7e3b6

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

.github/README.md

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff 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
4242
conda env create -f environment.yml
43+
44+
# Activate the conda env
4345
conda activate human-action-classification
4446

4547
# Basic installation
@@ -78,7 +80,7 @@ print(f"Action: {result['action']['top_class']}")
7880
print(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
9193
hac-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
9499
hac-demo --model weights/best.pth --share
95100
```
@@ -160,6 +165,17 @@ trainer = Trainer(
160165
trainer.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
```

0 commit comments

Comments
 (0)