You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
955 B
43 lines
955 B
#!/usr/bin/env python |
|
|
|
import sys |
|
import os.path |
|
|
|
# This is a tiny script to help you creating a CSV file from a face |
|
# database with a similar hierarchie: |
|
# |
|
# philipp@mango:~/facerec/data/at$ tree |
|
# . |
|
# |-- README |
|
# |-- s1 |
|
# | |-- 1.pgm |
|
# | |-- ... |
|
# | |-- 10.pgm |
|
# |-- s2 |
|
# | |-- 1.pgm |
|
# | |-- ... |
|
# | |-- 10.pgm |
|
# ... |
|
# |-- s40 |
|
# | |-- 1.pgm |
|
# | |-- ... |
|
# | |-- 10.pgm |
|
# |
|
|
|
if __name__ == "__main__": |
|
|
|
if len(sys.argv) != 2: |
|
print "usage: create_csv <base_path>" |
|
sys.exit(1) |
|
|
|
BASE_PATH=sys.argv[1] |
|
SEPARATOR=";" |
|
|
|
label = 0 |
|
for dirname, dirnames, filenames in os.walk(BASE_PATH): |
|
for subdirname in dirnames: |
|
subject_path = os.path.join(dirname, subdirname) |
|
for filename in os.listdir(subject_path): |
|
abs_path = "%s/%s" % (subject_path, filename) |
|
print "%s%s%d" % (abs_path, SEPARATOR, label) |
|
label = label + 1
|
|
|