如何在 tensorflow 上使用我自己的自定义数据训练新模型

Question

请原谅我，我是 Tensorflow 的新手，所以请尽可能详细地回复。非常感谢！！！

我现在的问题是我有运行这个带有一些参数的脚本

https://github.com/tensorflow/models/blob/master/inception/inception/data/build_image_data.py

# python build_image_data.py --train_directory /Input --validation_directory /validation --output_directory /output --labels_file /labels_file

构建图像数据并获得一些输出，例如 train-00000-of-00002，train-00001-of-00002，validation-00000-of-00002，validation-00001-of-00002

之后，如何在 tensorflow 上使用上述自定义数据训练新模型？

非常感谢！

Answer 1

我已经自己解决了这个愚蠢的问题，谢谢大家。

http://blog.twman.org/2016/06/tensorflow.html

bazel build inception/build_image_data

bazel-bin/inception/build_image_data   --train_directory="${TRAIN_DIR}"  \
--validation_directory="${VALIDATION_DIR}"   \
--output_directory="${OUTPUT_DIRECTORY}"   \
--labels_file="${LABELS_FILE}"   \
--train_shards=128   \
--validation_shards=24   \
--num_threads=8

如何在 tensorflow 上使用我自己的自定义数据训练新模型

How do I train a new model with my own custom data on tensorflow

tensorflow