Python 我应该使用哪个损失函数进行形状提取?

Python 我应该使用哪个损失函数进行形状提取?,python,tensorflow,neural-network,conv-neural-network,Python,Tensorflow,Neural Network,Conv Neural Network,我需要建立一个神经网络来识别图像中的多个特定形状配置(正方形、三角形…),并改变它们的颜色作为输出 因此,神经网络将(例如)以32x32x1(灰度)的图像作为输入,以32x32x3作为输出。我的问题是:我应该用什么样的损失函数来训练我的神经网络。因为我读了一些文章来找出答案,而这不是一个分类或回归问题(根据我的理解),所以我有点迷茫,因为我是一个新手 考虑到我有一个数据库,其中的输出图像与输入图像相对应,我可以对网络进行训练 我正在为这个项目使用Tensorflow 如果没有与图像对关联的形状

我需要建立一个神经网络来识别图像中的多个特定形状配置(正方形、三角形…),并改变它们的颜色作为输出

因此,神经网络将(例如)以32x32x1(灰度)的图像作为输入,以32x32x3作为输出。我的问题是:我应该用什么样的损失函数来训练我的神经网络。因为我读了一些文章来找出答案,而这不是一个分类或回归问题(根据我的理解),所以我有点迷茫,因为我是一个新手

  • 考虑到我有一个数据库,其中的输出图像与输入图像相对应,我可以对网络进行训练
  • 我正在为这个项目使用Tensorflow

如果没有与图像对关联的形状,则只有图像对: 您可以构建图像去噪网络()并使用反卷积生成彩色图像

损耗函数:您可以使用MSE损耗函数,通过比较网络输出图像和真实图像的每个通道中的每个像素来计算损耗

如果您有与每个输入图像关联的形状: 然后,您可以构建一个简单的深度神经网络分类器来预测形状,然后根据形状填充输出图像中的颜色值(这可以使用python中的cv2库执行)