From 2c23b99dfdd91973d5081cc2a6965efebfb73dbe Mon Sep 17 00:00:00 2001
From: sinofis <ognjen_j@yahoo.com>
Date: Sun, 25 Oct 2020 20:16:32 +0100
Subject: [PATCH] color aug typo fix

---
 samplelib/SampleProcessor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py
index eb515f9..2cfc8b3 100644
--- a/samplelib/SampleProcessor.py
+++ b/samplelib/SampleProcessor.py
@@ -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: