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

最新推荐文章于 2024-06-08 16:29:58 发布

原创最新推荐文章于 2024-06-08 16:29:58 发布 · 2.8k 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用Shell脚本来快速生成图片分类任务所需的训练集和测试集标签的方法。通过简单的脚本操作，可以批量为大量图片文件添加分类标签，大大提高了数据准备的工作效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

学习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"

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