Question

我是tensorflow的新手。

我已经使用docker安装并设置了tensorflow，也启动了一些示例（例如mnist和“如何重新训练”以重新训练Inception-v3。

现在，我试图找到输入我自己的图像的方法作为新的火车样本（不是重新训练检查-v3模型），但我不知道如何将目录的自己的图像（JPG）转换为新的tfrecords（或字节流？）

我找到了一些教程，例如Imageflow（https://github.com/HamedMP/ImageFlow），但我仍然不知道如何在目录上转换我自己的大量jpg文件for tensorflow。

另外，我是否需要输出这些我自己的图像目录作为label.txt和图像filename.txt等...？

非常感谢！我知道这可能是一个愚蠢的问题！拜托，有人可以帮我解决这个问题，谢谢!!!

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

Answer 2

除了@ Yaroslav的网址，您还可以从以下文件中读取图片：

img_file = tf.read_file('image.jpg')
img = tf.image.decode_jpeg(img_file)

如何转换我自己的大量jpg文件来训练张量流中的新模型

2 个答案: