WongKinYiu / CrossStagePartialNetworks

Cross Stage Partial Networks
https://github.com/WongKinYiu/CrossStagePartialNetworks
894 stars 172 forks source link

How many layers to extract from a .cfg ? #30

Open LukeAI opened 4 years ago

LukeAI commented 4 years ago

I want to retrain csmobilenet-v2.cfg for a different task starting with the provided weights.

How many layers should I extract from the provided weights?

There are 59 [convolutional] layers and since other layers like max-pooling and routes don't have weights, and since I am changing the number of convolutional filters immediately before the softmax, I guess that the answer is 58?

Like:

./darknet partial csmobilenet-v2.cfg csmobilenet-v2.weights csmobilenet-v2.conv.58 58