create_data.sh 文件内容

最新推荐文章于 2025-01-12 17:04:58 发布

原创最新推荐文章于 2025-01-12 17:04:58 发布 · 2.3k 阅读

0 ·

CC 4.0 BY-SA版权

本文为while10博主原创文章，未经博主允许不得转载。

目标检测专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一个用于Caffe深度学习框架的数据集准备脚本。该脚本通过修改特定路径，自动化创建标注文件，适用于VOC0712数据集。详细解释了脚本中各参数的作用，如数据根目录、数据集名称、标注类型等，并展示了如何使用Python脚本处理图像和标注数据。

注意使用之前请修改路径

#cur_dir="/media/caffeusr/diska/caffe/data/VOC0712"
#注意使用之前请修改路径
cur_dir=$(cd $( dirname ${BASH_SOURCE[0]} ) && pwd )
root_dir=$cur_dir/../..
#root_dir=$cur_dir

cd $root_dir

redo=1
data_root_dir="/media/caffeusr/diska/data/VOCdevkit"
dataset_name="VOC0712"
mapfile="$root_dir/data/$dataset_name/labelmap_voc.prototxt"
anno_type="detection"
db="lmdb"
min_dim=0
max_dim=0
width=0
height=0

extra_cmd="--encode-type=jpg --encoded"
if [ $redo ]
then
  extra_cmd="$extra_cmd --redo"
fi
for subset in test trainval
do
  python $root_dir/scripts/create_annoset.py --anno-type=$anno_type --label-map-file=$mapfile --min-dim=$min_dim --max-dim=$max_dim --resize-width=$width --resize-height=$height --check-label $extra_cmd $data_root_dir $root_dir/data/$dataset_name/$subset.txt $data_root_dir/$dataset_name/$db/$dataset_name"_"$subset"_"$db examples/$dataset_name
done