diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py index ba08c351e8b4f657dfe3f3173644ac1f785c7c38..01890a84d1df0f16533c0b398a408e8bdb029c85 100644 --- a/ppdet/data/transform/operators.py +++ b/ppdet/data/transform/operators.py @@ -824,7 +824,7 @@ class Resize(BaseOperator): im_scale_x = resize_w / im_shape[1] im = self.apply_image(sample['image'], [im_scale_x, im_scale_y]) - sample['image'] = im + sample['image'] = im.astype(np.float32) sample['im_shape'] = np.asarray([resize_h, resize_w], dtype=np.float32) if 'scale_factor' in sample: scale_factor = sample['scale_factor']