All

ganimation/openface_utils/create_csv.py

@@ -0,0 +1,11 @@
+import glob
+
+output_txt = 'list_attr_mydataset.txt'
+
+for idx, f in enumerate(glob.glob('./my-processed-dataset/*.csv')):
+    with open(f, 'r') as csv_file:
+        csv_file.readline()
+        csv_list = csv_file.readline().split(', ')
+        if float(csv_list[1]) >= 0.88:
+            aus = " ".join(csv_list[2:19])
+            open(output_txt, 'a').write(f.split('/')[-1].split('.')[0] + '.jpg ' + aus + '\n')

ganimation/openface_utils/create_image_folder.py

@@ -0,0 +1,12 @@
+import os
+import shutil
+
+output_dir = './images'
+
+os.mkdir(output_dir)
+
+for root, dirs, files in os.walk('./my-processed-dataset'):
+    for filename in files:
+        if 'jpg' in filename:
+            img_name = root.split('/')[-1].split('_')[0] + '.jpg'
+            shutil.copy2(os.path.join(root, filename), os.path.join(output_dir, img_name))