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

XSegEditor: fix

Browse source
This commit is contained in:
Colombo 2020-04-05 14:14:56 +04:00
parent 33b0aadb4e
commit 215277b376
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
XSegEditor/XSegEditor.py
View file

@ -366,7 +366,9 @@ class QCanvasOperator(QWidget):
def finalize(self): def finalize(self):
if self.initialized: if self.initialized:
if self.op_mode == OpMode.DRAW_PTS:
self.set_op_mode(OpMode.EDIT_PTS)
self.last_state = sn(op_mode = self.op_mode if self.op_mode in [OpMode.VIEW_BAKED, OpMode.VIEW_XSEG_MASK] else None, self.last_state = sn(op_mode = self.op_mode if self.op_mode in [OpMode.VIEW_BAKED, OpMode.VIEW_XSEG_MASK] else None,
color_scheme_id = self.color_scheme_id, color_scheme_id = self.color_scheme_id,
) )
@ -1200,6 +1202,7 @@ class MainWindow(QXMainWindow):
return True return True
def canvas_finalize(self, image_path): def canvas_finalize(self, image_path):
self.canvas.op.finalize()
if image_path.exists(): if image_path.exists():
dflimg = DFLIMG.load(image_path) dflimg = DFLIMG.load(image_path)
@ -1210,8 +1213,7 @@ class MainWindow(QXMainWindow):
self.image_paths_has_ie_polys[image_path] = new_ie_polys.has_polys() self.image_paths_has_ie_polys[image_path] = new_ie_polys.has_polys()
dflimg.set_seg_ie_polys( new_ie_polys ) dflimg.set_seg_ie_polys( new_ie_polys )
dflimg.save() dflimg.save()
self.canvas.op.finalize()
self.filename_label.setText("") self.filename_label.setText("")
def process_prev_image(self): def process_prev_image(self):