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')