Skip to content

Commit e3e67f2

Browse files
committed
minor updates
1 parent b7d4b97 commit e3e67f2

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

README.md

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -78,12 +78,12 @@ you can use darknet2pytorch to convert it yourself, or download my converted mod
7878
7979
# 2. Inference (Evolving)
8080
81-
## 2.1 416 * 416 Performance on MS COCO dataset (using pretrained DarknetWeights from <https://github.com/AlexeyAB/darknet>)
81+
## 2.1 Performance on MS COCO dataset (using pretrained DarknetWeights from <https://github.com/AlexeyAB/darknet>)
8282
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
86+
- val2017 dataset (416x416)
8787
8888
| Model type | AP | AP50 | AP75 | APS | APM | APL |
8989
| ------------------- | ----------: | ----------: | ----------: | ----------: | ----------: | ----------: |
@@ -92,7 +92,7 @@ See following sections for more details of conversions.
9292
| TensorRT FP32 + BatchedNMSPlugin | 0.472| 0.708 | 0.511 | 0.273 | 0.530 | 0.637 |
9393
| TensorRT FP16 + BatchedNMSPlugin | 0.472| 0.708 | 0.511 | 0.273 | 0.530 | 0.636 |
9494
95-
- testdev2017 dataset
95+
- testdev2017 dataset (416x416)
9696
9797
| Model type | AP | AP50 | AP75 | APS | APM | APL |
9898
| ------------------- | ----------: | ----------: | ----------: | ----------: | ----------: | ----------: |
@@ -101,6 +101,7 @@ See following sections for more details of conversions.
101101
| ONNX | incoming | incoming | incoming | incoming | incoming | incoming |
102102
| TensorRT FP32 + BatchedNMSPlugin | 0.412| 0.625 | 0.445 | 0.200 | 0.446 | 0.564 |
103103
104+
104105
## 2.2 Image input size for inference
105106
106107
Image input size is NOT restricted in `320 * 320`, `416 * 416`, `512 * 512` and `608 * 608`.
@@ -132,11 +133,11 @@ Larger input size could help detect smaller targets, but may be slower and GPU m
132133
133134
## 2.4 Inference output
134135
135-
There are 2 inference outputs.
136-
- One is locations of bounding boxes, its shape is `[batch, num_boxes, 1, 4]` which represents x1, y1, x2, y2 of each bounding box.
137-
- The other one is scores of bounding boxes which is of shape `[batch, num_boxes, num_classes]` indicating scores of all classes for each bounding box.
136+
There are 2 inference outputs.
137+
- One is locations of bounding boxes, its shape is `[batch, num_boxes, 1, 4]` which represents x1, y1, x2, y2 of each bounding box.
138+
- The other one is scores of bounding boxes which is of shape `[batch, num_boxes, num_classes]` indicating scores of all classes for each bounding box.
138139
139-
Until now, still a small piece of post-processing including NMS is required. We are trying to minimize time and complexity of post-processing.
140+
Until now, still a small piece of post-processing including NMS is required. We are trying to minimize time and complexity of post-processing.
140141
141142
142143
# 3. Darknet2ONNX (Evolving)

0 commit comments

Comments
 (0)