目標(biāo)檢測-YOLO系列

前言

yolo系列是目前應(yīng)用范圍比較廣的目標(biāo)的檢測算法,經(jīng)過幾輪的改進(jìn)昏滴,在實(shí)時性和檢測效果上同比第一代yolo也有了較大提升抛虫,本文梳理一下yolo系列算法的改進(jìn)思路懂缕。

YOLO

概述

在yolo之前侣集,目標(biāo)檢測更多的是two-stage算法键俱,即先篩選出部分候選框,在進(jìn)行選擇與分類世分,其中比較有代表性的網(wǎng)絡(luò)為:faster-RCNN编振,mask-RCNN等。
you only look once臭埋,yolo系列是眾多one-stage算法中比較有代表性的踪央,one- stage使用單一網(wǎng)絡(luò)完成定位與分類。

yolo框架思路

目標(biāo)檢測任務(wù)是定位出圖像中的物體并說明物體是什么瓢阴,yolo系列將其轉(zhuǎn)化為回歸問題畅蹂,輸入為圖片,輸出為x,y,w,h,c + 各分類概率

我們從v1 loss 入手看看它的設(shè)計(jì)思路:
image.png

loss分為三個大部分:
位置損失荣恐、置信度損失液斜、分類損失
位置損失比較容易理解,就是邊框x,y,w,h; w,h 取平方根用來平衡大小框?qū)p失影響大小的問題叠穆。
置信度指的是候選框內(nèi)為物體還是背景的置信度少漆,這里的C取值為0和1,圖像中背景區(qū)域一般來講遠(yuǎn)多于物體硼被,lamda用來平衡該差異對整體損失的影響示损。
第三部分為分類損失。
后面的YOLO都基本按照這個思路進(jìn)行優(yōu)化祷嘶。

YOLOV1

YOLO V1是YOLO的開山之作屎媳,它奠定了YOLO系列的基本框架。

基本思想

將圖片劃分為S*S個格子论巍,每個格子2個預(yù)選框烛谊,若該格子中心包含物體,則該物體由該格子進(jìn)行預(yù)測嘉汰。
預(yù)測產(chǎn)生 S*S*(B*5+C)
yolov1中S=7丹禀,B=2,C為數(shù)據(jù)集類別數(shù)量鞋怀。

網(wǎng)絡(luò)結(jié)構(gòu)

20200722170142957.png

網(wǎng)絡(luò)結(jié)構(gòu)相對簡單双泪, gooleNet作為卷積網(wǎng)絡(luò),兩層全聯(lián)接網(wǎng)絡(luò)進(jìn)行映射密似。

基本過程

輸入:448*448圖像
網(wǎng)絡(luò)預(yù)測邊框分類及其置信度
NMS去處冗余框

不足之處

1.網(wǎng)絡(luò)由于使用全連接網(wǎng)絡(luò)焙矛,輸入大小不可變。
2.預(yù)選框只有兩個残腌,對于密集重合以及小目標(biāo)效果不佳村斟。

YOLOV2

V2在V1的基礎(chǔ)上通過一些較有創(chuàng)造性的trick贫导,進(jìn)一步提升效果與速率。

V2優(yōu)化內(nèi)容

  1. 去除全連接層蟆盹,采用卷積與池化來進(jìn)行預(yù)測孩灯,該網(wǎng)絡(luò)結(jié)構(gòu)的改進(jìn)可以使用多尺度進(jìn)行訓(xùn)練。
  2. 去除了全連接層也不再使用dropout來防止過擬合逾滥,使用Batch Normalization來替代峰档。
  3. 優(yōu)化損失函數(shù),對位置的損失函數(shù)改用偏移量寨昙,更易收斂讥巡。
  4. 特征融合,除了使用最后一層特征圖外毅待,融合倒數(shù)第二層特征尚卫,改善V1對小目標(biāo)效果較差的問題归榕。
  5. 預(yù)選框使用K-means聚類得到尸红,同比V1提前指定預(yù)選框,該方式更貼近數(shù)據(jù)集刹泄,更合理外里。
  6. 更大分辨率,有效提升MAP特石。
  7. 引入聯(lián)合訓(xùn)練方式盅蝗,即檢測與分類聯(lián)合訓(xùn)練,訓(xùn)練分類時姆蘸,檢測部分的loss不生效墩莫,為0。

網(wǎng)絡(luò)結(jié)構(gòu)

yolov2.png

基本過程與細(xì)節(jié)

輸入416*416
卷積網(wǎng)絡(luò)
輸出13*13*5*25

