From 31a9d914884da23ea993058da7847abe6a072e05 Mon Sep 17 00:00:00 2001 From: Yashal Shakti Kanungo Date: Wed, 1 Nov 2023 16:28:02 -0700 Subject: [PATCH] Fix AddOriginalImageSizeAsTupleAndCropToSquare AbstractMapper to use the provided data_key Fix AddOriginalImageSizeAsTupleAndCropToSquare AbstractMapper to use the provided data_key --- sdata/mappers/sample_mappers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdata/mappers/sample_mappers.py b/sdata/mappers/sample_mappers.py index 82a5918..193cf1d 100644 --- a/sdata/mappers/sample_mappers.py +++ b/sdata/mappers/sample_mappers.py @@ -152,7 +152,7 @@ def __call__(self, x: Dict) -> Dict: if self.skip_this_sample(x): return x if self.use_data_key: - h, w = map(lambda y: x["json"][y], (self.h_key, self.w_key)) + h, w = map(lambda y: x[self.data_key][y], (self.h_key, self.w_key)) else: h, w = map(lambda y: x[y], (self.h_key, self.w_key)) x["original_size_as_tuple"] = torch.tensor([h, w])