Task 7 FCN_3

1. 了解tensorflow的API

為了更好更高效地進行數(shù)據分析怕吴,我學習了如何使用tf.summary觀察訓練過程落塑,用tf.records制作和讀取訓練集坪创,以及用tf.dataset按批讀取數(shù)據褐望。

2. 訓練心得

2.1. 背景

第一次做MIT Scene Parsing Benchmark的比賽數(shù)據集時蜕琴,由于經驗不足注整,栽了很多跟頭能曾。先是圖像讀取的代碼總是出錯,沒法正確運行肿轨。

好不容易能跑起來寿冕,準確率又很低:像素準確率有50左右,但miou只有20椒袍。

由于任務要求用FCN-8s驼唱,優(yōu)化不能改變網絡模型,學習率又從1e-4一直調到了1e-6都沒改善驹暑,最終確定優(yōu)化的方向在于圖像處理的方式曙蒸。

我主要研究了兩個樣例代碼捌治,分別是FCN.tensorflow以及semantic-segmentation-pytorch∨撸總結出來圖像處理的方式有以下幾種肖油。

2.2 圖像的按批處理

圖像的按批處理方式有3種:

  1. 一張一張?zhí)幚?/li>
  2. 按一小批處理(SGD)
  3. 全部一起處理(不可能,因為圖像集太大)

第一種方式的處理速度快臂港,收斂也快森枪,但因為每批只有一張圖,參數(shù)會反復跳動审孽,導致無法達到最佳值县袱。

第二種方式處理稍慢,但收斂結果會比第一種好些佑力,因為參數(shù)反復無常的次數(shù)會少些式散。最開始我參考的FCN.tensorflow的代碼就是使用每批兩張圖的。

2.3 圖像的讀取處理

  1. 全部縮小到統(tǒng)一比例打颤,比如224*224
  2. 全部放大到統(tǒng)一比例暴拄,只要能夠被32整除(以便通過卷積和逆卷積后正確恢復圖像比例)

第一種就是FCN.tensorflow的做法。它對所有圖片都縮放到224 * 224(很多圖片的尺寸都大于224 * 224)编饺,這樣雖然簡單快捷乖篷,但在處理較小物體時會表現(xiàn)得很差,因為縮小圖片會丟失細節(jié)透且。

第二種就是semantic-segmentation-pytorch的做法撕蔼。它對每一批的圖片都會放大到統(tǒng)一的尺寸,做法如下:

  1. 得到這一批圖片的最大的長和寬秽誊,比如(512鲸沮, 702)。

  2. 讓最大長和最大寬都能被32整除锅论,以便能順利通過卷積層和逆卷積層讼溺,并正確保持圖像比例,因為FCN-8s的卷積層全部通過時棍厌,圖像連續(xù)5次縮小一半肾胯,相當于被縮小了2^5 = 32倍,如果尺寸不是32的倍數(shù)耘纱,在恢復時可能出現(xiàn)尺寸無法匹配的情況敬肚。

    尺寸無法匹配的意思就是:比如某圖片在最后一層卷積層前尺寸是(31, 31)束析,經過卷積層(padding="SAME"艳馒, strides=[2, 2])縮小變成(16, 16),再放大就變成(32弄慰, 32)了第美。由于skip-architecture,(31陆爽, 31)要與(32什往, 32)相加,就出現(xiàn)了尺寸不同的情況慌闭。

    所以别威,(512, 702)的尺寸中驴剔,512可以整除32省古,而702不能整除。經過計算得到比702大的最近能夠整除32的數(shù)為704丧失,所以最終尺寸確定為(512豺妓, 704)

2.4 圖像的縮放處理

圖像的縮放方式很關鍵,自己寫的時候因為這里疏忽布讹,導致訓練結果基本作廢琳拭,那就是:對于annotation的縮放方式必須使用"nearest",對image的縮放則應該使用"bilinear"(別的也可以)炒事。

代碼中也就是這兩句

image = misc.imresize(image, [resize_height, resize_width], interp="bilinear")
annotation = misc.imresize(annotation, [resize_height, resize_width], interp="nearest")

