op นี้สร้างขอบเขตความสนใจจากขอบเขตที่กำหนด (bbox_deltas) ที่เข้ารหัสจุดยึด wrt ตาม eq.2 ใน arXiv:1506.01497
op เลือกกล่องให้คะแนน `pre_nms_topn` ระดับบนสุด ถอดรหัสตามจุดยึด ใช้การปราบปรามที่ไม่สูงสุดบนกล่องที่คาบเกี่ยวกันที่มีค่าจุดตัด-over-union (iou) ที่สูงกว่า `nms_threshold` ทิ้งกล่องที่มีด้านที่สั้นกว่าน้อยกว่า` min_size`. อินพุต: `คะแนน': เทนเซอร์ของรูปร่าง 4 มิติ [Batch, Height, Width, Num Anchors] ที่มีคะแนนต่อจุดยึดที่ตำแหน่งที่กำหนด `bbox_deltas`: เป็นเทนเซอร์ของรูปร่าง [Batch, Height, Width, 4 x Num Anchors] กล่องที่เข้ารหัส 'จุดยึด' ของสมอแต่ละอัน: เทนเซอร์รูปร่าง 1 มิติ [4 x Num Anchors] ซึ่งแสดงถึงจุดยึด ผลลัพธ์: `rois`: เอาต์พุต RoIs, เทนเซอร์รูปร่าง 3 มิติ [Batch, post_nms_topn, 4], เสริมด้วย 0 หากพบผู้สมัครน้อยกว่า post_nms_topn `roi_probabilities`: คะแนนความน่าจะเป็นของแต่ละ roi ใน 'rois', เมตริกซ์รูปร่าง 2 มิติ [Batch,post_nms_topn] เสริมด้วย 0 หากจำเป็น จัดเรียงตามคะแนน
คลาสที่ซ้อนกัน
| ระดับ | GenerateBoundingBoxProposals.Options | คุณลักษณะที่จำเป็นสำหรับการ GenerateBoundingBoxProposals | |
วิธีการสาธารณะ
| คง GenerateBoundingBoxProposals | สร้าง ( ขอบเขต ขอบเขต Operand <ลอย> คะแนน Operand <ลอย> bboxDeltas, Operand <ลอย> IMAGEINFO, Operand <ลอย> เบรก ตัวถูกดำเนินการ <ลอย> nmsThreshold, Operand <Integer> preNmsTopn, Operand <ลอย> MinSize, ตัวเลือก .. . ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ GenerateBoundingBoxProposals ใหม่ |
| คง GenerateBoundingBoxProposals.Options | postNmsTopn (Long postNmsTopn) |
| เอาท์พุท <ลอย> | roiProbabilities () เทนเซอร์รูปร่าง 2 มิติ `[num_images, post_nms_topn]` แทนคะแนนของขอบเขตพื้นที่ที่น่าสนใจในเมตริก 'rois' ที่ดัชนีเดียวกัน |
| เอาท์พุท <ลอย> | Rois () เทนเซอร์แบบลอยตัว 3 มิติของรูปร่าง `[num_images,post_nms_topn,4]` ซึ่งแสดงถึงขอบเขตของกล่องความสนใจที่เลือก |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง GenerateBoundingBoxProposals สร้าง ( ขอบเขต ขอบเขต Operand <ลอย> คะแนน Operand <ลอย> bboxDeltas, Operand <ลอย> IMAGEINFO, Operand <ลอย> เบรก ตัวถูกดำเนินการ <ลอย> nmsThreshold, Operand <Integer> preNmsTopn, Operand <ลอย> MinSize, ตัวเลือก ... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ GenerateBoundingBoxProposals ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| คะแนน | เมตริกซ์โฟลต 4 มิติของรูปร่าง `[num_images, height, width, num_achors]` ที่มีคะแนนของกล่องสำหรับจุดยึดที่กำหนด สามารถยกเลิกการจัดเรียงได้ |
| bboxDeltas | เมตริกซ์โฟลต 4 มิติของรูปร่าง `[num_images, height, width, 4 x num_anchors]` กล่องเข้ารหัสตามสมอแต่ละอัน พิกัดอยู่ในรูปแบบ [dy, dx, dh, dw] |
| ภาพข้อมูล | เทนเซอร์ลอยตัว 2 มิติของรูปร่าง `[num_images, 5]` ที่มีข้อมูลรูปภาพ ความสูง ความกว้าง มาตราส่วน |
| สมอ | เทนเซอร์รูปทรง 2 มิติ `[num_anchors, 4]` ที่อธิบายกล่องสมอ กล่องถูกจัดรูปแบบในรูปแบบ [y1, x1, y2, x2] |
| nmsThreshold | เทนเซอร์ลอยตัวแบบสเกลาร์สำหรับขีดจำกัดการกดทับที่ไม่สูงสุด |
| preNmsTopn | เมตริกซ์สเกลาร์สำหรับจำนวนกล่องคะแนนสูงสุดที่จะใช้เป็นอินพุต |
| ขนาดต่ำสุด | เทนเซอร์ลอยตัวแบบสเกลาร์ กล่องใดก็ตามที่มีขนาดที่เล็กกว่า min_size จะถูกละทิ้ง |
| ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ GenerateBoundingBoxProposals
สาธารณะคง GenerateBoundingBoxProposals.Options postNmsTopn (Long postNmsTopn)
พารามิเตอร์
| postNmsTopn | จำนวนเต็ม จำนวน rois สูงสุดในการส่งออก |
|---|
ประชาชน เอาท์พุท <ลอย> roiProbabilities ()
เทนเซอร์รูปร่าง 2 มิติ `[num_images, post_nms_topn]` แทนคะแนนของขอบเขตพื้นที่ที่น่าสนใจในเมตริก 'rois' ที่ดัชนีเดียวกัน
ประชาชน เอาท์พุท <ลอย> Rois ()
เทนเซอร์แบบลอยตัว 3 มิติของรูปร่าง `[num_images,post_nms_topn,4]` ซึ่งแสดงถึงขอบเขตของกล่องความสนใจที่เลือก เรียงลำดับจากมากไปน้อยในคะแนน