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

fix

Browse source
This commit is contained in:
iperov 2021-05-29 22:23:07 +04:00
commit 11a7993238
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
models/Model_AMP/Model.py
View file

@ -732,9 +732,9 @@ class AMPModel(ModelBase):
i = np.random.randint(n_samples) i = np.random.randint(n_samples)
st = [ np.concatenate ((S[i], D[i], DD[i]*DDM_000[i]), axis=1) ] st = [ np.concatenate ((S[i], D[i], DD[i]*DDM_000[i]), axis=1) ]
st += [ np.concatenate ((SS[i], DD[i], SD_065[i] ), axis=1) ] st += [ np.concatenate ((SS[i], DD[i], SD_075[i] ), axis=1) ]
result += [ ('AMP morph 0.65', np.concatenate (st, axis=0 )), ] result += [ ('AMP morph 0.75', np.concatenate (st, axis=0 )), ]
st = [ np.concatenate ((DD[i], SD_025[i], SD_050[i]), axis=1) ] st = [ np.concatenate ((DD[i], SD_025[i], SD_050[i]), axis=1) ]
st += [ np.concatenate ((SD_065[i], SD_075[i], SD_100[i]), axis=1) ] st += [ np.concatenate ((SD_065[i], SD_075[i], SD_100[i]), axis=1) ]
@ -756,7 +756,7 @@ class AMPModel(ModelBase):
#override #override
def get_MergerConfig(self): def get_MergerConfig(self):
morph_factor = np.clip ( io.input_number ("Morph factor", 0.65, add_info="0.0 .. 1.0"), 0.0, 1.0 ) morph_factor = np.clip ( io.input_number ("Morph factor", 0.75, add_info="0.0 .. 1.0"), 0.0, 1.0 )
def predictor_morph(face): def predictor_morph(face):
return self.predictor_func(face, morph_factor) return self.predictor_func(face, morph_factor)