|
class Normalize(object): |
|
def __init__(self, mean, std, to_bgr255=True): |
|
self.mean = mean |
|
self.std = std |
|
self.to_bgr255 = to_bgr255 |
|
|
|
def __call__(self, image, seg=None): |
|
if self.to_bgr255: |
|
image = image[[2, 1, 0]] * 255 |
|
image = F.normalize(image, mean=self.mean, std=self.std) |
|
seg = 255 * seg[0] if seg is not None else None |
|
return image, seg |
I guess to_bgr255 is used for Caffe2. This should be set to False.
ViTAA/vitaa/data/transforms/transforms.py
Lines 94 to 105 in 0bd7638
I guess to_bgr255 is used for Caffe2. This should be set to False.