From 17f4307d96de08a77212c39ea2f4b66d314d58a4 Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Tue, 15 Sep 2020 15:01:47 +0800
Subject: [PATCH] fix create_inputs of deploy/python/infer.py (#1411)

---
 deploy/python/infer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index 57214f577..7140736dd 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -272,7 +272,7 @@ def create_inputs(im, im_info, model_arch='YOLO'):
     if 'YOLO' in model_arch:
         im_size = np.array([origin_shape]).astype('int32')
         inputs['im_size'] = im_size
-    elif 'RetinaNet' or 'EfficientDet' in model_arch:
+    elif 'RetinaNet' in model_arch or 'EfficientDet' in model_arch:
         scale = scale_x
         im_info = np.array([resize_shape + [scale]]).astype('float32')
         inputs['im_info'] = im_info
-- 
GitLab