mmdetection_object_detection_demo
mmdetection_object_detection_demo copied to clipboard
Change Output layer
Hi, I have my own dataset of 6 objects say a,b,c,d,e,f. Now at the output I want to use two softmax layers. One for the objects a,b,c and one for the objects d,e,f. Could anyone guide me how to do it. Any help would be greatfull.