diff --git a/pdseg/reader.py b/pdseg/reader.py index 6fe395d5a003b2aeeffb7eaafc088c03c1802c1d..2e8f3908011eaf88c791adcf24a98e52d8d14392 100644 --- a/pdseg/reader.py +++ b/pdseg/reader.py @@ -318,6 +318,7 @@ class SegDataset(object): raise ValueError("Dataset mode={} Error!".format(mode)) # Normalize image + img = img[...,::-1] img = self.normalize_image(img) if ModelPhase.is_train(mode) or ModelPhase.is_eval(mode):