mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-08-20 13:33:24 -07:00
color aug typo fix
This commit is contained in:
parent
296488be6c
commit
2c23b99dfd
1 changed files with 1 additions and 1 deletions
|
@ -200,7 +200,7 @@ class SampleProcessor(object):
|
|||
|
||||
# Apply random color transfer
|
||||
if ct_mode is not None and ct_sample is not None or ct_mode == 'fs-aug':
|
||||
if 'fs-aug':
|
||||
if ct_mode == 'fs-aug':
|
||||
img = imagelib.color_augmentation(img)
|
||||
else:
|
||||
if ct_sample_bgr is None:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue