TensorFlow Object Detection Api: Classification Weights Initialization When Changing Number Of Classes At Training Using Pre-trained Models
I want to utilize not only the feature-extractor pre-trained weights but also the feature-map layers' classifier/localization pre-trained weights for fine-tuning tensorflow object
Solution 1:
As I read through the code I found the responsible code, which only retains the pre-trained model's weights if the shape of the layers between the newly-defined model and the pre-trained model match. So if I change the number of the class, the shape of the classifier layers change, and the pre-trained weights are not retained.
Baca Juga
- How To Classify Both Sentiment And Genres From Movie Reviews Using Cnn Tensorflow
- Valueerror When Loading A Previously Saved Retrained Vgg16 Model Using Keras
- Valueerror: Input 0 Of Layer Sequential_16 Is Incompatible With The Layer: Expected Ndim=5, Found Ndim=4. Full Shape Received: [none, 224, 224, 3]
Post a Comment for "TensorFlow Object Detection Api: Classification Weights Initialization When Changing Number Of Classes At Training Using Pre-trained Models"