如何转换我自己的大量 jpg 文件以在 tensorflow 中训练新模型

How to convert my own massive jpg files to train new model in tensorflow

我是张量流的新手。

我已经使用 docker 安装并设置了 tensorflow,还很好地启动了一些示例(例如 mnist,"How to Retrain" 以重新训练 Inception-v3。

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

找了Imageflow之类的教程(https://github.com/HamedMP/ImageFlow),但是还是不知道如何把自己目录下的海量jpg文件转换成tensorflow

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

非常感谢!我知道这可能是个愚蠢的问题! 请问谁能帮我解决这个问题,谢谢!!!

除了@Yaroslav 的 URL,您还可以从这样的文件中读取图像:

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

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

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