From 8f2e0946398b05b894482d605507163c7d93e780 Mon Sep 17 00:00:00 2001
From: KhubaibKhan <khubaibtakkar@gmail.com>
Date: Mon, 24 Jun 2024 13:57:44 +0900
Subject: [PATCH] Fix KeyError in MultiScaleFlipAug by ensuring correct key
handling
---
mmcv/transforms/processing.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mmcv/transforms/processing.py b/mmcv/transforms/processing.py
index 8b65b112b6..544e19f0ee 100644
--- a/mmcv/transforms/processing.py
+++ b/mmcv/transforms/processing.py
@@ -861,7 +861,7 @@ def transform(self, results: dict) -> Dict:
inputs.append(packed_results['inputs']) # type: ignore
data_samples.append(
- packed_results['data_sample']) # type: ignore
+ packed_results['data_samples']) # type: ignore
return dict(inputs=inputs, data_sample=data_samples)
def __repr__(self) -> str: