tensorflow :: ops :: NonMaxSuppressionV5

탐욕스럽게 점수의 내림차순으로 경계 상자의 하위 집합을 선택합니다.

요약

IOU (Intersection-over-union)가 높은 상자 잘라내 기는 이전에 선택한 상자와 겹칩니다. score_threshold 미만의 점수를 가진 경계 상자가 제거됩니다. 경계 상자는 [y1, x1, y2, x2]로 제공됩니다. 여기서 (y1, x1) 및 (y2, x2)는 대각선 상자 모서리 쌍의 좌표이며 좌표는 정규화 (예 : 간격 [0, 1]) 또는 절대. 이 알고리즘은 원점이 좌표계에있는 위치와 무관하며보다 일반적으로 좌표계의 직교 변환 및 변환에 불변합니다. 따라서 좌표계를 변환하거나 반사하면 알고리즘에 의해 동일한 상자가 선택됩니다. 이 작업의 출력은 선택한 상자를 나타내는 경계 상자의 입력 컬렉션으로 인덱싱되는 정수 집합입니다. 선택한 인덱스에 해당하는 경계 상자 좌표는 tf.gather operation 사용하여 얻을 수 있습니다. 예 : selected_indices = tf.image.non_max_suppression_v2 (boxes, scores, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather (boxes, selected_indices)이 작업은 Soft-NMS (가우스 가중치 포함) 모드도 지원합니다 (참조 : Bodla et al , https://arxiv.org/abs/1704.04503 ) 여기서 상자는 다른 겹치는 상자의 점수를 직접 잘라 내지 않고 줄입니다. 이 Soft-NMS 모드를 활성화하려면 soft_nms_sigma 매개 변수를 0보다 크게 설정하십시오.

인수 :

• 범위 : 범위 개체
• 상자 : 모양의 2 차원 부동 텐서 [num_boxes, 4] .
• scores : 각 상자 (각 상자 행)에 해당하는 단일 점수를 나타내는 1-D 부동 텐서 [num_boxes] 모양.
• max_output_size : 비 최대 억제에 의해 선택 될 최대 상자 수를 나타내는 정수 스칼라 텐서.
• iou_threshold : IOU와 관련하여 상자가 너무 많이 겹치는 지 여부를 결정하기위한 임계 값을 나타내는 0-D 부동 텐서.
• score_threshold : 점수에 따라 상자를 제거 할시기를 결정하기위한 임계 값을 나타내는 0-D 부동 텐서입니다.
• soft_nms_sigma : Soft NMS에 대한 시그마 매개 변수를 나타내는 0-D 부동 텐서. Bodla et al 참조 ( https://arxiv.org/abs/1704.04503 참조 ). soft_nms_sigma=0.0 (기본값)이면 표준 (하드) NMS로 돌아갑니다.

선택적 속성 ( Attrs 참조) :

• pad_to_max_output_size : true이면 selected_indices 출력이 max_output_size 길이로 채워집니다. 기본값은 false입니다.

보고:

• selected_indices Output : 상자 텐서에서 선택한 인덱스를 나타내는 [M] 모양의 1 차원 정수 텐서, 여기서 M <= max_output_size .
• selected_scores Output : 선택한 각 상자에 해당하는 점수를 나타내는 모양 [M] 의 1-D 부동 텐서, 여기서 M <= max_output_size . 점수는 Soft NMS를 사용할 때 (즉, soft_nms_sigma>0 ) 해당 입력 점수와 만 다릅니다.
• Output valid_outputs : selected_indices 의 유효한 요소 수를 나타내는 0-D 정수 텐서이며 유효한 요소가 먼저 나타납니다.

공용 속성

Operation operation

::tensorflow::Output selected_indices

::tensorflow::Output selected_scores

::tensorflow::Output valid_outputs

 NonMaxSuppressionV5(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  ::tensorflow::Input soft_nms_sigma
)

 NonMaxSuppressionV5(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  ::tensorflow::Input soft_nms_sigma,
  const NonMaxSuppressionV5::Attrs & attrs
)

Attrs PadToMaxOutputSize(
  bool x
)