yjh0410 1a3a1391c8 update há 10 meses atrás
..
README.md eb8e9e42d0 update há 11 meses atrás
build.py e777a55fba modify yolov3 há 11 meses atrás
loss.py 1a3a1391c8 update há 10 meses atrás
matcher.py e777a55fba modify yolov3 há 11 meses atrás
modules.py a77c0215af update v3 modules há 11 meses atrás
yolov3.py 1a10122af0 update há 11 meses atrás
yolov3_backbone.py e777a55fba modify yolov3 há 11 meses atrás
yolov3_fpn.py e777a55fba modify yolov3 há 11 meses atrás
yolov3_head.py e777a55fba modify yolov3 há 11 meses atrás
yolov3_neck.py e777a55fba modify yolov3 há 11 meses atrás

README.md

Redesigned YOLOv3:

  • VOC
  • COCO
Model Batch Scale APval
0.5
Weight
YOLOv3 1xb16 640 ckpt

Train YOLOv3

Single GPU

Taking training YOLOv3-S on COCO as the example,

python train.py --cuda -d coco --root path/to/coco -m yolov3 -bs 16 --fp16 

Multi GPU

Taking training YOLOv3-S on COCO as the example,

python -m torch.distributed.run --nproc_per_node=8 train.py --cuda --distributed -d coco --root path/to/coco -m yolov3 -bs 16 --fp16 

Test YOLOv3

Taking testing YOLOv3-S on COCO-val as the example,

python test.py --cuda -d coco --root path/to/coco -m yolov3 --weight path/to/yolov3.pth --show 

Evaluate YOLOv3

Taking evaluating YOLOv3-S on COCO-val as the example,

python eval.py --cuda -d coco --root path/to/coco -m yolov3 --weight path/to/yolov3.pth 

Demo

Detect with Image

python demo.py --mode image --path_to_img path/to/image_dirs/ --cuda -m yolov3 --weight path/to/weight --show

Detect with Video

python demo.py --mode video --path_to_vid path/to/video --cuda -m yolov3 --weight path/to/weight --show --gif

Detect with Camera

python demo.py --mode camera --cuda -m yolov3 --weight path/to/weight --show --gif
Model Batch Scale APval
0.5:0.95
APval
0.5
FLOPs
(G)
Params
(M)
Weight
YOLOv3 1xb16 640 ckpt