Skip to content

Commit b9eab58

Browse files
authored
Update preprocess.py
Add LaPa dataset
1 parent 3d4599d commit b9eab58

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

lib/preprocess.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -536,6 +536,10 @@ def gen_data(root_folder, data_name, target_size):
536536
f.write(str(x)+' '+str(y)+' ')
537537
f.write('\n')
538538
gen_meanface(root_folder, data_name)
539+
################################################################################################################
540+
elif data_name == 'LaPa':
541+
pass
542+
# TODO
539543
else:
540544
print('Wrong data!')
541545

@@ -546,6 +550,7 @@ def gen_data(root_folder, data_name, target_size):
546550
print('2. COFW')
547551
print('3. WFLW')
548552
print('4. AFLW')
553+
print('5. LaPa')
549554
exit(0)
550555
else:
551556
data_name = sys.argv[1]

0 commit comments

Comments
 (0)