Operasi ini menghasilkan Wilayah Minat dari kotak pembatas tertentu (bbox_deltas) jangkar wrt yang dikodekan sesuai dengan persamaan 2 di arXiv:1506.01497
Operasi memilih kotak penilaian `pre_nms_topn` teratas, mendekodekannya sehubungan dengan jangkar, menerapkan penekanan non-maksimal pada kotak yang tumpang tindih dengan nilai interseksi-over-union (iou) yang lebih tinggi dari `nms_threshold`, membuang kotak yang sisinya lebih pendek kurang dari ` ukuran_minimal`. Masukan: `scores`: Tensor bentuk 4D [Batch, Height, Wide, Num Anchors] yang berisi skor per jangkar pada posisi tertentu `bbox_deltas`: adalah tensor bentuk [Batch, Height, Wide, 4 x Num Anchors] kotak yang dikodekan ke setiap `jangkar` jangkar: Bentuk tensor 1D [4 x Num Jangkar], mewakili jangkar. Keluaran: `rois`: RoI keluaran, tensor 3D berbentuk [Batch, post_nms_topn, 4], diisi dengan 0 jika kandidat yang ditemukan kurang dari post_nms_topn. `roi_probabilities`: skor probabilitas setiap roi dalam 'rois', tensor 2D berbentuk [Batch,post_nms_topn], diisi dengan 0 jika diperlukan, diurutkan berdasarkan skor.
Kelas Bersarang
| kelas | HasilkanBoundingBoxProposals.Options | Atribut opsional untuk GenerateBoundingBoxProposals | |
Metode Publik
| statis GenerateBoundingBoxProposal | buat ( Lingkup lingkup, skor Operand <Float>, Operand <Float> bboxDeltas, Operand <Float> imageInfo, Operand <Float> jangkar, Operand <Float> nmsThreshold, Operand <Integer> preNmsTopn, Operand <Float> minSize, Options.. . pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi GenerateBoundingBoxProposals baru. |
| statis GenerateBoundingBoxProposals.Options | postNmsTopn (Posting panjangNmsTopn) |
| Keluaran <Mengambang> | roiProbabilitas () Tensor float 2-D berbentuk `[num_images, post_nms_topn]` yang mewakili skor kotak wilayah yang diminati dalam tensor `rois` pada indeks yang sama. |
| Keluaran <Mengambang> | rois () Tensor float 3-D berbentuk `[num_images,post_nms_topn,4]` yang mewakili kotak wilayah minat yang dipilih. |
Metode Warisan
Metode Publik
public static GenerateBoundingBoxProposals membuat ( Lingkup cakupan, skor Operand <Float>, Operand <Float> bboxDeltas, Operand <Float> imageInfo, Operand <Float> jangkar, Operand <Float> nmsThreshold, Operand <Integer> preNmsTopn, Operand <Float> minSize, Pilihan... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi GenerateBoundingBoxProposals baru.
Parameter
| cakupan | ruang lingkup saat ini |
|---|---|
| skor | Tensor float 4-D dengan bentuk `[num_images, height, width, num_achors]` yang berisi sejumlah kotak untuk jangkar tertentu, dapat tidak disortir. |
| bboxDelta | Tensor float 4-D dengan bentuk `[num_images, height, width, 4 x num_anchors]`. kotak pengkodean sehubungan dengan setiap jangkar. Koordinat diberikan dalam bentuk [dy, dx, dh, dw]. |
| info gambar | Tensor float 2-D berbentuk `[num_images, 5]` yang berisi informasi gambar Tinggi, Lebar, Skala. |
| jangkar | Tensor float 2-D dengan bentuk `[num_anchors, 4]` yang menjelaskan kotak jangkar. Kotak diformat dalam bentuk [y1, x1, y2, x2]. |
| nmsThreshold | Tensor float skalar untuk ambang penekanan non-maksimal. |
| sebelumNmsTopn | Tensor int skalar untuk jumlah kotak skor tertinggi yang akan digunakan sebagai masukan. |
| ukuran minimal | Tensor pelampung skalar. Kotak apa pun yang ukurannya lebih kecil dari min_size akan dibuang. |
| pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari GenerateBoundingBoxProposals
public static GenerateBoundingBoxProposals.Options postNmsTopn (Posting panjangNmsTopn)
Parameter
| postingNmsTopn | Sebuah bilangan bulat. Jumlah rois maksimum dalam output. |
|---|
Output publik <Float> roiProbabilities ()
Tensor float 2-D berbentuk `[num_images, post_nms_topn]` yang mewakili skor kotak wilayah yang diminati dalam tensor `rois` pada indeks yang sama.
Keluaran publik <Float> rois ()
Tensor float 3-D berbentuk `[num_images,post_nms_topn,4]` yang mewakili kotak wilayah minat yang dipilih. Diurutkan dalam urutan skor.