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

color aug typo fix

Browse source
This commit is contained in:
sinofis 2020-10-25 20:16:32 +01:00
commit 2c23b99dfd
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
samplelib/SampleProcessor.py
View file

@ -200,7 +200,7 @@ class SampleProcessor(object):
# Apply random color transfer # Apply random color transfer
if ct_mode is not None and ct_sample is not None or ct_mode == 'fs-aug': 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) img = imagelib.color_augmentation(img)
else: else:
if ct_sample_bgr is None: if ct_sample_bgr is None: