我正在使用深度学习模型,该模型具有在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
有人可以帮我吗?