mahyarnajibi / SNIPER

SNIPER / AutoFocus is an efficient multi-scale object detection training / inference algorithm
Other
2.69k stars 451 forks source link

Where to define # of superclasses in R-FCN-3k? #124

Open igygi opened 5 years ago

igygi commented 5 years ago

Hi,

According to the R-FCN-3k paper, we can group classes into superclasses by clustering closely related classes. May I ask where in the R-FCN-3k code can I set the preferred # of superclasses? I can't seem to find it. Thank you.

Lancelot365 commented 5 years ago

Its hardcoded to 2, you will have to make significant changes (in /dataset, /iterators, /symbols), which I did, to change the number of super class