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

pre-train use LAB randomization

Browse source
This commit is contained in:
Jeremy Hummel 2019-09-13 01:39:05 -07:00
commit a74e7f44d3
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_SAE/Model.py
View file

@ -469,7 +469,7 @@ class SAEModel(ModelBase):
face_type = t.FACE_TYPE_FULL if self.options['face_type'] == 'f' else t.FACE_TYPE_HALF face_type = t.FACE_TYPE_FULL if self.options['face_type'] == 'f' else t.FACE_TYPE_HALF
global t_mode_bgr global t_mode_bgr
t_mode_bgr = t.MODE_BGR if not self.pretrain else t.MODE_BGR_SHUFFLE t_mode_bgr = t.MODE_BGR if not self.pretrain else t.MODE_LAB_RAND_TRANSFORM
if self.options['random_color_change']: if self.options['random_color_change']:
t_mode_bgr = t.MODE_LAB_RAND_TRANSFORM t_mode_bgr = t.MODE_LAB_RAND_TRANSFORM