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

2024-05-16 05:10:55 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用一个深度学习模型,该模型以ResNet-50为主干,在ImageNet上进行预训练。我使用的数据集是CUB-200,这是一组200种鸟类。出于这个原因,我认为在一个具有类似域的数据集上有一个预训练的模型是很好的,我发现不自然的模型可能就是我正在寻找的那个

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

我试图使用MDNN库转换它,但它还需要“.ckpt.meta”文件扩展名,我只有“.ckpt”

这是一个如何使用MDNN库将tf模型转换为torch的示例:

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

有人能帮我吗


Tags: 数据模型鸟类tfpytorchmetav2resnet