未验证 提交 b1857334 编写于 作者: J JYChen 提交者: GitHub

fix solov2 not support multi-images (#6019)

上级 a6fc3bcf
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册