Add transform config

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 14e813281f
1 changed files with 2 additions and 1 deletions
--- 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: