This Repo is my first excerise for reproducing YOLOv3. Basically can work.
Please go to https://github.com/NuayHL/CrowdDetection for advanced implemention.
- pip install -r requirments.txt
- pytorch
- pip install 'git+https://github.com/katsura-jp/pytorch-cosine-annealing-with-warmup'
git clone ""
- For single GPU training, using
train_single GPU.py
. - For multiple GPUs training, using
train.py
- Models
- Loading pre-trained models
- Loading images: from dataset: from single image
Copyright© NuayHL 2022. All right reserved