From b185733495f0600c8e178e331f04f98d5d4ae25b Mon Sep 17 00:00:00 2001
From: JYChen <zoooo0820@qq.com>
Date: Mon, 23 May 2022 12:22:17 +0800
Subject: [PATCH] fix solov2 not support multi-images (#6019)

---
 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 368c5b688..b5a52887c 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -224,7 +224,7 @@ class Detector(object):
             for k, v in res.items():
                 results[k].append(v)
         for k, v in results.items():
-            if k != 'masks':
+            if k not in ['masks', 'segm']:
                 results[k] = np.concatenate(v)
         return results
 
-- 
GitLab