otb_get_labels.sh 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. #!/bin/bash
  2. dataset=$1
  3. w=$2
  4. h=$3
  5. # Parameters: Human3, CarScale, Human6, Biker
  6. #w=480
  7. #h=640
  8. IFS=','
  9. export LC_NUMERIC="en_US.UTF-8"
  10. wd=`pwd`
  11. dataset_path="data/$dataset"
  12. class_id=0
  13. num=1
  14. mkdir data
  15. wget http://cvlab.hanyang.ac.kr/tracker_benchmark/seq/$dataset.zip
  16. unzip -o $dataset.zip -d data
  17. sed -i.bak 's/\o11/,/g' $dataset_path/groundtruth_rect.txt
  18. sed -i.bak 's/\o11/,/g' $dataset_path/groundtruth_rect.txt
  19. dos2unix $dataset_path/groundtruth_rect.txt
  20. while read -r left right width height; do
  21. filename=$(printf "$dataset_path/img/%04d.txt" $num)
  22. #rm $filename.txt
  23. echo "$class_id " > $filename
  24. printf "%.5f " "$(($((left + width/2)) * 100000 / $w))e-5" >> $filename
  25. printf "%.5f " "$(($((right + height/2)) * 100000 / $h))e-5" >> $filename
  26. printf "%.5f " "$(($((width)) * 100000 / $w))e-5" >> $filename
  27. printf "%.5f " "$(($((height)) * 100000 / $h))e-5" >> $filename
  28. num=$((num + 1))
  29. done < $dataset_path/groundtruth_rect.txt
  30. echo "$dataset" > $dataset_path/otb.names
  31. find $dataset_path/img -name \*.jpg > data/$dataset/train.txt
  32. echo "classes = 1" > data/otb_$dataset.data
  33. echo "train = data/$dataset/train.txt" >> data/otb_$dataset.data
  34. echo "valid = data/$dataset/train.txt" >> data/otb_$dataset.data
  35. echo "names = $dataset_path/otb.names" >> data/otb_$dataset.data
  36. echo "backup = backup/" >> data/otb_$dataset.data
  37. echo "results= results/" >> data/otb_$dataset.data