diff --git a/hapi/vision/transforms/transforms.py b/hapi/vision/transforms/transforms.py index 82f96eefda3837aba80e0742e52cb2a6cb225015..b71b2571bafa23ae3ef58ec943d4e147749332f7 100644 --- a/hapi/vision/transforms/transforms.py +++ b/hapi/vision/transforms/transforms.py @@ -89,8 +89,10 @@ class Compose(object): def __call__(self, *data): for f in self.transforms: try: + # multi-fileds in a sample if isinstance(data, Sequence): data = f(*data) + # single field in a sample, call transform directly else: data = f(data) except Exception as e: