add bgr to rgb

22d86982 · chenguowei01 · 5d19dfa6 · 22d86982
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

pdseg/reader.py pdseg/reader.py +1 -0

未找到文件。
--- 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):