[Fix] Fix wrong key

3 years ago · b0610ea04e
parent 88a27ef3d5
commit b0610ea04e
1 changed files with 1 additions and 1 deletions
--- 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)