UPDATE
Pre-release
Pre-release
Based on zjykzj/YOLOv2, the entire project has been restructured, including training modules, data modules, and inference modules. In addition, the loss function YOLOv1Loss is reimplemented, which is completely defined by reference to YOLOv1 paper.
| Original (darknet) | Original (darknet) | abeardear/pytorch-YOLO-v1 | zjykzj/YOLOv1(This) | zjykzj/YOLOv1(This) | |
|---|---|---|---|---|---|
| arch | YOLOv1 | FastYOLOv1 | ResNet_YOLOv1 | YOLOv1 | FastYOLOv1 |
| train | VOC07+12 trainval | VOC07+12 trainval | VOC07+12 trainval | VOC07+12 trainval | VOC07+12 trainval |
| val | VOC2007 Test | VOC2007 Test | VOC2007 Test | VOC2007 Test | VOC2007 Test |
| VOC AP[IoU=0.50] | 63.4 | 52.7 | 66.5 | 52.95 | 43.65 |
| conf_thre | / | / | 0.5 | 0.005 | 0.005 |
| nms_thre | / | / | / | 0.45 | 0.45 |
| input_size | 448 | 448 | 448 | 448 | 448 |