This is a person detector that is based on MobileNetV2 backbone with ATSS head for 1280x720 resolution.
Metric | Value |
---|---|
AP @ [ IoU=0.50:0.95 ] | 0.444 (internal test set) |
GFlops | 24.7577 |
MParams | 3.6302 |
Source framework | PyTorch* |
Average Precision (AP) is defined as an area under the precision/recall curve.
Image, name: input
, shape: 1, 3, 720, 1280
in the format B, C, H, W
, where:
B
- batch sizeC
- number of channelsH
- image heightW
- image width
Expected color order is BGR
.
-
The
boxes
is a blob with the shape100, 5
in the formatN, 5
, whereN
is the number of detected bounding boxes. For each detection, the description has the format: [x_min
,y_min
,x_max
,y_max
,conf
], where:- (
x_min
,y_min
) - coordinates of the top left bounding box corner - (
x_max
,y_max
) - coordinates of the bottom right bounding box corner. conf
- confidence for the predicted class
- (
-
The
labels
is a blob with the shape100
in the formatN
, whereN
is the number of detected bounding boxes. In case of person detection, it is equal to1
for each detected box with person in it and0
for the background.
[*] Other names and brands may be claimed as the property of others.