kuangliu / pytorch-retinanet

RetinaNet in PyTorch
993 stars 249 forks source link

Clone boxes tensor to avoid mutating ground truth #19

Closed tarrencev closed 6 years ago

tarrencev commented 6 years ago

This resolves an issue where an items bounding box ground truths were being assigned by reference and mutated each time the datum is encoded. This resulted in inputs eventually having 0 width, so there would be no iou matches and loc loss would go to 0

kuangliu commented 6 years ago

Wow, that's a big one. THANKS a lot.