我当时是手动一层一层的对应输出,用pytorch复现了一遍,不过结果会有误差,自己随机一个输入,每次输出大小,用pytorch再搞一边,其他方法都不靠谱,还有就是论文中的话github可能有人复现了,多找找,backbone啥的pytorch自带
根据他定义的模型的各个层在pytorch里面调用相应的api实现相应的层就行了,但需要题主了解一点pytorch模型构建的知识和语法
不用转换,你直接找你的网络对应的torch版本,比如vgg16这些,很多网上都有torch的版本的
参考:
https://gereshes.com/2019/06/24/how-to-transfer-a-simple-keras-model-to-pytorch-the-hard-way/
import torch.nn as nn
下面是代码