From 14e813281fcfccd002cc4dd73b872704f4807f69 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Mon, 30 Jan 2023 14:35:29 +0800 Subject: [PATCH] Add transform config Signed-off-by: Jael Gu --- timm_image.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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: