이 작업은 arXiv:1506.01497의 eq.2에 따라 주어진 경계 상자(bbox_deltas)로 인코딩된 wrt 앵커로부터 관심 영역을 생성합니다.
op는 상위 `pre_nms_topn` 점수 상자를 선택하고 앵커와 관련하여 디코딩하며 `nms_threshold` 교차-결합(iou) 값보다 높은 중첩 상자에 최대 비최대 억제를 적용하고 짧은 변이 `보다 작은 상자를 삭제합니다. 최소_크기`. 입력: `scores`: 주어진 위치 `bbox_deltas`에서 앵커당 점수를 포함하는 [Batch, Height, Width, Num Anchors] 모양의 4D 텐서: [Batch, Height, Width, 4 x Num Anchors] 모양의 텐서입니다. 각 앵커 '앵커'에 인코딩된 상자: 앵커를 나타내는 [4 x Num Anchors] 모양의 1D 텐서. 출력: `rois`: [Batch, post_nms_topn, 4] 모양의 3D 텐서인 RoI를 출력합니다. post_nms_topn 후보보다 작을 경우 0으로 채워집니다. `roi_probabilities`: [Batch,post_nms_topn] 모양의 2D 텐서인 'rois'에 있는 각 roi의 확률 점수, 필요한 경우 0으로 채워지고 점수별로 정렬됩니다.
중첩 클래스
| 수업 | 생성BoundingBoxProposals.Options | GenerateBoundingBoxProposals 의 선택적 속성 | |
공개 방법
| 정적 생성BoundingBoxProposals | |
| 정적 생성BoundingBoxProposals.Options | postNmsTopn (긴 postNmsTopn) |
| 출력 <플로트> | roi확률 () 동일한 인덱스의 `rois` 텐서에서 관심 영역 상자의 점수를 나타내는 `[num_images, post_nms_topn]` 모양의 2D 부동 텐서입니다. |
| 출력 <플로트> | 로이스 () 선택된 관심 영역 상자를 나타내는 `[num_images,post_nms_topn,4]` 모양의 3D 부동 텐서. |
상속된 메서드
공개 방법
공개 정적 생성BoundingBoxProposals 생성 ( 범위 범위, 피연산자 <Float> 점수, 피연산자 <Float> bboxDeltas, 피연산자 <Float> imageInfo, 피연산자 <Float> 앵커, 피연산자 <Float> nmsThreshold, 피연산자 <Integer> preNmsTopn, 피연산자 <Float> minSize, 옵션... 옵션)
새로운 GenerateBoundingBoxProposals 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
| 범위 | 현재 범위 |
|---|---|
| 점수 | 주어진 앵커에 대한 상자 점수를 포함하는 `[num_images, height, width, num_achors]` 모양의 4-D 부동 텐서는 정렬 해제될 수 있습니다. |
| bbox델타 | `[num_images, height, width, 4 x num_anchors]` 모양의 4D 부동 텐서. 각 앵커와 관련된 인코딩 상자. 좌표는 [dy, dx, dh, dw] 형식으로 제공됩니다. |
| 이미지정보 | 이미지 정보 높이, 너비, 크기를 포함하는 '[num_images, 5]' 모양의 2D 부동 텐서. |
| 앵커 | 앵커 상자를 설명하는 '[num_anchors, 4]' 모양의 2D 부동 텐서. 상자는 [y1, x1, y2, x2] 형식으로 형식화됩니다. |
| nms임계값 | 비최대 억제 임계값을 위한 스칼라 부동 텐서. |
| preNmsTopn | 입력으로 사용할 최고 점수 상자 수에 대한 스칼라 int 텐서입니다. |
| 최소 크기 | 스칼라 부동 텐서. min_size보다 작은 크기의 상자는 모두 삭제됩니다. |
| 옵션 | 선택적 속성 값을 전달합니다. |
보고
- 생성BoundingBoxProposals의 새로운 인스턴스
공개 정적 생성BoundingBoxProposals.Options postNmsTopn (Long postNmsTopn)
매개변수
| postNmsTopn | 정수. 출력의 최대 Rois 수입니다. |
|---|
공개 출력 <Float> roiProbaries ()
동일한 인덱스의 `rois` 텐서에서 관심 영역 상자의 점수를 나타내는 `[num_images, post_nms_topn]` 모양의 2D 부동 텐서입니다.
공개 출력 <Float> rois ()
선택된 관심 영역 상자를 나타내는 `[num_images,post_nms_topn,4]` 모양의 3D 부동 텐서. 점수 내림차순으로 정렬됩니다.