為什么annotation的縮放方式必須使用"nearest"臀栈?因為annotation標記了每個像素的確切分類蔫慧,是整數(shù)值挠乳。如果使用"bilinear"縮放,會使縮放結果出現(xiàn)很多本來沒有的值姑躲。比如原本圖像只有2睡扬,5兩個分類,像素點的值都是2到5黍析,由于"bilinear"縮放卖怜,使得填充的點出現(xiàn)了3點幾,4點幾的值阐枣,但原圖片根本沒有3和4這兩個分類马靠,而且3.幾不是整數(shù),不能表示一個分類蔼两。這就扭曲了圖片原本的含義了甩鳄。

2.4 圖像處理的選用

最終我選擇了按批處理,每批2張圖的方式额划,對image使用"nearest"縮放妙啃,對annotation使用"bilinear"縮放,且尺寸統(tǒng)一放大,以便保留小物體的細節(jié)揖赴。

3. 指標理解:mean_iou

參考:深度學習中IU馆匿、IoU(Intersection over Union)的概念理解以及python程序實現(xiàn)
iou就是Intersection over Union,mean就是各個類別的平均iou燥滑。
比如渐北,一個物體類別的實際區(qū)域和預測區(qū)域可能如下:

image.png

那么iou就是預測與實際的重疊部分 / 預測與實際的合并區(qū)域:


對每個類別,都如此計算iou铭拧,最后取平均即可腔稀。

4. 訓練結果

對100張validation的圖片進行了測試,并計算了4項指標:
以下四項分別是pixel_acc, mean_acc, mean_iou, weighted_iou:

這是部分的預測結果:






相關閱讀

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末羽历,一起剝皮案震驚了整個濱河市焊虏,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌秕磷,老刑警劉巖诵闭,帶你破解...
    沈念sama閱讀 217,084評論 6 503
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異澎嚣,居然都是意外死亡疏尿,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,623評論 3 392
  • 文/潘曉璐 我一進店門易桃,熙熙樓的掌柜王于貴愁眉苦臉地迎上來褥琐,“玉大人,你說我怎么就攤上這事晤郑〉谐剩” “怎么了?”我有些...
    開封第一講書人閱讀 163,450評論 0 353
  • 文/不壞的土叔 我叫張陵造寝,是天一觀的道長磕洪。 經常有香客問我,道長诫龙,這世上最難降的妖魔是什么析显? 我笑而不...
    開封第一講書人閱讀 58,322評論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮签赃,結果婚禮上谷异,老公的妹妹穿的比我還像新娘。我一直安慰自己锦聊,他們只是感情好歹嘹,可當我...
    茶點故事閱讀 67,370評論 6 390
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著括丁,像睡著了一般荞下。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,274評論 1 300
  • 那天尖昏,我揣著相機與錄音仰税,去河邊找鬼。 笑死抽诉,一個胖子當著我的面吹牛陨簇,可吹牛的內容都是我干的。 我是一名探鬼主播迹淌,決...
    沈念sama閱讀 40,126評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼河绽,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了唉窃?” 一聲冷哼從身側響起耙饰,我...
    開封第一講書人閱讀 38,980評論 0 275
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎纹份,沒想到半個月后苟跪,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經...
    沈念sama閱讀 45,414評論 1 313
  • 正文 獨居荒郊野嶺守林人離奇死亡蔓涧,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 37,599評論 3 334
  • 正文 我和宋清朗相戀三年件已,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片元暴。...
    茶點故事閱讀 39,773評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡篷扩,死狀恐怖,靈堂內的尸體忽然破棺而出茉盏,到底是詐尸還是另有隱情鉴未,我是刑警寧澤,帶...
    沈念sama閱讀 35,470評論 5 344
  • 正文 年R本政府宣布援岩,位于F島的核電站歼狼,受9級特大地震影響掏导,放射性物質發(fā)生泄漏享怀。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,080評論 3 327
  • 文/蒙蒙 一趟咆、第九天 我趴在偏房一處隱蔽的房頂上張望添瓷。 院中可真熱鬧,春花似錦值纱、人聲如沸鳞贷。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,713評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽搀愧。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間咱筛,已是汗流浹背搓幌。 一陣腳步聲響...
    開封第一講書人閱讀 32,852評論 1 269
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留迅箩,地道東北人溉愁。 一個月前我還...
    沈念sama閱讀 47,865評論 2 370
  • 正文 我出身青樓,卻偏偏與公主長得像饲趋,于是被迫代替她去往敵國和親拐揭。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 44,689評論 2 354

推薦閱讀更多精彩內容