yjh0410 9c6622cdaa update hace 1 año
..
README.md cc84910643 update hace 1 año
build.py 264112178f build a new YOLO-Tutorial project for my book hace 1 año
loss.py 60d7d14b4b add RTCDet hace 1 año
matcher.py 264112178f build a new YOLO-Tutorial project for my book hace 1 año
yolov8.py 264112178f build a new YOLO-Tutorial project for my book hace 1 año
yolov8_backbone.py e87e86fb0a update hace 1 año
yolov8_basic.py 9c6622cdaa update hace 1 año
yolov8_head.py f6aa3d89dd update hace 1 año
yolov8_neck.py c304d2acc8 modify init hace 1 año
yolov8_pafpn.py f6aa3d89dd update hace 1 año
yolov8_pred.py f6aa3d89dd update hace 1 año

README.md

YOLOv8:

  • VOC
  • COCO
Model Batch Scale APval
0.5
Weight Logs
YOLOv8-S 1xb16 640 83.6 ckpt log

Train YOLOv8

Single GPU

Taking training YOLOv8-S on COCO as the example,

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

Multi GPU

Taking training YOLOv8-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 yolov8_s -bs 256 --fp16 

Test YOLOv8

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

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

Evaluate YOLOv8

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

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

Demo

Detect with Image

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

Detect with Video

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

Detect with Camera

python demo.py --mode camera --cuda -m yolov8_s --weight path/to/weight --show --gif
Model Batch Scale APval
0.5:0.95
APval
0.5
FLOPs
(G)
Params
(M)
Weight Logs
YOLOv8-S 1xb16 640 26.9 8.9