论文标题

学习结构几种颜色的图像

Learning to Structure an Image with Few Colors

论文作者

Hou, Yunzhong, Zheng, Liang, Gould, Stephen

论文摘要

颜色和结构是构建图像的两个支柱。通常,该结构通过丰富的颜色很好地表达,从而使图像中的对象可以被神经网络识别。但是,在颜色空间的极端局限性下,结构趋于消失,因此神经网络可能无法理解图像。对探索颜色和结构之间的这种相互作用感兴趣,我们研究了识别和保留最有用的图像结构的科学问题,同时将颜色空间限制为仅几个位,以便可以以很高的精度来识别所得的图像。为此,我们提出了一个颜色量化网络ColorCNN,该网络学会以端到端的方式从分类损失中构造图像。给定颜色空间尺寸,Colorcnn通过生成颜色索引图和RGB调色板来量化原始图像中的颜色。然后,将此颜色量化的图像馈送到预先训练的任务网络中以评估其性能。在我们的实验中,只有1位色彩空间(即两种颜色),提出的网络在CIFAR10数据集上实现了82.1%的TOP-1精度,以大幅度的差距超过传统的颜色量化方法。对于应用,当用PNG编码时,提出的颜色量化显示出优于比特率制度中其他图像压缩方法的优越性。该代码可在以下网址提供:https://github.com/hou-yz/color_distillation。

Color and structure are the two pillars that construct an image. Usually, the structure is well expressed through a rich spectrum of colors, allowing objects in an image to be recognized by neural networks. However, under extreme limitations of color space, the structure tends to vanish, and thus a neural network might fail to understand the image. Interested in exploring this interplay between color and structure, we study the scientific problem of identifying and preserving the most informative image structures while constraining the color space to just a few bits, such that the resulting image can be recognized with possibly high accuracy. To this end, we propose a color quantization network, ColorCNN, which learns to structure the images from the classification loss in an end-to-end manner. Given a color space size, ColorCNN quantizes colors in the original image by generating a color index map and an RGB color palette. Then, this color-quantized image is fed to a pre-trained task network to evaluate its performance. In our experiment, with only a 1-bit color space (i.e., two colors), the proposed network achieves 82.1% top-1 accuracy on the CIFAR10 dataset, outperforming traditional color quantization methods by a large margin. For applications, when encoded with PNG, the proposed color quantization shows superiority over other image compression methods in the extremely low bit-rate regime. The code is available at: https://github.com/hou-yz/color_distillation.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源