From f06c8b2ed96ac76c84f6d74c888e4b2685b61161 Mon Sep 17 00:00:00 2001 From: liaoxingyu Date: Tue, 25 Aug 2020 11:31:49 +0800 Subject: [PATCH] fix bug: make CJ works #239 fix #239 --- fastreid/data/transforms/build.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastreid/data/transforms/build.py b/fastreid/data/transforms/build.py index 743c8a2..52b8782 100644 --- a/fastreid/data/transforms/build.py +++ b/fastreid/data/transforms/build.py @@ -57,7 +57,7 @@ def build_transforms(cfg, is_train=True): res.extend([T.Pad(padding, padding_mode=padding_mode), T.RandomCrop(size_train)]) if do_cj: - T.RandomApply([T.ColorJitter(cj_brightness, cj_contrast, cj_saturation, cj_hue)], p=cj_prob) + res.append(T.RandomApply([T.ColorJitter(cj_brightness, cj_contrast, cj_saturation, cj_hue)], p=cj_prob)) if do_augmix: res.append(AugMix()) if do_rea: