国产成人精品久久免费动漫-国产成人精品天堂-国产成人精品区在线观看-国产成人精品日本-a级毛片无码免费真人-a级毛片毛片免费观看久潮喷

您的位置:首頁技術(shù)文章
文章詳情頁

python 實現(xiàn)非極大值抑制算法(Non-maximum suppression, NMS)

瀏覽:11日期:2022-07-08 11:26:28

NMS 算法在目標(biāo)檢測,目標(biāo)定位領(lǐng)域有較廣泛的應(yīng)用。

算法原理

非極大值抑制算法(Non-maximum suppression, NMS)的本質(zhì)是搜索局部極大值,抑制非極大值元素。

算法的作用

當(dāng)算法對一個目標(biāo)產(chǎn)生了多個候選框的時候,選擇 score 最高的框,并抑制其他對于改目標(biāo)的候選框

python 實現(xiàn)非極大值抑制算法(Non-maximum suppression, NMS)

適用場景

一幅圖中有多個目標(biāo)(如果只有一個目標(biāo),那么直接取 score 最高的候選框即可)。

算法的輸入

算法對一幅圖產(chǎn)生的所有的候選框,以及每個框?qū)?yīng)的 score (可以用一個 5 維數(shù)組 dets 表示,前 4 維表示四個角的坐標(biāo),第 5 維表示分?jǐn)?shù)),閾值 thresh。

算法的輸出

正確的候選框組(dets 的一個子集)。

細(xì)節(jié)

起始,設(shè)所有的框都沒有被抑制,所有框按照 score 從大到小排序。 從第 0 個框(分?jǐn)?shù)最高)開始遍歷:對于每一個框,如果該框沒有被抑制,就將所有與它 IoU 大于 thresh 的框設(shè)為抑制。 返回沒被抑制的框。

參考代碼

# --------------------------------------------------------# Fast R-CNN# Copyright (c) 2015 Microsoft# Licensed under The MIT License [see LICENSE for details]# Written by Ross Girshick# --------------------------------------------------------import numpy as npcimport numpy as npcdef inline np.float32_t max(np.float32_t a, np.float32_t b): return a if a >= b else bcdef inline np.float32_t min(np.float32_t a, np.float32_t b): return a if a <= b else bdef cpu_nms(np.ndarray[np.float32_t, ndim=2] dets, np.float thresh): cdef np.ndarray[np.float32_t, ndim=1] x1 = dets[:, 0] cdef np.ndarray[np.float32_t, ndim=1] y1 = dets[:, 1] cdef np.ndarray[np.float32_t, ndim=1] x2 = dets[:, 2] cdef np.ndarray[np.float32_t, ndim=1] y2 = dets[:, 3] cdef np.ndarray[np.float32_t, ndim=1] scores = dets[:, 4] cdef np.ndarray[np.float32_t, ndim=1] areas = (x2 - x1 + 1) * (y2 - y1 + 1) cdef np.ndarray[np.int_t, ndim=1] order = scores.argsort()[::-1] cdef int ndets = dets.shape[0] cdef np.ndarray[np.int_t, ndim=1] suppressed = np.zeros((ndets), dtype=np.int) # nominal indices cdef int _i, _j # sorted indices cdef int i, j # temp variables for box i’s (the box currently under consideration) cdef np.float32_t ix1, iy1, ix2, iy2, iarea # variables for computing overlap with box j (lower scoring box) cdef np.float32_t xx1, yy1, xx2, yy2 cdef np.float32_t w, h cdef np.float32_t inter, ovr keep = [] for _i in range(ndets): i = order[_i] if suppressed[i] == 1: continue keep.append(i) ix1 = x1[i] iy1 = y1[i] ix2 = x2[i] iy2 = y2[i] iarea = areas[i] for _j in range(_i + 1, ndets): j = order[_j] if suppressed[j] == 1:continue xx1 = max(ix1, x1[j]) yy1 = max(iy1, y1[j]) xx2 = min(ix2, x2[j]) yy2 = min(iy2, y2[j]) w = max(0.0, xx2 - xx1 + 1) h = max(0.0, yy2 - yy1 + 1) inter = w * h ovr = inter / (iarea + areas[j] - inter) if ovr >= thresh:suppressed[j] = 1 return keep

以上就是python 實現(xiàn)非極大值抑制算法(Non-maximum suppression, NMS)的詳細(xì)內(nèi)容,更多關(guān)于python 非極大值抑制算法的資料請關(guān)注好吧啦網(wǎng)其它相關(guān)文章!

標(biāo)簽: Python 編程
主站蜘蛛池模板: 亚洲在线看片 | 国产理论视频在线观看 | 欧美视频亚洲视频 | 亚洲在线网址 | 在线欧美一级毛片免费观看 | 久久在线免费视频 | 免费大片黄手机在线观看 | 黄色视屏免费 | 国产理论视频 | 色综合久久加勒比高清88 | 久久精品无遮挡一级毛片 | 精品中文字幕在线 | 手机在线精品视频 | 一级成人黄色片 | 国产成人精品综合在线 | 国产草草影院 | 午夜国产亚洲精品一区 | 欧美色性 | 国内自拍网址 | 国产微拍精品福利视频 | 女高中生被cao到哭视频 | 97视频在线免费观看 | 在线免费一区二区 | 国产系列在线观看 | 国产成人99精品免费观看 | 精品久久在线观看 | 国产黄网 | 午夜三级a三级三点在线观看 | 久久精品国产午夜伦班片 | 在线观看日本视频免费 | 精品亚洲欧美高清不卡高清 | 亚洲欧美在线视频免费 | 国产香蕉国产精品偷在线观看 | 黄录像欧美片在线观看 | 91最新地址永久入口 | 亚洲加勒比久久88色综合一区 | 美女扒开腿让男人桶爽免费动态图 | 国产在线观看一区精品 | a级片免费观看 | 色综合亚洲七七久久桃花影院 | v欧美精品v日本精品 |