This is a person detector that is based on Cascade R-CNN architecture with ResNet50 backbone.
Metric | Value |
---|---|
AP @ [ IoU=0.50:0.95 ] | 0.442 (internal test set) |
GFlops | 404.264 |
MParams | 71.565 |
Source framework | PyTorch* |
Average Precision (AP) is defined as an area under the precision/recall curve.
Image, name: image
, shape: 1, 3, 800, 1344
in the format B, C, H, W
, where:
B
- batch sizeC
- number of channelsH
- image heightW
- image width
Expected color order is BGR
.
Model has outputs with dynamic shapes.
-
The
boxes
is a blob with the shape-1, 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 shape-1
in the formatN
, whereN
is the number of detected bounding boxes. It contains predicted class ID (0 - person) per each detected box.
The model can be used in the following demos provided by the Open Model Zoo to show its capabilities:
[*] Other names and brands may be claimed as the property of others.