使用卷积神经网络(CNN)时如何忽略纯色图像背景?

时间:2015-09-20 13:28:34

标签: computer-vision neural-network deep-learning caffe conv-neural-network

给定具有已知背景颜色的对象的图像,如何影响CNN忽略/折扣背景的特征,从而强调对象?

为了您的信息,我的场景是基于内容的图像检索(CBIR)的特征提取(例如神经代码)。我正在使用Caffe。

1 个答案:

答案 0 :(得分:1)

我能想到的一个选项是为每个图像创建两个输入:第一个是3xHxW彩色图像,另一个是1xHxW蒙版,背景为零(“纯色”像素),否则为1。然后,您可以将遮罩与第一个转换层的输出进行逐元素相乘,从而强制“纯色”像素的所有特征为零。