add fddb download script

Author: Microos

README.md

@@ -1,5 +1,10 @@
 # FaceAnnotationTool
 
+### Testing with FDDB images:
+	get fddb images and annotations by executing `download_fddb.sh`
+
+
+
 ### TODO:
 - [x]  basic functions
 - [x]  ask for prefix and suffix when load imglist
@@ -10,3 +15,4 @@
 - [x]  one click after load -> disable load annotation button
 - [x]  load annotation -> disable load annotation button
 - [x]  output file function 
+

download_fddb.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+
+echo "Downloading fddb images"
+
+mkdir fddb && cd fddb
+wget http://tamaraberg.com/faceDataset/originalPics.tar.gz
+echo "Unzip..."
+tar -xzf originalPics.tar.gz
+rm -f originalPics.tar.gz
+
+wget http://vis-www.cs.umass.edu/fddb/FDDB-folds.tgz
+echo "Unzip..."
+tar -xzf FDDB-folds.tgz
+rm -f FDDB-folds.tgz
+
+cd ..
+python gen_absolute_path.py
+
+

gen_absolute_path.py

@@ -0,0 +1,60 @@
+'''
+(By default, this script will be automatically invoked by download_fddb.sh)
+
+Manually invoke this script to generate image list and annotation files with
+absolute paths.
+
+Please make sure this script stays aside with a folder fddb which contains
+sub-folders: 2002, 2003, FDDB-folds.
+
+Invoke this script again after you move the directory of images.
+
+'''
+import os
+
+
+assert os.path.exists("fddb/2002"), "fddb iamge folder 'fddb/2002' not found"
+assert os.path.exists("fddb/2003"), "fddb iamge folder 'fddb/2003' not found"
+assert os.path.exists("fddb/FDDB-folds"), "fddb annotation folder 'fddb/FDDB-folds' not found"
+
+os.chdir('fddb')
+prefix = os.getcwd()
+suffix = ".jpg"
+x=os.popen("pwd | sed 's/ /\\ /g'").read().strip()
+img_list_files = os.popen("find '{}/FDDB-folds' -name '*[0-9].txt'".format(x)).read().split('\n')
+anot_list_files = os.popen("find '{}/FDDB-folds' -name '*List.txt'".format(x)).read().split('\n')
+
+
+s = ""
+for p in img_list_files:
+    if p == '' :
+        continue
+    with open(p,'r') as f:
+      lines = f.readlines()
+    lines = [os.path.join(prefix, l.replace('\n','')+suffix) for l in lines]  
+    s += "\n".join(lines)+'\n'
+
+with open("./MergedImagePath.txt",'w') as f:
+    f.write(s)
+
+s = ""
+for p in anot_list_files:
+    if p == '' :
+        continue    
+    with open(p,'r') as f:
+        lines = f.readlines()
+
+    lines = [l.replace('\n','') for l in lines]
+    for i in range(len(lines)):
+        l = lines[i]
+        if(len(l.split("/")) == 5):
+            lines[i] = os.path.join(prefix, l+suffix)
+    s += '\n'.join(lines)+'\n'
+
+with open("./MergedAnnotations.txt",'w') as f:
+    f.write(s)
+
+print "Merged image path wrote to: "+os.path.join(os.getcwd(),"MergedImagePath.txt")
+print "Merged annotations wrote to: "+os.path.join(os.getcwd(),"MergedAnnotations.txt")
+
+