diff --git a/timm_image.py b/timm_image.py index bbd441f..e3ce234 100644 --- a/timm_image.py +++ b/timm_image.py @@ -92,7 +92,8 @@ class TimmImage(NNOperator): input_size=self.config['input_size'], interpolation=self.config['interpolation'], mean=self.config['mean'], - std=self.config['std'] + std=self.config['std'], + crop_pct=self.config['crop_pct'] ) self.skip_tfms = skip_preprocess else: