Keras预训练模型转Pytorch
Weights_Keras_2_Pytorch
最近想在Pytorch项目里使用一下谷歌的NIMA,但是发现没有预训练好的pytorch权重,于是整理了一下将Keras预训练权重转为Pytorch的代码,目前是支持Keras的Conv2D, Dense, DepthwiseConv2D, BatchNormlization的转换。需要注意的是在Pytorch模型内需要给每一层命名为与Keras每一层相同的名字,才能对应转换。
代码地址:
https://github.com/AgCl-LHY/Weights
原创
2021-12-11 22:15:50 ·
3894 阅读 ·
0 评论