导入训练好的权重文件时,报如下错误:
错误1:
File "/home/dongli/.conda/envs/tf2/lib/python3.7/site-packages/tensorflow_core/python/keras/saving/hdf5_format.py", line 165, in load_model_from_hdf5
raise ValueError('No model found in config file.')
ValueError: No model found in config file.
错误2:
ValueError: Cannot create group in read-only mode.
再重新训练自己的数据集以后,保存的h5模型在读取时出现Cannot create group in read only mode.的错误,通过查询发现是在模型保存时的错误,在保存时使用了model.save_weights。改为save就可以了。
model.save_weights和model.save()的区别
save()保存的模型结果,它既保持了模型的图结构,又保存了模型的参数。
save_weights()保存的模型结果,它只保存了模型的参数,但并没有保存模型的图结构。
由save()保存下来的h5文件才可以直接通过load_model()打开!
save_weights()保存的模型不含有模型结构信息,所以我们需要把模型结构再描述一遍才可以加载
keras保存模型时,使用的是m.save_weights,只保存了模型的权重而没有结构,直接使用keras.models.load_model就会报错。