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



我是tensorflow的新手。

我已经用docker安装并设置了tensorflow,还启动了一些示例(如mnist和"如何重新训练"来重新训练Inception-v3。

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

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

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

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

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

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

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

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

最新更新