diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py index 880f6936c926fdf437503ccf166b3eb8a780fbac..ae7023d799256dfd5a7e60c290b0ffef15c9415a 100644 --- a/ppdet/data/transform/operators.py +++ b/ppdet/data/transform/operators.py @@ -204,6 +204,8 @@ class DecodeCache(BaseOperator): sample['im_shape'] = np.array(im.shape[:2], dtype=np.float32) sample['scale_factor'] = np.array([1., 1.], dtype=np.float32) + sample.pop('im_file') + return sample @staticmethod