Hi,
I am planning to use SSD-MobileNet-v2 model to train my own datasets. I want to do transfer training, so choose SSD-MobileNet-v2 model, but there is a big problem that I cannot find the last classification layers. Since the SSD-MobileNet-v2 model default classes are 80 but mine is 2, I need to change the output layer to fit my case. I tried many way to modify the ClassProb layers, but non of work.
Does anyone can help me locate the layers I need to change? Thank for any help.
Attachments:
