yjh0410 1a3a1391c8 update 10 달 전
..
README.md eb8e9e42d0 update 11 달 전
build.py e777a55fba modify yolov3 11 달 전
loss.py 1a3a1391c8 update 10 달 전
matcher.py e777a55fba modify yolov3 11 달 전
modules.py a77c0215af update v3 modules 11 달 전
yolov3.py 1a10122af0 update 11 달 전
yolov3_backbone.py e777a55fba modify yolov3 11 달 전
yolov3_fpn.py e777a55fba modify yolov3 11 달 전
yolov3_head.py e777a55fba modify yolov3 11 달 전
yolov3_neck.py e777a55fba modify yolov3 11 달 전

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