david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
XSegEditor/XSegEditor.py
+3 -1
View File
@@ -366,6 +366,8 @@ 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)
@@ -1211,7 +1214,6 @@ class MainWindow(QXMainWindow):
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):