Skip to content

Commit 6d015a4

Browse files
committed
add test2017 result for TRT fp16
1 parent 174e720 commit 6d015a4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

README.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff 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.**
8484
See 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

0 commit comments

Comments
 (0)