Bladeren bron

[Fix] Fix wrong key

Bobholamovic 3 jaren geleden
bovenliggende
commit
b0610ea04e
1 gewijzigde bestanden met toevoegingen van 1 en 1 verwijderingen
  1. 1 1
      paddlers/tasks/change_detector.py

+ 1 - 1
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)