將hdf5模型保存成h5模型

resnet_mcnn_model = M_CNN(img_shape, img_shape, mask_shape, num_class)
resnet_mcnn_weight = "D:/workspace/weights-improvement-150-0.9472.hdf5"
resnet_mcnn_model.load_weights(resnet_mcnn_weight)
print('mcnn load weight done-')
resnet_mcnn_model_file = './models/resnetv1_mcnn_mask_1222.h5'
if not os.path.exists(resnet_mcnn_model_file):
    resnet_mcnn_model.save(resnet_mcnn_model_file)

將hdf5模型保存爲h5模型。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章