|
@@ -88,7 +88,7 @@ class SegPredictor(BasePredictor):
|
|
|
f"Transformation operators for data preprocessing will be inferred from config file."
|
|
f"Transformation operators for data preprocessing will be inferred from config file."
|
|
|
)
|
|
)
|
|
|
pre_transforms = self.other_src.pre_transforms
|
|
pre_transforms = self.other_src.pre_transforms
|
|
|
- pre_transforms.insert(0, image_common.ReadImage())
|
|
|
|
|
|
|
+ pre_transforms.insert(0, image_common.ReadImage(format="RGB"))
|
|
|
pre_transforms.append(image_common.ToCHWImage())
|
|
pre_transforms.append(image_common.ToCHWImage())
|
|
|
return pre_transforms
|
|
return pre_transforms
|
|
|
|
|
|