Files
ai-training/README.md
T
Oleksandr Bezdieniezhnykh 56017f2c23 add readme
update train.py with yolov10
fix generation of data.yaml
2024-06-08 19:32:04 +03:00

8 lines
308 B
Markdown

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
```