SSD代碼詳解(pytorch)

Prior Box生成

SSD中引入了prior box, 其實和anchor類似分蓖,就是一些目標的預選框湖苞;后續(xù)通過classification loss 和 bounding box regression loss確定真實的目標位置巴帮,SSD按照如下規(guī)則生成prior box:

  1. prior box生成由scale和aspect ratio決定邢享,其中每個特征圖的尺度計算公式如下:
    S_k = S_{min} + \frac{S_{max} - S_{min}}{m-1}*(k-1), k\in[1,m]
    在論文SSD300網(wǎng)絡中侨赡,其中的給出的minsizemaxsize與公式并不完全匹配摇庙,因為論文給第一層conv4_3設置了默認的scale, 為0.1。其余5層滿足上述公式計算絮姆,其中第二層至第五層scale分別為:0.2醉冤, 0.37, 0.54滚朵, 0.71冤灾, 0.88, 1.05辕近。
  2. 以feature map上每個點為中心韵吨,生成矩形prior box,其中中心點部分偏移(offset=0.5) 移宅。
  3. SSD提供兩個默認的正方形归粉,邊長分別為minsize\sqrt{minsize*maxsize}, SSD論文中默認的minsizemaxsize分別取0.2, 0.9。
  4. SSD會生成兩個長方形漏峰,采用的長寬比 aspect ratio

Multi loss function

SSD損失函數(shù)包含兩部分糠悼,一個是分類損失L_{loc},一個是定位損失L_{conf}, 整個損失函數(shù)表示如下:
L(x,c,l,g) =\frac{1}{N}*(L_{conf}(x,c)+L_{loc}(x,l,g))
其中N是先驗框的正樣本數(shù)量浅乔,c是類別置信度預測值倔喂,l是先驗框對應的邊界框預測值铝条,g是ground truth的位置參數(shù),x代表網(wǎng)絡的預測值席噩。
對于位置損失班缰,采用smooth L1 Loss, 位置信息都是encode后的值;
對于分類損失悼枢,首先需要使用hard negative mining將正負樣本按照1:3的比例把負樣本抽樣出來埠忘,抽樣的方法是:針對所有batch的confidence,按照置信度誤差進行降序排列馒索,取出前top_k個負樣本莹妒。

?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市绰上,隨后出現(xiàn)的幾起案子旨怠,更是在濱河造成了極大的恐慌,老刑警劉巖渔期,帶你破解...
    沈念sama閱讀 210,914評論 6 490
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件运吓,死亡現(xiàn)場離奇詭異,居然都是意外死亡疯趟,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 89,935評論 2 383
  • 文/潘曉璐 我一進店門谋梭,熙熙樓的掌柜王于貴愁眉苦臉地迎上來信峻,“玉大人,你說我怎么就攤上這事瓮床№镂瑁” “怎么了?”我有些...
    開封第一講書人閱讀 156,531評論 0 345
  • 文/不壞的土叔 我叫張陵隘庄,是天一觀的道長踢步。 經常有香客問我,道長丑掺,這世上最難降的妖魔是什么获印? 我笑而不...
    開封第一講書人閱讀 56,309評論 1 282
  • 正文 為了忘掉前任,我火速辦了婚禮街州,結果婚禮上兼丰,老公的妹妹穿的比我還像新娘。我一直安慰自己唆缴,他們只是感情好鳍征,可當我...
    茶點故事閱讀 65,381評論 5 384
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著面徽,像睡著了一般艳丛。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,730評論 1 289
  • 那天氮双,我揣著相機與錄音碰酝,去河邊找鬼。 笑死眶蕉,一個胖子當著我的面吹牛砰粹,可吹牛的內容都是我干的。 我是一名探鬼主播造挽,決...
    沈念sama閱讀 38,882評論 3 404
  • 文/蒼蘭香墨 我猛地睜開眼碱璃,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了饭入?” 一聲冷哼從身側響起嵌器,我...
    開封第一講書人閱讀 37,643評論 0 266
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎谐丢,沒想到半個月后爽航,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經...
    沈念sama閱讀 44,095評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡乾忱,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 36,448評論 2 325
  • 正文 我和宋清朗相戀三年讥珍,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片窄瘟。...
    茶點故事閱讀 38,566評論 1 339
  • 序言:一個原本活蹦亂跳的男人離奇死亡衷佃,死狀恐怖,靈堂內的尸體忽然破棺而出蹄葱,到底是詐尸還是另有隱情氏义,我是刑警寧澤,帶...
    沈念sama閱讀 34,253評論 4 328
  • 正文 年R本政府宣布图云,位于F島的核電站惯悠,受9級特大地震影響,放射性物質發(fā)生泄漏竣况。R本人自食惡果不足惜克婶,卻給世界環(huán)境...
    茶點故事閱讀 39,829評論 3 312
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望帕翻。 院中可真熱鬧鸠补,春花似錦、人聲如沸嘀掸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,715評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽睬塌。三九已至泉蝌,卻和暖如春歇万,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 31,945評論 1 264
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留寞宫,地道東北人串绩。 一個月前我還...
    沈念sama閱讀 46,248評論 2 360
  • 正文 我出身青樓垃瞧,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 43,440評論 2 348

推薦閱讀更多精彩內容