Upload run_training.sh
Browse files- run_training.sh +15 -0
run_training.sh
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
set -e
|
| 3 |
+
|
| 4 |
+
# Install system dependencies for OpenCV
|
| 5 |
+
apt-get update && apt-get install -y libgl1 libglib2.0-0 wget curl
|
| 6 |
+
|
| 7 |
+
# Install Python packages
|
| 8 |
+
pip install datasets==2.21.0 ultralytics huggingface_hub pyyaml Pillow
|
| 9 |
+
|
| 10 |
+
# Download and run training script
|
| 11 |
+
wget -O /app/train_ppe_improved_v2.py https://huggingface.co/baskarmother/ppe-training-scripts/resolve/main/train_ppe_improved_v2.py
|
| 12 |
+
|
| 13 |
+
# Run training
|
| 14 |
+
cd /app
|
| 15 |
+
python train_ppe_improved_v2.py
|