RCNNProposal
Input
- cls_score_softmax:
- Required: yes
- Type: float16
- Description: confidence value
- Restriction: NCHW, with shape [b * topn1, 4, 1, 1], where b indicates the batch size and topn1 equals top_n of the RPNProposalSSD layer.
- bbox_pred:
- Required: yes
- Type: float16
- Description: coordinate offset of the BBox, which is used to fine-tune rois
- Restriction: NCHW. Has shape is [b * topn1, 8, 1, 1] (foreground and background) when regress_agnostic attribute is set to true; else, [b* topn1, 16, 1, 1] (background and three classes), where, b indicates the batch size and topn1 equals top_n of the RPNProposalSSD layer.
- rois:
- Required: yes
- Type: float16
- Description: BBox coordinates
- Restriction: NCHW, with shape [b, topn1, 8, 1], where b indicates the batch size and topn1 equals top_n of the RPNProposalSSD layer.
- im_info:
- Required: yes
- Type: float16
- Description: image information
- Restriction: NCHW, with shape [b, 6, 1, 1], where b indicates the batch size
Attribute
- bbox_mean:
- Required: yes
- Type: listFloat
- Description: mean value for BBox normalization
- Restriction: either 0 or 4
- bbox_std:
- Required: yes
- Type: listFloat
- Description: standard deviation for BBox normalization
- Restriction: either 0 or 4
- num_class:
- Required: no
- Type: int
- Description: number of classes
- Restriction: Must be 3.
- rpn_proposal_output_score:
- Required: no
- Type: bool
- Description: whether to output the confidence value of each class
- Restriction: Must be true.
- regress_agnostic:
- Required: no
- Type: bool
- Description: whether to use the foreground class
- Restriction: none
- min_size_h:
- Required: yes
- Type: float
- Description: minimum height of the BBox (default = 8.8008)
- Restriction: none
- min_size_w:
- Required: yes
- Type: float
- Description: minimum width of the BBox (default = 8.8008)
- Restriction: none
- min_size_mode:
- Required: yes
- Type: str
- Description: minimum size mode enum, default: HEIGHT_OR_WIDTH (HEIGHT_OR_WIDTH=0,HEIGHT_OR_WIDTH=1)
- Restriction: none
- threshold_objectness:
- Required: yes
- Type: float
- Description: foreground threshold
- Restriction: Must be in the range (0, 1).
- threshold:
- Required: yes
- Type: listFloat
- Description: confidence threshold of each class (size = num_class)
- Restriction: Must be in the range (0, 1).
- refine_out_of_map_bbox:
- Required: yes
- Type: bool
- Description: whether to refine out-of-map BBoxes
- Restriction: none
- overlap_ratio:
- Required: yes
- Type: listFloat
- Description: Intersection over Union (IoU) threshold for NMS
- Restriction: none
- top_n:
- Required: yes
- Type: listInt
- Description: number of BBoxes after the NMS operation
- Restriction: none
- max_candidate_n:
- Required: yes
- Type: listInt
- Description: maximum number of candidate BBoxes for NMS
- Restriction: none
- bsz01:
- Required: no
- Type: float
- Description: BBox size scaling factor
- Restriction: none
- allow_border:
- Required: no
- Type: float
- Description: allowed border, in pixels
- Restriction: none
- allow_border_ratio:
- Required: no
- Type: float
- Description: allowed border ratio
- Restriction: none
Output
bboxes:
- Required: yes
- Type: float16
- Description: output BBox information, including the ID of the image, BBox coordinates, and confidence values: | img_idx | x1 | y1 | x2 | y2 | background_score | cls1_score | cls2_score | cls3_score |
- Restriction: NCHW, with shape [b, topn, 9, 1], where b indicates the batch size and topn equals top_n
Parent topic: Supported Caffe Operators