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

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

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

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

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

算法原理

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

算法的作用

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

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

適用場(chǎng)景

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

算法的輸入

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

算法的輸出

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

細(xì)節(jié)

起始,設(shè)所有的框都沒有被抑制,所有框按照 score 從大到小排序。 從第 0 個(gè)框(分?jǐn)?shù)最高)開始遍歷:對(duì)于每一個(gè)框,如果該框沒有被抑制,就將所有與它 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 實(shí)現(xiàn)非極大值抑制算法(Non-maximum suppression, NMS)的詳細(xì)內(nèi)容,更多關(guān)于python 非極大值抑制算法的資料請(qǐng)關(guān)注好吧啦網(wǎng)其它相關(guān)文章!

標(biāo)簽: Python 編程
主站蜘蛛池模板: 国产精品午夜波多野结衣性色 | 一级美国片免费看 | 亚洲视频自拍 | 成年午夜性视频免费播放 | 国产一区二区三区在线观看免费 | 欧美色欧美亚洲高清在线视频 | 亚洲成人免费视频在线 | 国内精品久久精品 | 精品国产亚洲一区二区在线3d | 欧美在线观看免费一区视频 | 婷婷亚洲久悠悠色在线播放 | 美国一级毛片不卡无毒 | 亚洲一级视频在线观看 | 亚洲视频手机在线 | 一区二区精品在线观看 | 国产成人v爽在线免播放观看 | 久久毛片免费看一区二区三区 | 精品国产一二三区在线影院 | 久青草国产手机在线观 | 久久亚洲精品无码观看不卡 | 亚洲视频中文字幕在线观看 | 亚洲在线免费视频 | 国产精品成人在线 | 国产日韩欧美综合一区二区三区 | 日本一级大毛片a一 | 欧美啊啊啊 | 久久久久久久久久久视频国内精品视频 | 欧美国产精品亚洲精品第一区 | 综合亚洲一区二区三区 | fulidown国产精品合集 | 播放一级毛片 | 国产精选在线视频 | 亚洲欧美综合国产不卡 | 97青娱国产盛宴精品视频 | 大美女香蕉丽人视频网站 | 不卡一级毛片免费高清 | 欧美日韩精品高清一区二区 | 国产欧美一区二区三区在线 | 亚洲欧美日本视频 | 国产一级免费片 | 黄色免费看片网站 |