細(xì)節(jié)一:輸入大小為32的奇數(shù)倍逞敷,保證最后框的大小為奇數(shù)狂秦,方便確定框的中心。
細(xì)節(jié)二: K-means聚類聚類距離為推捐,1-IOU裂问;規(guī)避大小框中心點(diǎn)距離差別大的問題。
細(xì)節(jié)三:優(yōu)化loss函數(shù)牛柒,使用相對偏移量計(jì)算損失堪簿,計(jì)算的是預(yù)測框與預(yù)選框的偏移(anchor box),只有最接近ground truth的檢測框進(jìn)行損失的計(jì)算皮壁。V5每個格子5個預(yù)選框椭更,同比V1召回明顯增加,準(zhǔn)確率下降蛾魄,map略有下降虑瀑。

image.png

YOLOV3

YOLOV3是極有標(biāo)志性的一代协饲,借鑒了ResNet的殘差思想,大幅度提升了時效性與效果缴川。

插曲佚事:V3是原YOLO作者的最后一代YOLO模型茉稠,因作者發(fā)現(xiàn)V3被用于軍事,從此退出了相關(guān)研究把夸。

網(wǎng)絡(luò)結(jié)構(gòu)

YOLOV3.jpeg

過程與細(xì)節(jié)

輸入圖像
卷積神經(jīng)網(wǎng)絡(luò)
多尺度預(yù)測

多尺度預(yù)測而线,
52*52 用于預(yù)測小目標(biāo)
26*26 用于預(yù)測中目標(biāo)
13*13 用于預(yù)測大目標(biāo)
低層融合高層特征,提高中小目標(biāo)的預(yù)測能力和預(yù)測效果
高層采用上采樣恋日,與中小兩層進(jìn)行拼接融合預(yù)測

取消池化層膀篮,使用卷積控制步長來進(jìn)行下采樣。
改進(jìn)loss岂膳,使得yolo可以進(jìn)行多標(biāo)簽預(yù)測誓竿。


image.png
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市谈截,隨后出現(xiàn)的幾起案子筷屡,更是在濱河造成了極大的恐慌,老刑警劉巖簸喂,帶你破解...
    沈念sama閱讀 219,589評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件毙死,死亡現(xiàn)場離奇詭異,居然都是意外死亡喻鳄,警方通過查閱死者的電腦和手機(jī)扼倘,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,615評論 3 396
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來除呵,“玉大人再菊,你說我怎么就攤上這事⊙赵” “怎么了纠拔?”我有些...
    開封第一講書人閱讀 165,933評論 0 356
  • 文/不壞的土叔 我叫張陵,是天一觀的道長泛啸。 經(jīng)常有香客問我绿语,道長,這世上最難降的妖魔是什么候址? 我笑而不...
    開封第一講書人閱讀 58,976評論 1 295
  • 正文 為了忘掉前任吕粹,我火速辦了婚禮,結(jié)果婚禮上岗仑,老公的妹妹穿的比我還像新娘匹耕。我一直安慰自己,他們只是感情好荠雕,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,999評論 6 393
  • 文/花漫 我一把揭開白布稳其。 她就那樣靜靜地躺著驶赏,像睡著了一般。 火紅的嫁衣襯著肌膚如雪既鞠。 梳的紋絲不亂的頭發(fā)上煤傍,一...
    開封第一講書人閱讀 51,775評論 1 307
  • 那天,我揣著相機(jī)與錄音嘱蛋,去河邊找鬼蚯姆。 笑死,一個胖子當(dāng)著我的面吹牛洒敏,可吹牛的內(nèi)容都是我干的龄恋。 我是一名探鬼主播,決...
    沈念sama閱讀 40,474評論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼凶伙,長吁一口氣:“原來是場噩夢啊……” “哼郭毕!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起函荣,我...
    開封第一講書人閱讀 39,359評論 0 276
  • 序言:老撾萬榮一對情侶失蹤显押,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后偏竟,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體煮落,經(jīng)...
    沈念sama閱讀 45,854評論 1 317
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡敞峭,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,007評論 3 338
  • 正文 我和宋清朗相戀三年踊谋,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片旋讹。...
    茶點(diǎn)故事閱讀 40,146評論 1 351
  • 序言:一個原本活蹦亂跳的男人離奇死亡殖蚕,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出沉迹,到底是詐尸還是另有隱情睦疫,我是刑警寧澤,帶...
    沈念sama閱讀 35,826評論 5 346
  • 正文 年R本政府宣布鞭呕,位于F島的核電站蛤育,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏葫松。R本人自食惡果不足惜瓦糕,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,484評論 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望腋么。 院中可真熱鬧咕娄,春花似錦、人聲如沸珊擂。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,029評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至圣贸,卻和暖如春挚歧,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背吁峻。 一陣腳步聲響...
    開封第一講書人閱讀 33,153評論 1 272
  • 我被黑心中介騙來泰國打工昼激, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人锡搜。 一個月前我還...
    沈念sama閱讀 48,420評論 3 373
  • 正文 我出身青樓橙困,卻偏偏與公主長得像,于是被迫代替她去往敵國和親耕餐。 傳聞我的和親對象是個殘疾皇子凡傅,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,107評論 2 356

推薦閱讀更多精彩內(nèi)容