Detectron2错误:KeyError:‘model‘

Detectron2错误:KeyError:‘model‘,第1张

我是在使用预训练模型是碰到的该问题:

问题所在:加载的预训练模型是pth格式的,但Detectron2因为你加载了pkl格式的预训练模型。而pkl格式与pth格式的预训练模型不同在于:

pkl是weights['model']['roi_haed.mask_head.predictor.weight']格式的,

pth是weights['roi_haed.mask_head.predictor.weight']格式的。

使用最简单的方法是修改官方的库文件:

点击到checkpoint.py文件:

将:

修改为:

不在pop出model,而是直接使用。不用从numpy转换到torch了。因为pth本身就是torch保存的。

开始完美运行:

 最后简单讲一下我的环境:Detectron2=0.3 python=3.7 torch=1.6

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/758521.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-01
下一篇 2022-05-01

发表评论

登录后才能评论

评论列表(0条)

保存