我正在使用一个深度学习模型,该模型以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
有人能帮我吗
目前没有回答
相关问题 更多 >
编程相关推荐