Task 7 FCN_3

1. 了解tensorflow的API

为了更好更高效地进行数据分析,我学习了如何使用tf.summary观察训练过程,用tf.records制作和读取训练集,以及用tf.dataset按批读取数据。

2. 训练心得

2.1. 背景

第一次做MIT Scene Parsing Benchmark的比赛数据集时,由于经验不足,栽了很多跟头。先是图像读取的代码总是出错,没法正确运行。

好不容易能跑起来,准确率又很低:像素准确率有50左右,但miou只有20。

由于任务要求用FCN-8s,优化不能改变网络模型,学习率又从1e-4一直调到了1e-6都没改善,最终确定优化的方向在于图像处理的方式

我主要研究了两个样例代码,分别是FCN.tensorflow以及semantic-segmentation-pytorch。总结出来图像处理的方式有以下几种。

2.2 图像的按批处理

图像的按批处理方式有3种:

  1. 一张一张处理
  2. 按一小批处理(SGD)
  3. 全部一起处理(不可能,因为图像集太大)

第一种方式的处理速度快,收敛也快,但因为每批只有一张图,参数会反复跳动,导致无法达到最佳值。

第二种方式处理稍慢,但收敛结果会比第一种好些,因为参数反复无常的次数会少些。最开始我参考的FCN.tensorflow的代码就是使用每批两张图的。

2.3 图像的读取处理

  1. 全部缩小到统一比例,比如224*224
  2. 全部放大到统一比例,只要能够被32整除(以便通过卷积和逆卷积后正确恢复图像比例)

第一种就是FCN.tensorflow的做法。它对所有图片都缩放到224 * 224(很多图片的尺寸都大于224 * 224),这样虽然简单快捷,但在处理较小物体时会表现得很差,因为缩小图片会丢失细节

第二种就是semantic-segmentation-pytorch的做法。它对每一批的图片都会放大到统一的尺寸,做法如下:

  1. 得到这一批图片的最大的长和宽,比如(512, 702)。

  2. 让最大长和最大宽都能被32整除,以便能顺利通过卷积层和逆卷积层,并正确保持图像比例,因为FCN-8s的卷积层全部通过时,图像连续5次缩小一半,相当于被缩小了2^5 = 32倍,如果尺寸不是32的倍数,在恢复时可能出现尺寸无法匹配的情况

    尺寸无法匹配的意思就是:比如某图片在最后一层卷积层前尺寸是(31, 31),经过卷积层(padding="SAME", strides=[2, 2])缩小变成(16, 16),再放大就变成(32, 32)了。由于skip-architecture,(31, 31)要与(32, 32)相加,就出现了尺寸不同的情况。

    所以,(512, 702)的尺寸中,512可以整除32,而702不能整除。经过计算得到比702大的最近能够整除32的数为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标记了每个像素的确切分类,是整数值。如果使用"bilinear"缩放,会使缩放结果出现很多本来没有的值。比如原本图像只有2,5两个分类,像素点的值都是2到5,由于"bilinear"缩放,使得填充的点出现了3点几,4点几的值,但原图片根本没有3和4这两个分类,而且3.几不是整数,不能表示一个分类。这就扭曲了图片原本的含义了。

2.4 图像处理的选用

最终我选择了按批处理,每批2张图的方式,对image使用"nearest"缩放,对annotation使用"bilinear"缩放,且尺寸统一放大,以便保留小物体的细节。

3. 指标理解:mean_iou

参考:深度学习中IU、IoU(Intersection over Union)的概念理解以及python程序实现
iou就是Intersection over Union,mean就是各个类别的平均iou。
比如,一个物体类别的实际区域和预测区域可能如下:

image.png

那么iou就是预测与实际的重叠部分 / 预测与实际的合并区域:


对每个类别,都如此计算iou,最后取平均即可。

4. 训练结果

对100张validation的图片进行了测试,并计算了4项指标:
以下四项分别是pixel_acc, mean_acc, mean_iou, weighted_iou:

这是部分的预测结果:






相关阅读

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,294评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,493评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,790评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,595评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,718评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,906评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,053评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,797评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,250评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,570评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,711评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,388评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,018评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,796评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,023评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,461评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,595评论 2 350

推荐阅读更多精彩内容