search

tadax / m2det

M2Det implementation using TensorFlow
MIT License
63 stars 26 forks source link

assign_boxes (ypred, ytrue ) #8

Closed baldmonkeey closed 5 years ago

baldmonkeey commented 5 years ago

Just wanted to check if I am missing anything! Assign boxes seems to add 2 to num classes for gt while m2det seems to add only 1

https://github.com/tadax/m2det/blob/ed2181e1897a58812755eee2f1da774fa6296121/utils/assign_boxes.py#L31

https://github.com/tadax/m2det/blob/ed2181e1897a58812755eee2f1da774fa6296121/m2det.py#L7

baldmonkeey commented 5 years ago

Never mind its being used to update box loss https://github.com/tadax/m2det/blob/ed2181e1897a58812755eee2f1da774fa6296121/utils/loss.py#L66