我使用了Keras中的预制模型来训练关于URL的数据集,结果已经保存为.h5格式。我想知道是否可以将保存的.h5文件转换成JSON格式,这样我就可以查看结果并将JSON文件应用到我的系统中,从而将坏的URL与好的URL分开。如果可能的话,我也希望避免重新进行训练。
我使用了Keras中的预制模型来训练关于URL的数据集,结果已经保存为.h5格式。我想知道是否可以将保存的.h5文件转换成JSON格式,这样我就可以查看结果并将JSON文件应用到我的系统中,从而将坏的URL与好的URL分开。如果可能的话,我也希望避免重新进行训练。
tensorflowjs_converter --input_format=keras /tmp/model.h5 /tmp/tfjs_model
首先
!sudo pip3 install tensorflowjs
那么
!sudo tensorflowjs_converter
只需将model.h5替换为您训练好的权重即可
!tensorflowjs_converter --input_format=keras '/content/drive/MyDrive/model.h5' "/content/drive/MyDrive/json"
你已经准备就绪,你的文件将被转换!