我正在使用TensorFlow Inception v3和GTX 1060 GPU实现图像分类。我在9种不同的分类器中训练了大约3000张图像,其中每个分类器包含100至500张图像(100 * 100像素),使用张诗作为诗人。 但是当我的图像超过500时,它会抛出错误并且训练失败。相同类别的图像较少,没有错误。我的每个类的图像数据大小约为5000~10000,但我只能训练大约300~400的数据。更多的数据,它给出了以下GPU错误。
规范如下:
答案 0 :(得分:0)
我是Tensorflow的新手,但我遇到了同样的问题。 我的建议是:
尝试重命名所有图片,如:
TrainImage001.jpg
TrainImage002.jpg
TrainImage003.jpg
并尝试将图片转换为jpg。
仍然我不确定,但至少试一试.. :))