add readme

update train.py with yolov10 fix generation of data.yaml
2026-04-22 07:06:36 +00:00 · 2024-06-08 19:32:04 +03:00
parent 1f7b4a95d2
commit 56017f2c23
3 changed files with 29 additions and 16 deletions
@@ -0,0 +1,8 @@
+1. Install dependencies first
+```
+    python -m pip install --upgrade pip
+    pip install --upgrade huggingface_hub
+    pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
+    pip install -q git+https://github.com/THU-MIG/yolov10.git
+    pip install albumentations
+```