Trả lời cho sự kiện TensorFlow Everywhere tại địa phương của bạn ngay hôm nay!
Trang này được dịch bởi Cloud Translation API.
Switch to English

dòng chảy :: ops :: NonMaxSuppression

#include <image_ops.h>

Tham lam chọn một tập hợp con các hộp giới hạn theo thứ tự điểm số giảm dần,.

Tóm lược

loại bỏ các hộp có nhiều giao nhau (IOU) trùng với các hộp đã chọn trước đó. Các hộp giới hạn được cung cấp dưới dạng [y1, x1, y2, x2], trong đó (y1, x1) và (y2, x2) là tọa độ của bất kỳ cặp góc chéo nào của góc hộp và tọa độ có thể được cung cấp dưới dạng chuẩn hóa (nghĩa là nằm trong khoảng [0, 1]) hoặc tuyệt đối. Lưu ý rằng thuật toán này không xác định được điểm gốc trong hệ tọa độ. Lưu ý rằng thuật toán này bất biến đối với các phép biến đổi trực giao và phép tịnh tiến của hệ tọa độ; do đó dịch chuyển hoặc phản xạ của hệ tọa độ dẫn đến các hộp giống nhau được chọn bởi thuật toán. Đầu ra của thao tác này là một tập hợp các số nguyên lập chỉ mục vào bộ sưu tập đầu vào của các hộp giới hạn đại diện cho các hộp đã chọn. Sau đó có thể lấy tọa độ hộp giới hạn tương ứng với các chỉ số đã chọn bằng cách sử dụng tf.gather operation . Ví dụ: select_indices = tf.image.non_max_suppression (hộp, điểm, max_output_size, iou_threshold) select_boxes = tf.gather (ô, select_indices)

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • hộp: Một tenxơ nổi 2-D có hình dạng [num_boxes, 4] .
  • điểm: Một hàng chục float 1-D của hình [num_boxes] biểu thị một điểm duy nhất tương ứng với mỗi hộp (mỗi hàng hộp).
  • max_output_size: Một tensor số nguyên vô hướng biểu thị số lượng hộp tối đa được chọn bằng cách triệt tiêu không tối đa.

Các thuộc tính tùy chọn (xem Phần Attrs ):

  • iou_threshold: Một phao biểu thị ngưỡng để quyết định xem các hộp có chồng chéo quá nhiều so với IOU hay không.

Lợi nhuận:

  • Output : Một tensor số nguyên 1-D của hình [M] đại diện cho các chỉ số được chọn từ tensor hộp, trong đó M <= max_output_size .

Người xây dựng và Người phá hủy

NonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size)
NonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, const NonMaxSuppression::Attrs & attrs)

Thuộc tính công cộng

operation
selected_indices

Chức năng công cộng

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Chức năng tĩnh công khai

IouThreshold (float x)

Cấu trúc

tensorflow :: ops :: NonMaxSuppression :: Attrs

Bộ thiết lập thuộc tính tùy chọn cho NonMaxSuppression .

Thuộc tính công cộng

hoạt động

Operation operation

lựa chọn

::tensorflow::Output selected_indices

Chức năng công cộng

NonMaxSuppression

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size
)

NonMaxSuppression

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  const NonMaxSuppression::Attrs & attrs
)

nút

::tensorflow::Node * node() const 

toán tử :: tensorflow :: Đầu vào

 operator::tensorflow::Input() const 

toán tử :: tensorflow :: Đầu ra

 operator::tensorflow::Output() const 

Chức năng tĩnh công khai

IouThreshold

Attrs IouThreshold(
  float x
)