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

XSeg: added pretrain option.

Browse source
This commit is contained in:
iperov 2021-07-30 17:24:21 +04:00
parent 83b1412da7
commit 55b947eab5
3 changed files with 111 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

4
facelib/XSegNet.py
View file

@ -81,8 +81,8 @@ class XSegNet(object):
def get_resolution(self):
return self.resolution
def flow(self, x):
return self.model(x)
def flow(self, x, pretrain=False):
return self.model(x, pretrain=pretrain)
def get_weights(self):
return self.model_weights