github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 05:22:06 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactoring after @kcimit

Browse source
This commit is contained in:
Colombo 2019-10-14 17:20:43 +04:00
parent d26ac4896e
commit 5ad4b1011c
1 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
mainscripts/MaskEditorTool.py
View file

@ -319,7 +319,14 @@ class MaskEditor:
def get_ie_polys(self):
return self.ie_polys
def set_ie_polys(self, saved_ie_polys):
self.state = self.STATE_NONE
self.ie_polys = saved_ie_polys
self.redo_to_end_point()
self.mask_finish()
def mask_editor_main(input_dir, confirmed_dir=None, skipped_dir=None, no_default_mask=False):
input_path = Path(input_dir)
@ -485,6 +492,7 @@ def mask_editor_main(input_dir, confirmed_dir=None, skipped_dir=None, no_default
break
elif filepath is not None:
if chr_key == 'e':
saved_ie_polys = ed.ie_polys
do_save_move_count = 1 if not shift_pressed else 10
elif chr_key == 'c':
saved_ie_polys = ed.ie_polys
@ -494,10 +502,7 @@ def mask_editor_main(input_dir, confirmed_dir=None, skipped_dir=None, no_default
elif chr_key == 'x':
do_skip_count = 1 if not shift_pressed else 10
elif chr_key == 'r' and saved_ie_polys != None:
ed.state = 0
ed.ie_polys = saved_ie_polys
ed.redo_to_end_point()
ed.mask_finish()
ed.set_ie_polys(saved_ie_polys)
if do_prev_count > 0:
do_prev_count -= 1