From b0610ea04e8ee5423ca22f501f9c33848fe16433 Mon Sep 17 00:00:00 2001 From: Bobholamovic Date: Sat, 30 Apr 2022 12:41:17 +0800 Subject: [PATCH] [Fix] Fix wrong key --- paddlers/tasks/change_detector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddlers/tasks/change_detector.py b/paddlers/tasks/change_detector.py index 64d3fd9..f5c9586 100644 --- a/paddlers/tasks/change_detector.py +++ b/paddlers/tasks/change_detector.py @@ -539,7 +539,7 @@ class BaseChangeDetector(BaseModel): if isinstance(sample['image_t1'], str) or \ isinstance(sample['image_t2'], str): sample = ImgDecoder(to_rgb=False)(sample) - ori_shape = sample['image'].shape[:2] + ori_shape = sample['image_t1'].shape[:2] im1, im2 = transforms(sample)[:2] batch_im1.append(im1) batch_im2.append(im2)