python - TensorFlow 在类别验证中找不到图像

Question

我正在 python 中使用 TensorFlow 库。我正在尝试重新训练我自己的图像数据集，以便从我训练的图像中获取预测结果。我正在关注this教程。我收到一些错误，因为“My_image_folder”在类别验证中没有图像且除以零，尽管当我使用以下命令时“My_image_folder”包含flower_photos文件夹内的图像 -

python -m scripts.retrain \
  --bottleneck_dir=tf_files/bottlenecks \
  --how_many_training_steps=500 \
  --model_dir=tf_files/models/ \
  --summaries_dir=tf_files/training_summaries/"${ARCHITECTURE}" \
  --output_graph=tf_files/retrained_graph.pb \
  --output_labels=tf_files/retrained_labels.txt \
  --architecture="${ARCHITECTURE}" \
  --image_dir=tf_files/flower_photos

My_image_folder 中有 10 张图片，全部为 jpeg/jpg 格式，每张图片大小小于 2 MB。

使用上述教程进行训练是否有任何预定义的图像格式、图像大小、内存问题等？

为什么我在类别验证中遇到错误？请帮帮我。我已经被这个问题困扰了很长时间了。

Answer 1

您可以修改retrain.py以确保目录中至少有一张图像，如引用here 。我尚未测试对训练的影响，但它可以让您运行端到端测试而不会失败。

编辑此内容(第 220 行左右):

  if percentage_hash < validation_percentage:
    validation_images.append(base_name)
  elif percentage_hash < (testing_percentage + validation_percentage):
    testing_images.append(base_name)
  else:
  ...

像这样:

  if len(validation_images) == 0:
     validation_images.append(base_name)
  elif percentage_hash < validation_percentage:
    validation_images.append(base_name)
  elif percentage_hash < (testing_percentage + validation_percentage):
    testing_images.append(base_name)
  else:
  ...