ColorNet

2019年,清华大学团队写了一篇探究颜色空间对图像分类重要性的文章colorNet

paper《ColorNet: Investigating the importance of color spaces for image classification》。

目录

文章结论

网络结构

复现关键点

复现结果


文章结论

  1. 在图像分类领域,使用不同的颜色空间(RGB/HSV等),分类效果具有显著差异;
  2. 不同的颜色空间对于分类中的部分类别有独立性。例如,一颜色空间对A类的分类效果好,另一颜色空间对A类的分类效果差;
  3.  基于上述2中的独立性,可以选择集成(融合)多个颜色空间,可以明显提升分类效果。

网络结构

每个颜色空间使用一个DenseNet提取特征,得到类别向量(本人复现时,未作softmax),把所有的类别向量输出到一个Dense Layer,学习类别向量的权重,最后给出综合的类别向量。

图1 网络结构

复现关键点

论文的融合过程特别重要。本人理解如下。以10分类为例。

  • 设每个DenseNet的输出为o_i \in R^{10}, w_i \in R^{10},i=1,2,3,\dots,nn为颜色空间数量;
  • 融合的输出为\small z_i = \sum w_i.*o_i

复现结果

本人在cifar-10数据集复现论文的效果,使用的基本网络为resnet18,使用了3个颜色空间RGB/HSV/LAB。

表1 论文结果
颜色空间(组合)测试集准确率
RGB0.8396
HSV0.7842
LAB0.8028
RGB+HSV+LAB0.8594

上述结果体现了之前的结论。

 

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值