fix solov2 not support multi-images (#6019)

b1857334 · JYChen · GitHub · a6fc3bcf · b1857334
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

deploy/python/infer.py deploy/python/infer.py +1 -1

未找到文件。
--- 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