Not training for other classes for object detection

[kshitij3112]

First of all, thanks for sharing the code. I have been trying to train for the traffic light and traffic sign classes for object detection. I have frozen the segmentation branch and changed the obj_list and obj_combine as follows:

obj_list: ['traffic light', 'traffic sign'] obj_combine: []

However the network does not seem to train. Could you please provide some insight on what I could be missing?

I have also set the conf_thres to 0.5. Hoping for a reply soon.

Thanks in advance.