0

我是张量流的新手。

我已经使用 docker 安装和设置了 tensorflow,还很好地启动了一些示例(例如 mnist,以及“如何重新训练”来重新训练 Inception-v3.

现在,我尝试找到将自己的图像输入为新的训练样本(不是重新训练检查-v3 模型)的方法,但我不确定如何将我自己的目录图像(JPG)转换为 tenslfow 作为新的 tfrecords(或字节流?)

我找到了一些教程,例如 Imageflow ( https://github.com/HamedMP/ImageFlow ),但我仍然不知道如何将我自己的海量 jpg 文件转换为 tensorflow 目录。

另外,我是否需要将这些我自己的图像目录输出为 label.txt 和图像文件名.txt 等...?

非常感谢 !我知道这可能是一个愚蠢的问题!求大神帮忙解答一下这个问题,谢谢!!!

4

2 回答 2

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
于 2016-06-29T18:23:04.830 回答
0

除了@Yaroslav 的 URL,您还可以从文件中读取图像,如下所示:

img_file = tf.read_file('image.jpg')
img = tf.image.decode_jpeg(img_file)
于 2016-05-04T23:58:55.310 回答