diff --git a/deploy/python/infer.py b/deploy/python/infer.py index 368c5b6887b7d4f47c9ac795e9ff68026f7fcaea..b5a52887caa7e00b7f8167231a2da905daeee89e 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