tensorflow :: ops :: NonMaxSuppressionWithOverlaps

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

요약

이전에 선택한 상자와 겹치는 부분이 많은 상자를 잘라냅니다. score_threshold 미만의 점수를 가진 경계 상자가 제거됩니다. Nxn 오버랩 값은 정사각형 매트릭스로 제공되어 사용자 지정 오버랩 기준을 정의 할 수 있습니다 (예 : 결합을 통한 교차, 영역에 대한 교차 등).

이 작업의 출력은 선택한 상자를 나타내는 경계 상자의 입력 컬렉션으로 인덱싱되는 정수 집합입니다. 선택한 인덱스에 해당하는 경계 상자 좌표는 tf.gather operation 사용하여 얻을 수 있습니다. 예를 들면 :

selected_indices = tf.image.non_max_suppression_with_overlaps (overlaps, scores, max_output_size, overlay_threshold, score_threshold) selected_boxes = tf.gather (boxes, selected_indices)

인수 :

• 범위 : 범위 개체
• 겹침 : [num_boxes, num_boxes] 상자 겹침 값을 나타내는 모양 [num_boxes, num_boxes] 의 2 차원 부동 텐서.
• scores : 각 상자 (각 상자 행)에 해당하는 단일 점수를 나타내는 1-D 부동 텐서 [num_boxes] 모양.
• max_output_size : 비 최대 억제에 의해 선택 될 최대 상자 수를 나타내는 정수 스칼라 텐서.
• overlap_threshold : 상자가 겹치는 지 여부를 결정하기위한 임계 값을 나타내는 0-D 부동 텐서입니다.
• score_threshold : 점수에 따라 상자를 제거 할시기를 결정하기위한 임계 값을 나타내는 0-D 부동 텐서입니다.

보고:

• Output : 상자 텐서에서 선택한 인덱스를 나타내는 [M] 모양의 1 차원 정수 텐서, 여기서 M <= max_output_size .

공용 속성

Operation operation

::tensorflow::Output selected_indices

 NonMaxSuppressionWithOverlaps(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input overlaps,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input overlap_threshold,
  ::tensorflow::Input score_threshold
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const