From d911c56552665185819eb60f7d1957609d3229c1 Mon Sep 17 00:00:00 2001 From: Eugenio Date: Mon, 29 Apr 2024 16:13:56 +0200 Subject: [PATCH] fixed bug that prevented to use center crop --- diffusion_policy/model/vision/multi_image_obs_encoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/diffusion_policy/model/vision/multi_image_obs_encoder.py b/diffusion_policy/model/vision/multi_image_obs_encoder.py index de6aa658..c4f6755e 100644 --- a/diffusion_policy/model/vision/multi_image_obs_encoder.py +++ b/diffusion_policy/model/vision/multi_image_obs_encoder.py @@ -98,7 +98,7 @@ def __init__(self, pos_enc=False ) else: - this_normalizer = torchvision.transforms.CenterCrop( + this_randomizer = torchvision.transforms.CenterCrop( size=(h,w) ) # configure normalizer