GenerateBoundingBoxProposals

lớp cuối cùng công khai TạoBoundingBoxProposals

Hoạt động này tạo ra Vùng sở thích từ các hộp giới hạn (bbox_deltas) đã cho các neo wrt được mã hóa theo eq.2 trong arXiv:1506.01497

Cơ quan điều hành chọn các hộp tính điểm `pre_nms_topn` hàng đầu, giải mã chúng đối với các điểm neo, áp dụng tính năng triệt tiêu không tối đa trên các hộp chồng chéo có giá trị giao nhau (iou) cao hơn `nms_threshold`, loại bỏ các hộp có cạnh ngắn hơn ` kích thước tối thiểu`. Đầu vào: `scores`: Một tensor 4D có hình dạng [Batch, Height, Height, Num Anchors] chứa điểm trên mỗi neo tại vị trí nhất định `bbox_deltas`: là một tensor có hình dạng [Batch, Height, Height, 4 x Num Anchors] các hộp được mã hóa cho mỗi neo `neo`: Một tensor 1D có hình dạng [4 x Num Anchors], đại diện cho các neo. Đầu ra: `rois`: đầu ra RoIs, một tenxơ 3D có hình dạng [Batch, post_nms_topn, 4], được đệm bằng 0 nếu tìm thấy ít hơn ứng cử viên post_nms_topn. `roi_probabilities`: điểm xác suất của mỗi roi trong 'rois', một tensor 2D có hình dạng [Batch,post_nms_topn], được đệm bằng 0 nếu cần, được sắp xếp theo điểm.

Các lớp lồng nhau

lớp học TạoBoundingBoxProposals.Options Các thuộc tính tùy chọn cho GenerateBoundingBoxProposals

Hằng số

Sợi dây OP_NAME Tên của op này, được biết đến bởi công cụ lõi TensorFlow

Phương pháp công khai

Đề xuất generateBoundingBox tĩnh
tạo ( Phạm vi phạm vi , Điểm toán hạng < TFloat32 >, Toán hạng < TFloat32 > bboxDeltas, Toán hạng < TFloat32 > imageInfo, Toán hạng < TFloat32 > neo, Toán hạng < TFloat32 > nmsThreshold, Toán hạng < TInt32 > preNmsTopn, Toán hạng < TFloat32 > minSize, Tùy chọn.. . tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao bọc một hoạt động TạoBoundingBoxProposals mới.
tĩnh GeneratorBoundingBoxProposals.Options
postNmsTopn (Bài đăng dàiNmsTopn)
Đầu ra < TFloat32 >
roiXác suất ()
Một tenxơ nổi 2-D có hình dạng `[num_images, post_nms_topn]` đại diện cho điểm của vùng quan tâm trong tenxơ `rois` ở cùng một chỉ mục.
Đầu ra < TFloat32 >
roi ()
Một tenxơ nổi 3-D có hình dạng `[num_images,post_nms_topn,4]` đại diện cho vùng đã chọn của các hộp quan tâm.

Phương pháp kế thừa

Hằng số

Chuỗi cuối cùng tĩnh công khai OP_NAME

Tên của op này, được biết đến bởi công cụ lõi TensorFlow

Giá trị không đổi: "GenerateBoundingBoxProposals"

Phương pháp công khai

public static generateBoundingBoxProposals tạo ( Phạm vi phạm vi, Toán hạng < TFloat32 > điểm, Toán hạng < TFloat32 > bboxDeltas, Toán hạng < TFloat32 > imageInfo, Toán hạng < TFloat32 > neo, Toán hạng < TFloat32 > nmsThreshold, Toán hạng < TInt32 > preNmsTopn, Toán hạng < TFloat32 > minSize, Tùy chọn... tùy chọn)

Phương thức xuất xưởng để tạo một lớp bao bọc một hoạt động TạoBoundingBoxProposals mới.

Thông số
phạm vi phạm vi hiện tại
điểm số Một tenxơ nổi 4-D có hình dạng `[số_hình ảnh, chiều cao, chiều rộng, num_achors]` chứa điểm số của các hộp cho các điểm neo nhất định, có thể không được sắp xếp.
bboxDeltas Một tenxơ nổi 4-D có hình dạng `[số_hình ảnh, chiều cao, chiều rộng, 4 x số_anchors]`. hộp mã hóa tương ứng với từng điểm neo. Tọa độ được cho dưới dạng [dy, dx, dh, dw].
thông tin hình ảnh Một tenxơ nổi 2-D có hình dạng `[num_images, 5]` chứa thông tin hình ảnh Chiều cao, Chiều rộng, Tỷ lệ.
mỏ neo Một tensor nổi 2-D có hình dạng `[num_anchors, 4]` mô tả các hộp neo. Các hộp được định dạng theo dạng [y1, x1, y2, x2].
ngưỡng bước sóng Một tenxơ float vô hướng cho ngưỡng triệt tiêu không cực đại.
preNmsTopn Một tensor int vô hướng cho số lượng ô ghi điểm cao nhất sẽ được sử dụng làm đầu vào.
kích thước tối thiểu Một tenxơ phao vô hướng. Bất kỳ hộp nào có kích thước nhỏ hơn min_size sẽ bị loại bỏ.
tùy chọn mang các giá trị thuộc tính tùy chọn
Trả lại
  • một phiên bản mới của GeneratorBoundingBoxProposals

public static generateBoundingBoxProposals.Options postNmsTopn (Bài đăng dàiNmsTopn)

Thông số
bàiNmsTopn Một số nguyên. Số rois tối đa ở đầu ra.

public Output < TFloat32 > roiProbabilities ()

Một tenxơ nổi 2-D có hình dạng `[num_images, post_nms_topn]` đại diện cho điểm của vùng quan tâm trong tenxơ `rois` ở cùng một chỉ mục.

Đầu ra công khai < TFloat32 > rois ()

Một tenxơ nổi 3-D có hình dạng `[num_images,post_nms_topn,4]` đại diện cho vùng đã chọn của các hộp quan tâm. Sắp xếp theo thứ tự giảm dần về điểm số.