Add train/test split for Caltech pedestrian dataset

Austin Blanton
1 parent aa8a724f
Showing 3 changed files with 15 additions and 0 deletions
.gitignore
scripts/downloadDatasets.sh
scripts/writeCaltechPedestrianSigset.sh
@@ -41,6 +41,7 @@ scripts/results
 data/INRIAPerson/sigset
 data/KTH/sigset
 data/CaltechPedestrians/annotations
+data/CaltechPedestrians/*.xml
 ### Sublime ###
 *.check_cache
@@ -51,6 +51,8 @@ if [ ! -d ../data/CaltechPedestrians/vid ]; then
     tar -xf $fname
   done
   rm *.tar
+  ./writeCaltechPedestrianSigset.sh 0 5 > ../data/CaltechPedestrians/train.xml
+  ./writeCaltechPedestrianSigset 6 10 > ../data/CaltechPedestrians/test.xml
   mv set* ../data/CaltechPedestrians/vid
   if hash curl 2>/dev/null; then
     curl -OL "$prefix/annotations.zip"
+#!/bin/bash
+
+echo '<?xml version="1.0" encoding="UTF-8"?>'
+echo '<biometric-signature-set>'
+for ((set=$1; set <= $2; set++)); do
+	echo -e "\t<biometric-signature name=\"\">"
+	for vid in `printf "set%02d/*.seq" $set`; do
+		echo -e "\t\t<presentation file-name=\"vid/$vid\" />"
+	done
+	echo -e "\t</biometric-signature>"
+done
+echo '</biometric-signature-set>'