提交 22d86982 编写于 作者: C chenguowei01

add bgr to rgb

上级 5d19dfa6
...@@ -318,6 +318,7 @@ class SegDataset(object): ...@@ -318,6 +318,7 @@ class SegDataset(object):
raise ValueError("Dataset mode={} Error!".format(mode)) raise ValueError("Dataset mode={} Error!".format(mode))
# Normalize image # Normalize image
img = img[...,::-1]
img = self.normalize_image(img) img = self.normalize_image(img)
if ModelPhase.is_train(mode) or ModelPhase.is_eval(mode): if ModelPhase.is_train(mode) or ModelPhase.is_eval(mode):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册