0

我正在使用带有 GTX 1060 GPU 的 TensorFlow Inception v3 实现图像分类。我在 9 个不同的分类器中训练了大约 3000 张图像，其中每个分类器包含 100 到 500 张图像（100*100 像素），使用诗人的张量。但是当我有超过 500 个图像时，它会抛出错误并且训练失败。具有较少图像的同一类训练没有错误。我每个班级的图像数据大小约为 5000~10000，但只有我可以训练大约 300~400 的数据大小。更多的数据，它会给出以下 GPU 错误。

在此处输入图像描述

规格如下：

  • 带有 GeForce GTX 1060 的 Windows 7
  • 蟒蛇 3 与 python 3.5
  • TensorFlow Inception V3
4

1 回答 1

0

我是 Tensorflow 的新手，但我遇到了同样的问题。我的建议是：

尝试重命名所有图像，例如：

火车图片001.jpg

火车图片002.jpg

火车图片003.jpg

并尝试将您的图像转换为 jpg。

我仍然不确定，但至少试一试.. :)

于 2017-07-20T20:39:11.383 回答