50预训练权重_MMDetection笔记：修改预训练模型权重类别数

如果想了解MMDetection请点击下面链接：

初识CV：MMDetection中文文档—详解​zhuanlan.zhihu.com

将coco预训练模型类别权重修改类训练数据集权重，代码如下:

如何修改可以看key里面值进行修改：

    for key, value in model_coco["state_dict"].items():
        print(key)

# -*- coding: utf-8 -*-
# @Time    : 20-1-7 上午10:28
# @Author  : wusaifei
# @FileName: Modify_category.py
# @Software: PyCharm
def main():
    #gen coco pretrained weight
    import torch
    num_classes = 11
    mod