dòng chảy :: hoạt động :: Kết hợpNonMaxSuppression

#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

Thao tác này thực hiện non_max_suppression trên các đầu vào mỗi lô, trên tất cả các lớp. Cắt bỏ các hộp có giao điểm cao (IOU) chồng chéo 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 độ. Cũng lưu ý rằng thuật toán này là 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à các ô cuối cùng, điểm số và lớp được trả về sau khi thực hiện non_max_suppression.

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • hộp: Một tenxơ nổi 4-D có hình dạng [batch_size, num_boxes, q, 4] . Nếu q là 1 thì các hộp giống nhau được sử dụng cho tất cả các lớp, ngược lại, nếu q bằng số lớp, các hộp dành riêng cho lớp được sử dụng.
  • điểm: Một tensor nổi 3-D của hình dạng [batch_size, num_boxes, num_classes] đại diện cho 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_per_class: Một tensor số nguyên vô hướng đại diện cho số lượng hộp tối đa được chọn bằng cách triệt tiêu không max cho mỗi lớp
  • max_total_size: Một đại lượng vô hướng đại diện cho số lượng hộp tối đa được giữ lại trên tất cả các lớp.
  • iou_threshold: Một bộ căng phao 0-D đại diện cho 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.
  • score_threshold: Một dấu thăng 0-D biểu thị ngưỡng để quyết định thời điểm loại bỏ các hộp dựa trên điểm số.

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

  • pad_per_class: Nếu sai, các hộp, điểm số và lớp nmsed đầu ra được đệm / cắt thành max_total_size . Nếu đúng, các hộp, điểm và lớp có nmsed đầu ra được đệm để có độ dài max_size_per_class * num_classes , trừ khi nó vượt quá max_total_size trong trường hợp đó nó được cắt thành max_total_size . Giá trị mặc định là false.
  • clip_boxes: Nếu đúng, giả sử tọa độ hộp nằm giữa [0, 1] và kẹp các hộp đầu ra nếu chúng nằm ngoài [0, 1]. Nếu sai, không cắt và xuất ra tọa độ hộp như nó vốn có.

Lợi nhuận:

  • Output nmsed_boxes: A [batch_size, max_detices, 4] float32 tensor chứa các hộp không bị triệt tiêu tối đa.
  • Output nmsed_scores: A [batch_size, max_detilities] float32 tensor chứa điểm cho các hộp.
  • Output nmsed_classes: Một tensor [batch_size, max_detilities] float32 chứa các lớp cho các hộp.
  • Output xác định giá trị_của_đầu Output : Một tensor [batch_size] int32 cho biết số lượng các phát hiện hợp lệ cho mỗi mục lô. Chỉ các mục nhập num_detection [i] hàng đầu trong nms_boxes [i], nms_scores [i] và nms_class [i] là hợp lệ. Phần còn lại của các mục không có khoảng đệm.

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

CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)
CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, const CombinedNonMaxSuppression::Attrs & attrs)

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

nmsed_boxes
nmsed_classes
nmsed_scores
operation
valid_detections

Các chức năng tĩnh công khai

ClipBoxes (bool x)
PadPerClass (bool x)

Cấu trúc

tensorflow :: ops :: CombineNonMaxSuppression :: Attrs

Bộ thiết lập thuộc tính tùy chọn cho Kết hợpNonMaxSuppression .

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

nmsed_boxes

::tensorflow::Output nmsed_boxes

nmsed_classes

::tensorflow::Output nmsed_classes

nmsed_scores

::tensorflow::Output nmsed_scores

hoạt động

Operation operation

hợp_lý

::tensorflow::Output valid_detections

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

Kết hợpNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold
)

Kết hợpNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const CombinedNonMaxSuppression::Attrs & attrs
)

Các chức năng tĩnh công khai

ClipBoxes

Attrs ClipBoxes(
  bool x
)

PadPerClass

Attrs PadPerClass(
  bool x
)