用im2txt设置大型jpeg

时间:2016-10-25 16:01:39

标签: python tensorflow deep-learning

我只想知道是否可以使用im2txt为存储在同一文件夹中的数千个jpeg图像生成标题。我应该选择那样的东西吗?生成标题,然后将结果写入.txt文件。

# Directory containing model checkpoints.
CHECKPOINT_DIR="${HOME}/im2txt/model/train"

# Vocabulary file generated by the preprocessing script.
VOCAB_FILE="${HOME}/im2txt/data/mscoco/word_counts.txt"

# JPEG image file to caption.
IMAGE_FILE="${HOME}/im2txt/data/lotofimages/

# Build the inference binary.
bazel build -c opt im2txt/run_inference

# Ignore GPU devices (only necessary if your GPU is currently memory
# constrained, for example, by running the training script).
export CUDA_VISIBLE_DEVICES=""

# Run inference to generate captions.
bazel-bin/im2txt/run_inference >> 1.txt \
  --checkpoint_path=${CHECKPOINT_DIR} \
  --vocab_file=${VOCAB_FILE} \
  --input_files=${IMAGE_FILE}

谢谢!

2 个答案:

答案 0 :(得分:1)

我使用这样的方法:

IMAGE_FILE = “/路径/到/ image1.jpg,/路径/到/ image2.jpg /,...,/路径/到/ image1000.jpg”

我将此脚本放入文件script.sh并使用命令运行它:

./script.sh > captions.txt

文件 captions.txt 将包含所有图片的所有标题。

也许有更好的方法,但这个对我来说很好。

答案 1 :(得分:0)

我很好地找到了最好的方法:使用文件模式

... / yourimagesfolder / *。JPG