使用tensorflow开始v3训练图像时出错

时间:2017-05-24 02:56:50

标签: python tensorflow deep-learning image-recognition tensorflow-gpu

我正在使用TensorFlow Inception v3和GTX 1060 GPU实现图像分类。我在9种不同的分类器中训练了大约3000张图像,其中每个分类器包含100至500张图像(100 * 100像素),使用张诗作为诗人。 但是当我的图像超过500时,它会抛出错误并且训练失败。相同类别的图像较少,没有错误。我的每个类的图像数据大小约为5000~10000,但我只能训练大约300~400的数据。更多的数据,它给出了以下GPU错误。

enter image description here

规范如下:

  • 使用GeForce GTX 1060的Windows 7
  • Anaconda 3 with python 3.5
  • Tensorflow Inception V3

1 个答案:

答案 0 :(得分:0)

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

尝试重命名所有图片,如:

TrainImage001.jpg

TrainImage002.jpg

TrainImage003.jpg

并尝试将图片转换为jpg。

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