JdeRobot / ObjectDetector

16 stars 14 forks source link

Problem when using full_model.h5 #42

Closed alexandre2r closed 5 years ago

alexandre2r commented 5 years ago

Hi, I am having some problems when loading the provided keras model (full_model.h5). The _loadmodel from Keras is not working and it does not create the model. I printed the exception from line 53 in network.py and it shows bad marshal data (unknown type code). So I searched for the problem and it seems to be a version mismatch when saving and loading models with Keras... My versions of the main libraries needed are the following:

Thanks (I tried the provided Tensorflow model _ssd_mobilenet_v2_coco_2018_0329.pb and it works perfect! Great work!)

alexandre2r commented 5 years ago

Hello again, I realized the problem was that I was using the full model and the application works using just the weights and after that it saves the full model in a .h5. Closing the issue.