markshih91/mobilenet_v2_ssdlite_keras

How sub_sample the weights?

Opened this issue · 5 comments

zhoujinhai commented

@markshih91, I want to training the model on my dataset that have two classess, how to sub-sample the weight tensors of all the classification layers?

DubiousCactus commented

The original writer of this SSD implementation has a very nice tutorial for that: https://github.com/pierluigiferrari/ssd_keras/blob/master/weight_sampling_tutorial.ipynb

zhoujinhai commented

yes, I have tried it, and change the layers name from conv4_3_norm_mbox_conf to ssd_cls1conv2, but it show that bias:0 does not exist after running.

DubiousCactus commented

Well I can't help you, I haven't tried it myself...

zhoujinhai commented
liguilan1227 commented

yes, I have tried it, and change the layers name from conv4_3_norm_mbox_conf to ssd_cls1conv2, but it show that bias:0 does not exist after running.

I met the problem,too. Have you solve it? Please help me.