StarMirror/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2024-03-22 13:10:55 +08:00
Code Issues Projects Releases Wiki Activity

_

Browse Source
This commit is contained in:
Colombo 2020-03-21 11:11:52 +04:00
parent 46eacd6fd2
commit deb705ad53
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
DFLIMG/DFLJPG.py
View File

@ -259,12 +259,8 @@ class DFLJPG(object):
self.dfl_dict['ie_polys'] = ie_polys
def get_seg_ie_polys(self): return self.dfl_dict.get('seg_ie_polys',None)
def get_seg_ie_polys(self): return self.dfl_dict.get('seg_ie_polys',None)
def set_seg_ie_polys(self, seg_ie_polys):
if seg_ie_polys is not None and \
not isinstance(seg_ie_polys, dict):
seg_ie_polys = seg_ie_polys.dump()
self.dfl_dict['seg_ie_polys'] = seg_ie_polys

2
mainscripts/XSegUtil.py
View File

@ -52,7 +52,7 @@ def merge(input_dir):
io.log_info(f"No points found in {json_filepath}, skipping.")
continue
dflimg.set_seg_ie_polys (seg_ie_polys)
dflimg.set_seg_ie_polys ( seg_ie_polys.dump() )
dflimg.save()
json_filepath.unlink()