tensorflow之准备好自己的train.txt和val.txt

   学习tensorflow时需要准备好自己的训练集train.txt 把训练图片给打上分类标签  和训练过程中的测试集val.txt 要给打上分类标签。

这样就需要你准备好train.txt和val.txt。怎样快速准备好自己的标签。可用shell脚本进行快速生成。

</pre><pre name="code" class="plain">#!/usr/bin/env sh
DATA=data/zhao/
MY=examples/zhaofile

echo "Create train.txt..."
rm -rf $MY/train.txt
for i in 3 4 5 6 7 
do
find $DATA/train -name $i*.jpg | cut -d '/' -f4-5 | sed "s/$/ $i/">>$MY/train.txt
done
echo "Create test.txt..."
rm -rf $MY/test.txt
for i in 3 4 5 6 7
do
find $DATA/test -name $i*.jpg | cut -d '/' -f4-5 | sed "s/$/ $i/">>$MY/test.txt
done
echo "All done"

用上这样的脚本就可以生成自己的标签。