From 1abb4b8bf6498be3bb418fd3f0cdfd2c6180eba9 Mon Sep 17 00:00:00 2001
From: Wenyu <wenyu.lyu@gmail.com>
Date: Wed, 13 Oct 2021 17:12:33 +0800
Subject: [PATCH] pop str im_file (#4299)

---
 ppdet/data/transform/operators.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py
index 880f6936c..ae7023d79 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
-- 
GitLab