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 | |
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 <Float>, Toán hạng <Float> bboxDeltas, Toán hạng <Float> imageInfo, Neo toán hạng <Float>, Toán hạng <Float> nmsThreshold, Toán hạng <Integer> preNmsTopn, Toán hạng <Float> 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 <Float> | 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 <Float> | 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
Phương pháp công khai
public static generateBoundingBoxProposals tạo ( Phạm vi phạm vi , Điểm toán hạng <Float>, Toán hạng <Float> bboxDeltas, Toán hạng <Float> imageInfo, Neo toán hạng <Float>, Toán hạng <Float> nmsThreshold, Toán hạng <Integer> preNmsTopn, Toán hạng <Float> 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 <Float> 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 <Float> 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ố.