Commit 22d25bed authored by WXinlong's avatar WXinlong
Browse files

fix matrix_nms

parent 17f46d9f
......@@ -18,7 +18,7 @@ def matrix_nms(seg_masks, cate_labels, cate_scores, kernel='gaussian', sigma=2.0
n_samples = len(cate_labels)
if n_samples == 0:
return []
if sum_masks == None:
if sum_masks is None:
sum_masks = seg_masks.sum((1, 2)).float()
seg_masks = seg_masks.reshape(n_samples, -1).float()
# inter.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment