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



我已经构建并执行了TensorFlow Android演示,但是现在我想生成另一个图形。我需要先训练另一个数据集。我想使用Imagenet。我实际上想从Imagenet下载所有图像。我需要大约500GB。这里有一个脚本在这里执行此操作
在运行此脚本并获得大量培训文件后,我想知道它们是JPEG吗?他们将采用哪种格式?因为然后我想使用结果(培训文件(来创建可以使用TensorFlow构建的图形。

我如何使用Inception脚本的结果使用以下培训脚本创建图形:

 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

根据您提供的页面:

每个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。

之后,您可以使用要使用Retrain Tool训练的实际数据重新培训模型。

如果您完成了培训,则将其转换为优化和/或量化,以便可以在Android移动演示中使用。(请参阅此页面的方法:https://petewarden.com/2016/09/27/tensorflow-for-mobile-poets/(

相关内容

最新更新