Created
September 30, 2017 16:41
-
-
Save ahmetozlu/5663f0b56f4339c2e72096877f57d6a5 to your computer and use it in GitHub Desktop.
Automatically creates you a CSV file
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#---------------------------------------------- | |
#--- Author : Ahmet Ozlu | |
#--- Mail : ahmetozlu93@gmail.com | |
#--- Date : 21st September 2017 | |
#---------------------------------------------- | |
import sys | |
import os.path | |
import csv | |
# This is a tiny script to help you creating a CSV file from a face database with a similar hierarchie: | |
# . | |
# |-- README | |
# |-- s1 | |
# | |-- 1.pgm | |
# | |-- ... | |
# | |-- 10.pgm | |
# |-- s2 | |
# | |-- 1.pgm | |
# | |-- ... | |
# | |-- 10.pgm | |
# ... | |
# |-- s40 | |
# | |-- 1.pgm | |
# | |-- ... | |
# | |-- 10.pgm | |
# | |
def CreateCsv(csvLine): | |
BASE_PATH=csvLine | |
SEPARATOR="," | |
label = 0 | |
with open("face_dataset.csv", "w") as file: | |
writer = csv.writer(file) | |
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) | |
csv_line = "%s%s%d" % (abs_path, SEPARATOR, label) | |
print csv_line | |
writer.writerows([csv_line.split(',')]) | |
label = label + 1 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment