简体   繁体   English

使用TensorFlow Inception V3进行错误训练图像

[英]Error training images using tensorflow inception v3

I am implementing image classification using TensorFlow Inception v3 with GTX 1060 GPU. 我正在使用TensorFlow Inception v3和GTX 1060 GPU实现图像分类。 I trained around 3000 images in 9 different classifier, where each classifier contains from 100 to 500 images(100*100 px) using tensor for poets. 我在9个不同的分类器中训练了大约3000张图像,其中每个分类器都使用张量用于诗人,包含100到500张图像(100 * 100 px)。 But when I have images more than 500, it throws out error and training fails. 但是,当我的图像超过500张时,它会抛出错误,并且训练失败。 Same class with less images trains without error. 图像少的同一个类训练不会出错。 My Image data size for each class is around 5000~10000 but only I can train data size of around 300~400. 我每个班级的图像数据大小约为5000〜10000,但只有我可以训练300〜400左右的数据大小。 more of the data and it gives out following GPU error. 更多的数据,它会给出以下GPU错误。

在此处输入图片说明

The specification as follows: 规格如下:

  • Windows 7 with GeForce GTX 1060 Windows 7和GeForce GTX 1060
  • Anaconda 3 with python 3.5 Anaconda 3与python 3.5
  • Tensorflow Inception V3 Tensorflow盗梦空间V3

I am novice for Tensorflow but I had faced the same problem. 我是Tensorflow的新手,但我也遇到过同样的问题。 My suggestion would be: 我的建议是:

Try to rename all the images like: 尝试重命名所有图像,例如:

TrainImage001.jpg TrainImage001.jpg

TrainImage002.jpg TrainImage002.jpg

TrainImage003.jpg TrainImage003.jpg

and try converting your images to jpg. 然后尝试将图像转换为jpg。

Still I am not sure but at least give it a try.. :) 我仍然不确定,但至少可以尝试一下.. :)

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM