image-processing - Tensorflow:标签初始阶段在类别训练中没有图像

Question

我正在根据此处的代码重新训练 Inception v3 网络:https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#0 。我有一套 200 个标签。每个标签都有图像，其数量范围从 50 到 15000。重新训练时，我收到以下错误:

2017-01-08 07:42:09.683263: Step 30: Train accuracy = 6.0%
2017-01-08 07:42:09.683384: Step 30: Cross entropy = 6.297137
2017-01-08 07:42:09.808175: Step 30: Validation accuracy = 0.0%
2017-01-08 07:42:11.083850: Step 40: Train accuracy = 2.0%
2017-01-08 07:42:11.083964: Step 40: Cross entropy = 6.296147
CRITICAL:tensorflow:Label corset has no images in the category validation.
Traceback (most recent call last):
 File "tensorflow/examples/image_retraining/retrain.py", line 1012, in <module>
   tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)
 File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py", line 43, in run
   sys.exit(main(sys.argv[:1] + flags_passthrough))
 File "tensorflow/examples/image_retraining/retrain.py", line 839, in main
   bottleneck_tensor))
 File "tensorflow/examples/image_retraining/retrain.py", line 480, in get_random_cached_bottlenecks
   bottleneck_tensor)
 File "tensorflow/examples/image_retraining/retrain.py", line 388, in get_or_create_bottleneck
   bottleneck_dir, category)
 File "tensorflow/examples/image_retraining/retrain.py", line 245, in get_bottleneck_path
   category) + '.txt'
 File "tensorflow/examples/image_retraining/retrain.py", line 221, in get_image_path
   mod_index = index % len(category_list)
ZeroDivisionError: integer division or modulo by zero`

Google 上的搜索让我了解到，如果图像少于 20 个，则可能会发生此类错误。发生这种情况的原因是它们可能没有足够的图像可供验证。不过我至少有 50 张图片。那么为什么我仍然收到此错误？

Answer 1

我查看了tensorflow/examples/image_retraining/retrain.py文件是从您帖子中的教程链接的。

运行的部分代码是在函数 create_image_lists(image_dir, testing_percentage, validation_percentage) 中进行的训练/验证拆分。

分割的默认值为 10%，并在 FLAGS.validation_percentage 中说明。

由于您的类的图像少于 200 张，因此拆分返回的类的照片少于 20 张以进行验证，这就是您出现错误的地方。

尝试仅使用具有超过 200 个图像的类来运行代码，看看是否有帮助。如果确实如此，您可以考虑添加更多图像或操作 create_image_lists 函数以返回至少 20 张照片进行验证。