Question

我已经构建并执行了TensorFlow Android Demo，但现在我想生成另一个图表。我需要先训练另一个数据集。我想使用ImageNet。我实际上想从imageNet下载所有图像。我需要大约500GB。 do this here有一个脚本我想知道在我运行这个脚本并得到大量的培训文件后，它们会成为jpegs吗？他们会采用什么格式？因为我然后想要使用结果（训练文件）来创建一个我可以用tensorflow构建的图形。

如何使用初始脚本中的结果使用以下培训脚本创建图表：

 cd /tensorflow
 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/flower_photos

Answer 1

根据您提供的页面：

每个tf.Example proto包含ImageNet图像（JPEG编码）为以及标签和边界框信息等元数据。看到 parse_example_proto了解详情。

所以你下载的所有imageNet文件看起来都像是jpeg格式。

您所说的工具是用于重新培训已经训练过的模型。我想你想从头开始训练所有的图像，对吗？

您提供的页面：https://github.com/tensorflow/models/tree/master/inception 还解释了如何从头开始训练数据。

因此，如果您使用

下载了imageNet数据

bazel-bin/inception/download_and_preprocess_imagenet "${DATA_DIR}"

（当然你必须在使用前设置DATA_DIR并构建download_and_preprocess_imagenet）

然后，你可以开始训练：

bazel-bin/inception/imagenet_train --num_gpus=1 --batch_size=32 --train_dir=${TRAIN_DIR} --data_dir=${DATA_DIR}

您可以根据自己的需要和条件更改上述选项，还必须指定TRAIN_DIR。

之后，您可以使用重新训练工具重新训练模型中的实际数据。

如果您完成了训练，那么将其转换为优化和/或量化，以便您可以在Android移动演示中使用。（请参阅此页面了解如何执行此操作：https://petewarden.com/2016/09/27/tensorflow-for-mobile-poets/）

TensorFlow Android演示 - 如何使用使用ImageNet数据集

1 个答案: