瀏覽代碼

[Fix] Fix wrong key

Bobholamovic 3 年之前
父節點
當前提交
b0610ea04e
共有 1 個文件被更改,包括 1 次插入1 次删除
  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)