我想训练一个卷积网络来执行以下操作:
- 输入是一组具有给定对象的单通道(从黑色到灰色到白色)图片,比如汽车。
- 对于集合中的每张图片,目标是相同的图片,但是像素不是黑色就是白色。与汽车对象对应的像素为白色(即强度 255),与背景对应的像素为黑色(即强度 0)。
训练后,我想向网络提供汽车图片,并且我希望预测(至少是理想的预测)是具有黑色或白色像素的图片,其中白色对应于对象,黑色对应于背景。
我假设输入层是一个二维卷积层,输出层也是一个二维卷积层,每个卷积层的神经元数量与图片中的像素一样多。
谁能解释一下什么样的网络架构可以做到这一点?
它可以是架构(理论上)或以代码实现。
我希望对其进行调整,但最好不要从头开始。