0

我目前正在从事一项分类任务,将传送带上的物体图像分类为几个不同的类别。为此,我使用了通过 Python Tensorflow 实现的 MobileNet 架构。此外,物体的重量(例如 1.2 公斤)是可用的,所以我正在考虑通过在图像数据之外使用重量来使神经网络成为多模态,以实现更好的预测。

我正在考虑通过连接它们然后应用softmax来组合两个输入(图像和权重)。此外,输入的权重似乎很有用,因为图像数据比权重重要得多(可能是 95% 到 5%)。

我的问题:

  • 对于权重数据分类,您会推荐哪种模型架构?
  • 你将如何组合这两个输入?

提前致谢。

4

0 回答 0