Unknown layer: AnchorBoxes

Sorry ,I’m s a rookie in keras & Neural network. Pardom,can I ask you a question?

I complete the model training and save model & weights，but when I want to load model next time ， there will be mistakes : ValueError: Unknown layer: AnchorBoxes #load model && load weights model = load_model('./model/ssd7_0.h5') model.load_weights('./model/ssd7_0_weights.h5') I really want to know how to solve this problem，would like to ask about the experienced elders. thankes!

[pierluigiferrari]

@gloddream There definitely is, i just haven’t figured out how yet. i’ve recently implemented the get_config() method in both of the custom layers (L2Normalization and AnchorBoxes), so you should theoretically be able to load the model using

model = load_model('./model/ssd7_0.h5', custom_objects={'AnchorBoxes': AnchorBoxes, 'L2Normalization': L2Normalization, 'SSDLoss': SSDLoss})

but for some reason either model.save or load_model isn’t able to use the dictionaries that get_config() provides correctly. Since Keras documentation regarding saving and loading models with custom objects is basically non-existent and as far as I can tell I implemented the relevant methods in my layers exactly like the Keras core layers implement them, I didn’t bother trying around any longer since I had other stuff to do. If you figure out how to make it work, let me know 🙂

[sachinkmohan]

Hi @pierluigiferrari ! I am so thankful for finding your repository! Great work!!

I know this issue is resolved. But I was trying to apply quantization(tensorflow model optimisation technique) to your repository and again encountered the AnchorBoxes issue. As an author of this repository, I thought you would be the best person to check the issue with. Any inputs from your end will be of great help.

I have detailed this on tensorflow model optimization issues page. Link here

The above mentioned problem is now resolved under tensorflow/model-optimization#620.

All thanks to @Hackerman28 !! 😄