如何将Tensorflow模型检查点转换为Pytorch?

时间:2020-05-03 13:17:01

标签: python tensorflow deep-learning pytorch imagenet

我正在使用深度学习模型,该模型具有在ImageNet上预训练的ResNet-50作为主干。我使用的数据集是CUB-200,它是200种鸟类的集合。因此,我认为在具有相似域的数据集上拥有经过预训练的模型可能会很好,而且我发现iNaturalist可能是我正在寻找的模型。

问题是我没有找到任何针对Pytorch的预训练模型,而只有Tensorflow一个here

我尝试使用MDNN库进行转换,但它还需要扩展'.ckpt.meta'文件,而我只有'.ckpt'。

这是如何使用MDNN库将tf模型转换为割炬的示例:

mmconvert -sf tensorflow -in imagenet_resnet_v2_152.ckpt.meta -iw imagenet_resnet_v2_152.ckpt --dstNode MMdnn_Output -df pytorch -om tf_to_pytorch_resnet_152.pth

有人可以帮我吗?

0 个答案:

没有答案