valeoai / LOST

Pytorch implementation of LOST unsupervised object discovery method
Other
234 stars 26 forks source link

Doesn't save boxes #3

Closed bradezard131 closed 2 years ago

bradezard131 commented 2 years ago

The predicted boxes are never assigned to preds_dict and therefore it just saves an empty dictionary

osimeoni commented 2 years ago

Thanks @bradezard131 for spotting the issue. I have pushed a fix.