Operasi ini menghasilkan Wilayah Kepentingan dari kotak pembatas yang diberikan (bbox_deltas) jangkar wrt yang disandikan 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 yang lebih tinggi dari `nms_threshold` persimpangan-over-union (iou), membuang kotak di mana sisi yang lebih pendek kurang dari ` ukuran_min`. Input: `scores`: Tensor bentuk 4D [Batch, Height, Width, Num Anchors] yang berisi skor per anchor pada posisi tertentu `bbox_deltas`: adalah tensor bentuk [Batch, Height, Width, 4 x Num Anchors] kotak yang dikodekan ke setiap jangkar `jangkar`: Tensor 1D berbentuk [4 x Num Jangkar], mewakili jangkar. Keluaran: `rois`: RoI keluaran, tensor 3D bentuk [Batch, post_nms_topn, 4], diisi oleh 0 jika kurang dari kandidat post_nms_topn yang ditemukan. `roi_probabilities`: skor probabilitas setiap roi dalam 'rois', tensor bentuk 2D [Batch,post_nms_topn], diisi dengan 0 jika diperlukan, diurutkan berdasarkan skor.
Kelas Bersarang
kelas | HasilkanBoundingBoxProposals.Options | Atribut opsional untuk GenerateBoundingBoxProposals |
Metode Publik
statis GenerateBoundingBoxProposals | membuat ( Lingkup lingkup, Operan <Lampung> skor, Operan <Lampung> bboxDeltas, Operan <Lampung> imageInfo, Operan <Lampung> jangkar, Operan <Lampung> nmsThreshold, Operan <Integer> preNmsTopn, Operan <Lampung> minSize, Pilihan .. . Pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi GenerateBoundingBoxProposals baru. |
statis GenerateBoundingBoxProposals.Options | postNmsTopn (Long postNmsTopn) |
Output <Lampung> | roiProbabilities () Tensor float 2-D berbentuk `[num_images, post_nms_topn]` yang mewakili skor wilayah kotak yang diinginkan dalam tensor `rois` pada indeks yang sama. |
Output <Lampung> | rois () Tensor float 3-D berbentuk `[num_images,post_nms_topn,4]` yang mewakili wilayah kotak yang diinginkan. |
Metode yang Diwarisi
Metode Publik
public static GenerateBoundingBoxProposals membuat ( Scope lingkup, Operan <Lampung> skor, Operan <Lampung> bboxDeltas, Operan <Lampung> imageInfo, Operan <Lampung> jangkar, Operan <Lampung> nmsThreshold, Operan <Integer> preNmsTopn, Operan <Lampung> minSize, Pilihan ... Pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi GenerateBoundingBoxProposals baru.
Parameter
cakupan | lingkup saat ini |
---|---|
skor | Tensor float 4-D berbentuk `[num_images, height, width, num_achors]` yang berisi skor kotak untuk jangkar tertentu, dapat disortir. |
bboxDelta | Tensor float 4-D berbentuk `[jumlah_gambar, tinggi, lebar, 4 x jumlah_jangkar]`. kotak penyandian dengan respek untuk 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 2D berbentuk `[num_anchors, 4]` yang menjelaskan kotak anchor. Kotak diformat dalam bentuk [y1, x1, y2, x2]. |
nmsThreshold | Tensor float skalar untuk ambang penekanan non-maksimal. |
praNmsTopn | Tensor int skalar untuk jumlah kotak dengan skor tertinggi yang akan digunakan sebagai input. |
ukuran min | Tensor float skalar. Kotak apa pun yang memiliki ukuran lebih kecil dari min_size akan dibuang. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari GenerateBoundingBoxProposals
public static GenerateBoundingBoxProposals.Options postNmsTopn (Long postNmsTopn)
Parameter
postNmsTopn | Sebuah bilangan bulat. Jumlah maksimum rois dalam output. |
---|
publik Keluaran <Lampung> roiProbabilities ()
Tensor float 2-D berbentuk `[num_images, post_nms_topn]` yang mewakili skor wilayah kotak yang diinginkan dalam tensor `rois` pada indeks yang sama.
publik Keluaran <Lampung> rois ()
Tensor float 3-D berbentuk `[num_images,post_nms_topn,4]` yang mewakili wilayah kotak yang diinginkan. Diurutkan dalam urutan menurun dalam skor.