具有Fashion MNIST数据集的TensorFlow模型

时间:2018-10-28 10:46:23

标签: java python tensorflow machine-learning keras

我对TensorFlow很陌生。我想将TensorFlow用于我的时尚电子商务网站(在Java上运行),以便所有产品图片都应使用CNN进行自动标记。

我使用inception5h训练有素的tensorflow模型检查了LabelImage.java示例,用于自动标记/标记新图像。

似乎inception5h不适合时尚数据集。我还找到了一个使用Fashion MNIST数据集进行基本分类的火车优先神经网络解释的指南。

https://www.tensorflow.org/tutorials/keras/basic_classification

我还想创建自己的模型,在Fashion MNIST数据集上对其进行训练,并在我的Java代码中使用它进行自动标记。

问题-

  1. 我们也可以为Fashion MNIST数据集重新训练inception5h模型吗?
  2. 时装数据集是否有任何预训练的张量流模型?
  3. 如果我训练了模型,是否可以通过参考LabelImage.java示例在Java代码中使用相同的模型? LabelImage示例需要模型和图形标签。

1 个答案:

答案 0 :(得分:-1)

我只是在研究Fashion-MNIST。据说F-MNIST刻意模仿了MNIST的所有特性,并且只需稍微更改数据集路径即可在F-MNIST上使用在MNIST上训练的任何模型。

尽管将我自己的MNIST模型转移到Fashion-MNIST数据集时存在过度拟合的问题。我仍在检查我的代码并尝试找出问题所在。