HybridNets
HybridNets copied to clipboard
Not training for other classes for object detection
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.