pytorch_to_keras

pytorch2kerasインポートコンバータから
torch.autogradインポート変数から
の輸入torch.onnx

dummy_input =可変(torch.rand(1、3、224、224))

モデル= ResNet()
pretrained_net = torch.load( '/ホーム/ chenyang / PycharmProjects / show_face_decetor / predict_face /モデル/ model_RAFD_facial_50_0.8000.pth')
model.load_state_dict(pretrained_net)

model.eval()
k_model = converter.pytorch_to_keras(モデル、dummy_input、[(3224224)]、冗長= TRUE)

k_model.summary()
モデルの保存#
k_model.save( 'my_model.h5')

公開された172元の記事 ウォン称賛52 ビュー40000 +

おすすめ

転載: blog.csdn.net/weixin_32759777/article/details/103930740