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-11-07 10:03:15 +04:00
parent 8489949f2c
commit 30ba51edf7
24 changed files with 663 additions and 459 deletions
Download patch file Download diff file Expand all files Collapse all files

22
scripts/dev.py
View file

@ -53,14 +53,10 @@ def extract_FaceSynthetics(inputdir_path : Path, faceset_path : Path):
"""
if faceset_path.suffix != '.dfs':
raise ValueError('faceset_path must have .dfs extension.')
filepaths = lib_path.get_files_paths(inputdir_path)
fs = lib_face.Faceset(faceset_path)
fs.recreate()
for filepath in lib_con.progress_bar_iterator(filepaths, 'Processing'):
fs = lib_face.Faceset(faceset_path, write_access=True, recreate=True)
for filepath in lib_con.progress_bar_iterator(filepaths, desc='Processing'):
if filepath.suffix == '.txt':
image_filepath = filepath.parent / f'{filepath.name.split("_")[0]}.png'
@ -93,17 +89,13 @@ def extract_FaceSynthetics(inputdir_path : Path, faceset_path : Path):
ufm.set_UImage_uuid(uimg.get_uuid())
ufm.set_FRect(flmrks.get_FRect())
ufm.add_FLandmarks2D(flmrks)
fs.add_UFaceMark(ufm)
fs.add_UImage(uimg, format='png')
fs.add_UFaceMark(ufm)
fs.shrink()
fs.optimize()
fs.close()
import code
code.interact(local=dict(globals(), **locals()))
# seg_filepath = input_path / ( Path(image_filepath).stem + '_seg.png')
# if not seg_filepath.exists():
# raise ValueError(f'{seg_filepath} does not exist')