YOLOv8 re-implementation for person detection using PyTorch
conda create -n YOLO python=3.8
conda activate YOLO
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch-lts
pip install opencv-python==4.5.5.64
pip install PyYAML
pip install tqdm
- Configure your dataset path in
main.pyfor training - Run
bash main.sh $ --trainfor training,$is number of GPUs
- Configure your dataset path in
main.pyfor testing - Run
python main.py --testfor testing
| Version | COCO weights | CrowdHuman weights |
|---|---|---|
| v8_n | model | model |
| v8_s | model | - |
| v8_m | model | - |
| v8_l | model | - |
| v8_x | model | - |
- the weights are ported from original repo, see reference
├── Person
├── images
├── train2017
├── 1111.jpg
├── 2222.jpg
├── val2017
├── 1111.jpg
├── 2222.jpg
├── labels
├── train2017
├── 1111.txt
├── 2222.txt
├── val2017
├── 1111.txt
├── 2222.txt
- Public person detection datasets