This model is an instance segmentation network for 80 classes of objects. It is a Mask R-CNN with ResNet101 backbone, light-weight FPN, RPN, detection and segmentation heads.
Metric | Value |
---|---|
MS COCO val2017 box AP | 38.85% |
MS COCO val2017 mask AP | 33.9% |
Max objects to detect | 100 |
GFlops | 147.2352 |
MParams | 49.8328 |
Source framework | PyTorch* |
Average Precision (AP) is defined and measured according to standard MS COCO evaluation procedure.
- name:
image
, shape: [1x3x608x608] - An input image in the format [1xCxHxW]. The expected channel order is BGR.
- name:
labels
, shape: [100] - Contiguous integer class ID for every detected object. - name:
boxes
, shape: [100, 5] - Bounding boxes around every detected objects in (top_left_x, top_left_y, bottom_right_x, bottom_right_y) format and its confidence score in range [0, 1]. - name:
masks
, shape: [100, 14, 14] - Segmentation heatmaps for every output bounding box.
[*] Other names and brands may be claimed as the property of others.