yolo2-pytorch icon indicating copy to clipboard operation
yolo2-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon longcw

_ious[cell_ind, a, :] = ious_reshaped[cell_ind, a, i]

Open 22wei22 opened this issue 7 years ago • 1 comments

Ask for help,in your code,you use _ious[cell_ind, a, :] = ious_reshaped[cell_ind, a, i], I think _ious[cell_ind, a, :] = anchor_ious[a, i]

22wei22 avatar Mar 26 '18 07:03 22wei22

No, the predicted IOU should be compared to the IOU with the groundtruth box, not the anchor box. I'm vetting this code fairly extensively right now, and I believe this is right.

See: https://github.com/longcw/yolo2-pytorch/issues/19

Erotemic avatar Mar 26 '18 18:03 Erotemic