pytorch_to_keras

from pytorch2keras import converter
from torch.autograd import Variable
import torch.onnx

dummy_input = Variable(torch.rand(1, 3, 224, 224))

# #
model = ResNet()
pretrained_net = torch.load('/home/chenyang/PycharmProjects/show_face_decetor/predict_face/models/model_RAFD_facial_50_0.8000.pth')
model.load_state_dict(pretrained_net)

model.eval()
k_model = converter.pytorch_to_keras(model,dummy_input,[(3,224,224,)],verbose=True)

k_model.summary()
#保存模型
k_model.save('my_model.h5')