From fd7d98e00d89266df03fc3c7cb5cc5d2a4b6a874 Mon Sep 17 00:00:00 2001 From: Jianwen Song Date: Tue, 18 Nov 2025 09:37:41 +1100 Subject: [PATCH] [fix] input_augmentation_bypass for remote inference --- compressai_vision/datasets/image.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/compressai_vision/datasets/image.py b/compressai_vision/datasets/image.py index 1f2ad84..90d64df 100644 --- a/compressai_vision/datasets/image.py +++ b/compressai_vision/datasets/image.py @@ -263,9 +263,13 @@ def __init__(self, root, dataset_name, imgs_folder, **kwargs): if self.input_agumentation_bypass: emptyAugList = AugmentationList([]) - self._org_mapper_func.augmentations = emptyAugList mapper.augmentations = emptyAugList + if hasattr(self._org_mapper_func, "_obj"): + self._org_mapper_func._obj.augmentations = emptyAugList + else: + self._org_mapper_func.augmentations = emptyAugList + self.mapDataset = MapDataset(_dataset, mapper) metaData = MetadataCatalog.get(dataset_name)