diff --git a/paddlers/tasks/segmenter.py b/paddlers/tasks/segmenter.py index 1ee8118..3c29e3f 100644 --- a/paddlers/tasks/segmenter.py +++ b/paddlers/tasks/segmenter.py @@ -524,7 +524,7 @@ class BaseSegmenter(BaseModel): for im in images: sample = {'image': im} if isinstance(sample['image'], str): - sample = ImgDecode(to_rgb=False)(sample) + sample = ImgDecoder(to_rgb=False)(sample) ori_shape = sample['image'].shape[:2] im = transforms(sample)[0] batch_im.append(im)