论文笔记 Finding Tiny Faces in the Wild with Generative Adversarial Network

文章来源:CVPR  2018

下载链接:Paper Download


解决的问题:

无约束条件下的低分辨率的人脸检测(即小人脸检测)


文章内容

目前人脸检测主要面临的挑战是:

 (1) 低分辨率的人脸缺乏详细的细节信息,以及图像模糊等问题。

 (2) 目前CNN模型卷积核步长太长,对于低分辨率人脸识别容易丢失大部分信息。

 本文算法的创新点:

(1) 提出一种新的人脸检测的统一端到端卷积神经网络结构,采用超分辨率和细化网络生成真实清晰的高分辨率图像,并引入判别网络对人脸与非人脸进行分类。

(2) 引入新的判别器loss。

 本文算法的核心思想:

本文方法主要是针对实际应用而提出的人脸检测算法,由于在无约束的环境下,人脸可能会以模糊、侧脸、低分辨率等方式呈现。针对该问题,作者通过采用生成对抗网络(GAN)直接从模糊的低分辨率人脸中生成清晰的高分辨率人脸,然后进行人脸检测。对于生成的超分辨率图像,可能是非人脸,因此作者在鉴别器网络中加入分类分支来判断生成图像是人脸还是非人脸,从而增强人脸检测算法的鲁棒性。下图为本文算法的网络结构图。

Figure 1. The pipeline of the proposed tiny face detector system.

对于本文的GAN网络,生成器包含两个子网络:超分辨率网络和细化网络;判别器中加入分支网络,用于区分人脸、非人脸和生成图片、真实图片。超分辨率网络(SRN)对小人脸进行小尺度上采样,在较大尺度上减小伪影,提高上采样图像质量。细化网络(network, RN)可以恢复上采样图像中缺失的一些细节,生成清晰的高分辨率图像进行分类。鉴别子网络利用一种新的损失函数,使鉴别器网络同时区分real/fake face 和 face/non-face(即区分是生成器合成的图像还是ground truth图像,以及是人脸还是非人脸)。该算法的损失函数主要包括三部分,分别为:

  (1) Pixel-wise loss:MSE loss,通过像素级的MSE损失来强制生成器的输出接近超分辨率的ground truth值,具体公式如图2所示。 

   (2) Adversarial loss:促进生成器生成更逼真的图片,具体公式如图2所示。 

    (3) Classification loss:其作用是为了使生成网络重构的图像更易于分类,同时抑制非人脸转变为人脸的过程,(如,将模糊桌子转变成人脸的过程),具体公式如图2所示 。

图2 本文算法所用的损失函数

图2中Classification loss存在一个问题,文章介绍yn=1表示人脸,而yn=0表示非人脸。那么,当yi=0时,log(yi-Dθ(IiHR))该如何取值,文中未对此进行说明(或许本人理解错误,此文仅供参考)。

最终将上图中三个损失函数相结合,得到最终的objective function,其公式如图3所示:

图3 objective function

图4中给出了本文GAN网络的结构设计表。其中Generator网络使用deep CNN, batch-normalization, ReLU,而Discriminator使用VGG19网络。

Figure 4. Architecture of the generator and discriminator network. “conv” represents a convolutional layer, “x8” denotes a residual block which has 8 convolutional layers, “de-conv” means a fractionally-stride convolutional layer, “2x” denotes up-sampling by a factor of 2, and “fc” indicates a fully connected layer.

实验结果

本文算法在WIDER FACE和FDDB两个公开的人脸检测数据集上进行训练以及测试。Trade-off weights α = 0.001 and β = 0.01。另外,图1中的MB-FCN detector是17年提出的一种人脸检测算法,知乎上有关于其的详细介绍,本文中仅使用它在数据集中随机截取人脸图片和非人脸图片,用于训练网络的输入。其中,训练使用的low-resolution images通过下采样生成,high-resolution images使用双三次插值生成。实验结果如下图所示。

Figure 5. Qualitative detection results of our proposed method. Green bounding boxes are ground truth annotations and red bounding boxes are the results from our method. Best seen on the computer, in color and zoomed in.

文中还对各个损失函数的性能进行了验证,即Ablation Studies, 其性能对比结果如下图。

Figure 6. Performance of the baseline model trained with and without GAN, refinement network, adversarial loss and classification loss on the WIDER FACE invalidation set.

本文方法与state-of-the-art方法对比的实验结果如下图。

Figure 7. On the WIDER FACE validation set, we compare our method with several state-of-the-art methods: MSCNN[31], MTTCNN[33], CMS-RCNN[37], HR[10], SSH[19], SFD[35]. The average precision (AP) is reported in the legend. Best viewed in color.

本文方法性能提升的原因

(1)本文方法主要是针对无约束、低分辨率的人脸进行检测,而这种情况下的人脸大部分是模糊不清晰的。既然这些人脸分辨率低,所以作者就通过所设计的网络(超分辨率网络SRN)将其分辨率变大,这样得到的图像可能是不清晰的,因此作者就通过细化网络将其变清晰,从而使生成的图像含有详细的细节信息,以便于人脸和非人脸的判别。基于这样的思路,网络均可将输入的低分辨率人脸合成高分辨率的人脸,同时也可以准确鉴别合成图像是否是人脸。

(2)本文方法在鉴别器网络中加入了分类分支,这可以抑制低分辨率的非人脸图像向人脸图像转化。GAN网络本来就是用来生成我们所期望输出的图像,这样鉴别器同样可以引导非人脸图像向人脸图像转化(如GAN网格可以将低分辨率模糊的桌子引导生成至清晰人脸),所以为了阻止该过程的发生,作者引入了Classification loss,当生成初步可判断的非人脸图像时,网络就可以直接阻止生成网络将其合成至人脸图像。


相关参考:Finding Tiny Faces in the Wild with Generative Adversarial NetworkFinding Tiny Faces in the Wild with Generative Adversarial Network 论文学习 


内容说明:

博客:论文笔记 | 低分辨率人脸识别:Finding Tiny Faces in the Wild with Generative Adversarial Network(AI_ISCAS)同为本人所写,两者之间所涉及的内容几乎相同。

上述内容仅个人的点滴粗见,如有不当之处,请同行批评指正。

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

推荐阅读更多精彩内容