I am training detectnet_v2 with tlt-launcher using trafficCamNet pretrained model, on a dataset which you can find info about in the result_dataset_convert.txt attached file.
I’ve run several training specifications. You can find two spec_train files attached as well as the respective results they yield.
What I want to highlight is that everything work fine when using the spec_train1.txt file, meaning when using the original classes of the dataset. But when using the spec_train2.txt file, meaning using the classes twowheels, lightvehicle, intermediatevehicule, heavyvehicule and person.
- with twowheels mapping motorcycle and bicycle.
- with heavyvehicule mapping heavyvehicule and bus.
I get an AP of 0 for the twowheels class.
I’ve checked and compared the two spec files and I didn’t find any error in the spec_train2.txt file. Can you confirm me that the concerned spec file is correct ?
And if that’s the case, how can I train a model while grouping the original classes of my dataset together ?