Skip to content

Refactor Data Module

Pre-release
Pre-release
Compare
Choose a tag to compare
@zjykzj zjykzj released this 28 Jun 13:09
· 70 commits to master since this release
v0.2.1
caa33e8
  1. Refactor data module;
  2. Fix demo.py;
  3. Update latest training results.
  • Train using the VOC07+12 trainval dataset and test using the VOC2007 Test dataset with an input size of 448x448. give the result as follows
Original (darknet) tztztztztz/yolov2.pytorch zjykzj/YOLOv2(This) zjykzj/YOLOv2(This) zjykzj/YOLOv2(This)
ARCH YOLOv2 YOLOv2 YOLOv2+Darknet53 YOLOv2 YOLOv2-tiny
VOC AP[IoU=0.50] 76.8 72.7 76.33 72.00 64.36
  • Train using the COCO train2017 dataset and test using the COCO val2017 dataset with an input size of 416x416. give the result as follows (Note: The results of the original paper were evaluated on the COCO test-dev2015 dataset)
Original (darknet) zjykzj/YOLOv2(This) zjykzj/YOLOv2(This)
ARCH YOLOv2 YOLOv2+Darknet53 YOLOv2
COCO AP[IoU=0.50:0.95] 21.6 24.98 22.01
COCO AP[IoU=0.50] 44.0 46.85 42.70
Assets 10
Loading