search

flyingshan / Learning-Polar-Encodings-For-Arbitrary-Oriented-Ship-Detection-In-SAR-Images

Code for paper "Learning-Polar-Encodings-For-Arbitrary-Oriented-Ship-Detection-In-SAR-Images"
MIT License
24 stars 7 forks source link

assertion error #3

Closed Fly-dream12 closed 2 years ago

Fly-dream12 commented 3 years ago

Thanks for you code, when i train the model, this error appears: in decoder.py in polar_decode assert num_targets >0 This happens in the evaluation process, can you give some suggestions?

flyingshan commented 3 years ago

I think it is caused by the function in . When there is no target predicted, the function will fail. But recently I have no time to fix this bug, if you help fix the bug, you can give a pull request here. p.s. I think you are using your personal dataset? if you use the SSDD dataset, I think the bug will not appear.

Fly-dream12 commented 3 years ago

I'm using the SSDD dataset. So what may be wrong? @flyingshan

flyingshan commented 3 years ago

Well, I updated the README page just now, added some explanations and updated the training and testing script. See if it can help you.