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

Merge branch 'master' into config_updates

Browse source
This commit is contained in:
Jan 2021-12-09 16:06:02 +01:00
commit 2c911ce234
2 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_Quick96/config_schema.json
View file

@ -12,7 +12,7 @@
} }
}, },
"required": [ "required": [
"batch_size", "batch_size"
], ],
"title": "dfl_config" "title": "dfl_config"
} }

2
samplelib/SampleProcessor.py
View file

@ -266,6 +266,8 @@ class SampleProcessor(object):
img_v = np.clip (img_v + (rnd_state.random()-0.5)*a, 0, 1 ) img_v = np.clip (img_v + (rnd_state.random()-0.5)*a, 0, 1 )
img = np.clip( cv2.cvtColor(cv2.merge([img_h, img_s, img_v]), cv2.COLOR_HSV2BGR) , 0, 1 ) img = np.clip( cv2.cvtColor(cv2.merge([img_h, img_s, img_v]), cv2.COLOR_HSV2BGR) , 0, 1 )
img = imagelib.warp_by_params (warp_params, img, warp, transform, can_flip=True, border_replicate=border_replicate)
img = np.clip(img.astype(np.float32), 0, 1)
# Transform from BGR to desired channel_type # Transform from BGR to desired channel_type
if channel_type == SPCT.BGR: if channel_type == SPCT.BGR: