Question

我在python中使用TensorFlow库。我正在尝试重新训练我自己的图像数据集，以便从我训练过的图像中获取预测结果。我正在关注this教程。我收到了一些错误＆＃34; My_image_folder＆＃34;在类别验证中没有图像并且除以零，尽管＆＃34; My_image_folder＆＃34;当我使用以下命令时，在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。

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

为什么我在类别验证中遇到错误？请帮帮我。我已经坚持了很长时间。

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

0 个答案: