Mask_RCNN icon indicating copy to clipboard operation
Mask_RCNN copied to clipboard

How to save weights and models

Open mengxiabing opened this issue 6 years ago • 9 comments

How to save weights and models

mengxiabing avatar Mar 14 '18 08:03 mengxiabing

Weights are saved after each training epoch inside logs folder (.h5 files). You can check https://keras.io/getting-started/faq/ (Saving/loading only a model's weights section) for further info.

fastlater avatar Mar 15 '18 05:03 fastlater

@fastlater Thank you. I need a model of the network structure, not just a weight file. How can I export it?

mengxiabing avatar Mar 15 '18 07:03 mengxiabing

@mengxiabing in the same link, you will find how to save the architecture of the model.

In model.py, you can see that they used keras so you need to follow the instructions from keras about how to save the model. I am not sure if there is a method inside this code to save the model architecture.

fastlater avatar Mar 15 '18 07:03 fastlater

It appears the model cannot be easily saved. I encountered this error when trying to save MaskRCNN model: https://github.com/keras-team/keras/issues/10140

CMCDragonkai avatar May 11 '18 03:05 CMCDragonkai

How to save weights and models

Now,is the question you face resolved? if “yes” ,please tell me how to solve it thanks

lovebabychen avatar Oct 17 '18 12:10 lovebabychen

Run into the same thing, anyone knows?

shen338 avatar Dec 13 '18 23:12 shen338

I am facing the same problem. Anyone know about it, please help me..

AI-ML-Enthusiast avatar Jul 12 '19 08:07 AI-ML-Enthusiast

how to save mask rcnn cfg and weights file to serve the model using tensorflow serving?

NayanDharviya avatar Nov 27 '20 06:11 NayanDharviya

Weights are saved after each training epoch inside logs folder (.h5 files). You can check https://keras.io/getting-started/faq/ (Saving/loading only a model's weights section) for further info.

the saved weights of the model don't appear for me after each epoch, but only after some selected ones (only epochs 1 and 3 for example), has this happened to you? does anybody know how to fix it?

avnerst avatar Jun 01 '22 12:06 avnerst