search

liulei01 / DRBox

A deep learning based algorithm to detect rotated object, for example, objects in remote sensing images
Other
421 stars 142 forks source link

how to detect airplane, ship and vehicle all together as a multiclass problem #13

Open arasharchor opened 6 years ago

arasharchor commented 6 years ago

Hi,

I understand that you have trained separate networks for each object class. I am trying to train a network on these three classes together as a multi-class detection task, but I am stuck in changing prototxts. Any idea?

Best,

evercherish commented 4 years ago

Hi, have you solved this problem? Could you please tell me how to train a multi-classes model? Thanks!

evercherish commented 4 years ago

No I spent time on creating another algorithm for multi-class detection. I had to train on a custom dataset which we got from https://ternowai.eu/ so I did not focus on DRBox

thanks for your reply!