github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-21 22:13: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": [
"batch_size",
"batch_size"
],
"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 = 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
if channel_type == SPCT.BGR: