Question

我正在尝试使用张力流为诗人训练我的图像识别模型。它正在构建花卉数据的示例情况并生成瓶颈，但当我尝试重新训练它用于另一个测试用例时（我在其中创建了四个文件夹，即Chandler，Darth，Ross，Joey，在一个名为tf_files的文件夹中，此文件夹存在在我的家里）

当我运行此命令时

python tensorflow/examples/image_retraining/retrain.py \
--bottleneck_dir=/tf_files/bottlenecks \
--how_many_training_steps 500 \
--model_dir=/tf_files/inception \
--output_graph=/tf_files/retrained_graph.pb \
--output_labels=/tf_files/retrained_labels.txt \
--image_dir /tf_files

我收到此错误

InvalidArgumentError (see above for traceback): Invalid JPEG data, size 
43[[Node: DecodeJpeg = DecodeJpeg[acceptable_fraction=1, channels=3, 
dct_method="", fancy_upscaling=true, ratio=1, try_recover_truncated=false, 
_device="/job:localhost/replica:0/task:0/cpu:0"](_recv_DecodeJpeg
/contents_0)]]

和许多警告

但是当我为花标准例子运行上述命令时

python tensorflow/examples/image_retraining/retrain.py \
--bottleneck_dir=/tf_files/bottlenecks \
--how_many_training_steps 500 \
--model_dir=/tf_files/inception \
--output_graph=/tf_files/retrained_graph.pb \
--output_labels=/tf_files/retrained_labels.txt \
--image_dir /flower_photos

我没有收到任何错误。我的模型得到了再培训，我得到了预测

在这个领域非常新，请指导我错误的安装错误或其他什么

Answer 1

您得到的错误是因为数据集中可能有一些png格式的图像。另一方面，花卉数据集具有jpeg格式的所有图像。

此外，编写retrain.py来处理jpeg格式的图像。您可能希望关注this thread以了解有关处理png图片的更多信息

无效的JPEG数据，构建部分再培训部分43的错误，以及诗人

1 个答案: