yjh0410 60fbb4ddd7 update 1 年之前
..
README.md 60fbb4ddd7 update 1 年之前
build.py 60fbb4ddd7 update 1 年之前
loss.py 60fbb4ddd7 update 1 年之前
matcher.py 60fbb4ddd7 update 1 年之前
rtcdet.py 60fbb4ddd7 update 1 年之前
rtcdet_backbone.py 60fbb4ddd7 update 1 年之前
rtcdet_basic.py 60fbb4ddd7 update 1 年之前
rtcdet_head.py 60fbb4ddd7 update 1 年之前
rtcdet_neck.py 60fbb4ddd7 update 1 年之前
rtcdet_pafpn.py 60fbb4ddd7 update 1 年之前
rtcdet_pred.py 60fbb4ddd7 update 1 年之前

README.md

RTCDet: My Empirical Study of Real-Time Convolutional Object Detectors.

  • VOC
  • COCO
Model Batch Scale APval
0.5
Weight Logs
RTCDet-S 1xb16 640

Train RTCDet

Single GPU

Taking training RTCDet-S on COCO as the example,

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

Multi GPU

Taking training RTCDet-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 rtcdet_s -bs 256 --fp16 

Test RTCDet

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

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

Evaluate RTCDet

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

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

Demo

Detect with Image

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

Detect with Video

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

Detect with Camera

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