Fix the error of transforms in deploy (PaddlePaddle#2490) (PaddlePadd…

…le#2492)
skuley · Aug 29, 2022 · 6fabf30 · 6fabf30
1 parent 5a154b7
commit 6fabf30
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/deploy/python/collect_dynamic_shape.py b/deploy/python/collect_dynamic_shape.py
@@ -86,7 +86,8 @@ def collect_dynamic_shape(args):
  # collect
  progbar_val = progbar.Progbar(target=len(img_path_list))
  for idx, img_path in enumerate(img_path_list):
- data = np.array([cfg.transforms(img_path)[0]])
+ data = {'img': img_path}
+ data = np.array([cfg.transforms(data)['img']])
  input_handle.reshape(data.shape)
  input_handle.copy_from_cpu(data)
 

diff --git a/deploy/python/infer.py b/deploy/python/infer.py
@@ -176,7 +176,8 @@ def auto_tune(args, imgs, img_nums):
 
  for i in range(0, num):
  if isinstance(imgs[i], str):
- data = np.array([cfg.transforms(imgs[i])[0]])
+ data = {'img': imgs[i]}
+ data = np.array([cfg.transforms(data)['img']])
  else:
  data = imgs[i]
  input_handle.reshape(data.shape)