MMDet加载旧版本代码权重及自定义数据集的问题及解决

文章首发见博客:https://mwhls.top/4881.html。
无图/格式错误/后续更新请见首发页。
更多更新请到mwhls.top查看
欢迎留言提问或批评建议,私信不回。

场景
  • PyTorch说它们2.0很快,那么就安个最新的Torch和MMDet。
  • 打开两年前的顶会文章,复制更早几年的代码文件以及权重。
  • 找个数据集。
  • 按照最新的config进行配置权重路径。
  • 然而,找不到加载权重。
解决
  • config 配置
    # 权重加载为 init_cfg=dict(checkpoint='pretrained/p2t_tiny.pth'),和仓库中其它config一样,区别在于一个是网上链接,一个是本地路径。
    model = dict(
        type='Retina

你可能感兴趣的:(python,python,pytorch,mmdet)