Giters

Epiphqny / SOLO

SOLO: Segmenting Objects by Locations

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Bug in Mask-NMS

wjfwzzc opened this issue · comments

commented

There seems to be a bug in the below line.

SOLO/mmdet/models/anchor_heads/solo_head.py

Line 223 in 59a4c03

iou = float(overlap.sum())+1/(float(union.sum())+1)

Might it be the following line?

iou = float(overlap.sum() + 1) / float(union.sum() + 1)

And it might be faster to apply mask-nms on gpu.
image

commented

There seems to be a bug in the below line.

SOLO/mmdet/models/anchor_heads/solo_head.py

Line 223 in 59a4c03

iou = float(overlap.sum())+1/(float(union.sum())+1)

Might it be the following line?

iou = float(overlap.sum() + 1) / float(union.sum() + 1)

And it might be faster to apply mask-nms on gpu.
image

Thank you very much for reporting this bug. I fixed the bug in 28a82d0