CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

XSegEditor: fixed incorrect count of labeled images.

Browse Source
This commit is contained in:
Colombo
2020-06-21 00:56:25 +04:00
parent 2b8e8f0554
commit 27f219fec7
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
XSegEditor/XSegEditor.py
+9 -3
View File
@@ -1295,9 +1295,15 @@ class MainWindow(QXMainWindow):
new_ie_polys = self.canvas.op.get_ie_polys() new_ie_polys = self.canvas.op.get_ie_polys()
if not new_ie_polys.identical(ie_polys): if not new_ie_polys.identical(ie_polys):
new_has_ie_polys = new_ie_polys.has_polys() prev_has_polys = self.image_paths_has_ie_polys[image_path]
self.set_has_ie_polys_count ( self.get_has_ie_polys_count() + (1 if new_has_ie_polys else -1) ) self.image_paths_has_ie_polys[image_path] = new_ie_polys.has_polys()
self.image_paths_has_ie_polys[image_path] = new_has_ie_polys new_has_polys = self.image_paths_has_ie_polys[image_path]
if not prev_has_polys and new_has_polys:
self.set_has_ie_polys_count ( self.get_has_ie_polys_count() +1)
elif prev_has_polys and not new_has_polys:
self.set_has_ie_polys_count ( self.get_has_ie_polys_count() -1)
dflimg.set_seg_ie_polys( new_ie_polys ) dflimg.set_seg_ie_polys( new_ie_polys )
dflimg.save() dflimg.save()