More classes

[curiousKid007]

@Ozzyz @Fredrik00

Trying to create an issue. But since this fork doesn't have that option, creating a PR to reach out.

I was wondering if this branch 'more_classes' is complete and if it can be used for training with multiple classes? I am trying to train KittiBox model on multiple classes (8 classes). Any insights will be helpful.

[Ozzyz]

Hey @curiousKid007 ! I think the evaluation of more classes should work as long as the number of classes are specified correctly, and modify the loss function. In addition, I think the code as it is now uses BDD as its dataset, but this should be easy to modify to suit your needs. I am not 100% sure that this branch is complete, as I haven't personally looked at this code in quite a while, and this code was part of a side project that is not maintained anymore.

Regarding not being able to create issues, this was the default setting on github for forks, and it should be fixed now :)

In order to work on this, I would advice you to look at decoder/fastBox.py where the loss function is modified, and submodules/KittiObjective2/evaluate_object.cpp where the number of classes etc are specified. The latest commit should contain the most correct code, but I can not guarantee this seeing as I haven't worked on the project in ages. Also: this issue was addressed (but not fixced) here

Good luck with your project! Although I have vacation, I will try to keep an eye on this fork just in case you have more issues :)

[curiousKid007]

@Ozzyz : thank you so much for responding. I will try to create an issue, so it becomes more visible to someone who might encounter something similar.

Thank you once again!