create_list.sh 文件内容

最新推荐文章于 2024-09-16 20:45:40 发布

原创最新推荐文章于 2024-09-16 20:45:40 发布 · 1.7k 阅读

0 ·

CC 4.0 BY-SA版权

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

目标检测专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一个用于处理VOCdevkit数据集的bash脚本，该脚本可以为训练、验证和测试数据生成对应的图片和标签列表，并进行必要的格式转换，最后将这些信息整合到一个文件中，便于后续的机器学习模型训练。

注意使用之前请修改文件路径

#!/bin/bash
#注意使用之前请修改文件路径

root_dir=/media/caffeusr/diska/data/VOCdevkit
sub_dir=ImageSets/Main
#bash_dir="/media/caffeusr/diska/caffe/data/VOC0712"
bash_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
for dataset in trainval test
do
  dst_file=$bash_dir/$dataset.txt
  if [ -f $dst_file ]
  then
    rm -f $dst_file
  fi
  for name in VOC2007 VOC2012
  do
    if [[ $dataset == "test" && $name == "VOC2012" ]]
    then
      continue
    fi
    echo "Create list for $name $dataset..."
    dataset_file=$root_dir/$name/$sub_dir/$dataset.txt

    img_file=$bash_dir/$dataset"_img.txt"
    cp $dataset_file $img_file
    sed -i "s/^/$name\/JPEGImages\//g" $img_file
    sed -i "s/$/.jpg/g" $img_file

    label_file=$bash_dir/$dataset"_label.txt"
    cp $dataset_file $label_file
    sed -i "s/^/$name\/Annotations\//g" $label_file
    sed -i "s/$/.xml/g" $label_file

    paste -d' ' $img_file $label_file >> $dst_file

    rm -f $label_file
    rm -f $img_file
  done

  # Generate image name and size infomation.
  if [ $dataset == "test" ]
  then
    $bash_dir/../../build/tools/get_image_size $root_dir $dst_file $bash_dir/$dataset"_name_size.txt"
  fi

  # Shuffle trainval file.
  if [ $dataset == "trainval" ]
  then
    rand_file=$dst_file.random
    cat $dst_file | perl -MList::Util=shuffle -e 'print shuffle(<STDIN>);' > $rand_file
    mv $rand_file $dst_file
  fi
done