生成train.txt和val.txt

最新推荐文章于 2025-06-21 15:17:52 发布

转载最新推荐文章于 2025-06-21 15:17:52 发布 · 2.4k 阅读

文章标签：

#caffe #linux #python

本文介绍了一个用于Caffe框架的数据集划分脚本。该脚本能够将图像文件按类别划分为训练集和验证集，并生成对应的txt文件，便于进行深度学习模型的训练和验证。

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

#!/usr/bin/env sh
DATA=/home/cjj/caffe/data/myimage
MY=/home/cjj/caffe/data/myimage
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 '/' -f8 | 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/val -name $i*.jpg | cut -d '/' -f8 | sed "s/$/ $i/">>$MY/test.txt
done

echo "All done"

#!/usr/bin/env sh
DATA=/home/cjj/caffe/data/myimage
    echo "Create train.txt..."
    rm -rf $DATA/train.txt
    find $DATA/train/s -name *.JPEG | cut -d '/' -f8-9 | sed "s/$/ 1/">>$DATA/train.txt
    find $DATA/train/x -name *.JPEG | cut -d '/' -f8-9 | sed "s/$/ 2/">>$DATA/tmp.txt
     cat $DATA/tmp.txt>>$DATA/train.txt
     rm -rf $DATA/tmp.txt
    echo "Done.."

#!/usr/bin/env sh
DATA=/home/cjj/caffe/data/myimage
    echo "Create val.txt..."
    rm -rf $DATA/val.txt
    find $DATA/val -name n01704323*.JPEG | cut -d '/' -f8 | sed "s/$/ 1/">>$DATA/val.txt
    find $DATA/val -name n01770393*.JPEG | cut -d '/' -f8 | sed "s/$/ 2/">>$DATA/val.txt
    find $DATA/val -name n01734418*.JPEG | cut -d '/' -f8 | sed "s/$/ 3/">>$DATA/val.txt
    find $DATA/val -name n01773157*.JPEG | cut -d '/' -f8 | sed "s/$/ 4/">>$DATA/val.txt
    # cat $DATA/tmp.txt>>$DATA/train.txt
    # rm -rf $DATA/tmp.txt
    echo "Done.."