github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-07-06 13:02:16 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactoring

Browse source
This commit is contained in:
iperov 2021-10-26 20:06:50 +04:00
parent d90ec2d024
commit 8b385f6d80
11 changed files with 263 additions and 137 deletions
Download patch file Download diff file Expand all files Collapse all files

7
scripts/dev.py
View file

@ -26,7 +26,7 @@ def split_large_files(delete_original=False):
SplittedFile.split(filepath, part_size=part_size, delete_original=delete_original)
print('Done')
def extract_facesynthetics_dataset(input_dir):
def extract_FaceSynthetics(input_dir):
"""
extract FaceSynthetics dataset https://github.com/microsoft/FaceSynthetics
@ -53,7 +53,7 @@ def extract_facesynthetics_dataset(input_dir):
"""
input_path = Path(input_dir)
faceset_path = input_path.parent / f'{input_path.name}.dfs'
# fs = lib_face.Faceset(output_dbpath)
# for ufm in fs.iter_UFaceMark():
# uimg = fs.get_UImage_by_uuid( ufm.get_UImage_uuid() )
@ -65,7 +65,7 @@ def extract_facesynthetics_dataset(input_dir):
filepaths = lib_path.get_files_paths(input_path)[:100] #TODO
fs = lib_face.Faceset(faceset_path)
fs.clear_db()
fs.recreate()
for filepath in lib_con.progress_bar_iterator(filepaths, 'Processing'):
@ -104,6 +104,7 @@ def extract_facesynthetics_dataset(input_dir):
fs.add_UImage(uimg, format='png')
fs.add_UFaceMark(ufm)
fs.shrink()
fs.close()