File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -83,7 +83,7 @@ you can use darknet2pytorch to convert it yourself, or download my converted mod
8383**ONNX and TensorRT models are converted from Pytorch (TianXiaomo): Pytorch->ONNX->TensorRT.**
8484See following sections for more details of conversions.
8585
86- - val2017 dataset (416x416)
86+ - val2017 dataset (input size: 416x416)
8787
8888| Model type | AP | AP50 | AP75 | APS | APM | APL |
8989| ------------------- | ----------: | ----------: | ----------: | ----------: | ----------: | ----------: |
@@ -93,14 +93,15 @@ See following sections for more details of conversions.
9393| TensorRT FP32 + BatchedNMSPlugin | 0.472| 0.708 | 0.511 | 0.273 | 0.530 | 0.637 |
9494| TensorRT FP16 + BatchedNMSPlugin | 0.472| 0.708 | 0.511 | 0.273 | 0.530 | 0.636 |
9595
96- - testdev2017 dataset (416x416)
96+ - testdev2017 dataset (input size: 416x416)
9797
9898| Model type | AP | AP50 | AP75 | APS | APM | APL |
9999| ------------------- | ----------: | ----------: | ----------: | ----------: | ----------: | ----------: |
100100| DarkNet (YOLOv4 paper)| 0.412 | 0.628 | 0.443 | 0.204 | 0.444 | 0.560 |
101101| Pytorch (TianXiaomo)| 0.404 | 0.615 | 0.436 | 0.196 | 0.438 | 0.552 |
102102| ONNX | incoming | incoming | incoming | incoming | incoming | incoming |
103103| TensorRT FP32 + BatchedNMSPlugin | 0.412| 0.625 | 0.445 | 0.200 | 0.446 | 0.564 |
104+ | TensorRT FP16 + BatchedNMSPlugin | 0.412| 0.625 | 0.445 | 0.200 | 0.446 | 0.563 |
104105
105106
106107## 2.2 Image input size for inference
You can’t perform that action at this time.
0 commit comments