-
pytorch实现yolov4-tiny: yolov4-tiny的pytorch版本
资源介绍
yolov4-tiny
pytorch implementation of yolov4-tiny
requirements
torch
torchvision
opencv-python
Pillow
matplotlib
tqdm
train
add datasets
数据集按coco2017数据集形式存放
coco
├── labels/
│ ├── train2017/
│ │ └── 0001.txt (classes cx cy w h)
│ └── val2017/
└── images/
├── train2017/
│ └── 0001.jpg
└── val2017/
set datasets
修改config.py中的train_datasets_images_path、train_datasets_labels