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 weightimport torchnum_classes = 11model_coco = torch.load("../checkpoints/cascade_rcnn_r50_fpn_1x_20190501-3b6211ab.pth") # weightfor key, value in model_coco["state_dict"].items():print(key)
####################################################################################### faster_rcnn_r50_fpn_1x_20181010-3d1b3351.pth# model_coco["state_dict"]["bbox_head.fc_cls.weight"] = # model_coco["state_dict"]["bbox_head.fc_cls.weight"][:num_classes, :]### model_coco["state_dict"]["bbox_head.fc_cls.bias"] = # model_coco["state_dict"]["bbox_head.fc_cls.bias"][:num_classes]
###################################################################################### # cascade_rcnn_r50_fpn_1x_20190501-3b6211ab.pthmodel_coco["state_dict"]["bbox_head.0.fc_cls.weight"] = model_coco["state_dict"]["bbox_head.0.fc_cls.weight"][:num_classes, :]model_coco["state_dict"]["bbox_head.1.fc_cls.weight"] = model_coco["state_dict"]["bbox_head.1.fc_cls.weight"][:num_classes, :]model_coco["state_dict"]["bbox_head.2.fc_cls.weight"] = model_coco["state_dict"]["bbox_head.2.fc_cls.weight"][:num_classes, :]model_coco["state_dict"]["bbox_head.0.fc_cls.bias"] = model_coco["state_dict"]["bbox_head.0.fc_cls.bias"][:num_classes]model_coco["state_dict"]["bbox_head.1.fc_cls.bias"] = model_coco["state_dict"]["bbox_head.1.fc_cls.bias"][:num_classes]model_coco["state_dict"]["bbox_head.2.fc_cls.bias"] = model_coco["state_dict"]["bbox_head.2.fc_cls.bias"][:num_classes]# save new modeltorch.save(model_coco, "cascade_rcnn_r50_fpn_1x_coco_pretrained_weights_classes_%d.pth" % num_classes)
if __name__ == "__main__":main()

各个模型下载路径

初识CV：MMDetection中文文档—3.基准和模型汇总zhuanlan.zhihu.com

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

训练dnn_[预训练语言模型专题] MT-DNN(KD) : 预训练、多任务、知识蒸馏的结合
本文为预训练语言模型专题系列第八篇快速传送门 1-4:[萌芽时代].[风起云涌].[文本分类通用技巧].[GPT家族] 5-7:[BERT来临].[浅析BERT代码].[ERNIE合集] 感谢清华大 ...
训练dnn_[预训练语言模型专题] MTDNN(KD) : 预训练、多任务、知识蒸馏的结合
本文为预训练语言模型专题系列第八篇快速传送门 1-4:[萌芽时代].[风起云涌].[文本分类通用技巧] . [GPT家族]5-7:[BERT来临].[浅析BERT代码].[ERNIE合集]感谢清华 ...
[预训练语言模型专题] MT-DNN(KD) : 预训练、多任务、知识蒸馏的结合
本文为预训练语言模型专题系列第八篇快速传送门 1-4:[萌芽时代].[风起云涌].[文本分类通用技巧] . [GPT家族] 5-7:[BERT来临].[浅析BERT代码].[ERNIE合集] 感 ...
《预训练周刊》第9期：TABBIE：表格数据的预训练表示、「视觉预训练神作」：不用图片却训出图像识别SOTA？...
No.09 智源社区预训练组预训练研究观点资源活动关于周刊超大规模预训练模型是当前人工智能领域研究的热点,为了帮助研究与工程人员了解这一领域的进展和资讯,智源社区整理了第9期< ...
最强 NLP 预训练模型库 PyTorch-Transformers 正式开源：支持 6 个预训练框架，27 个预训练模型...
先上开源地址: https://github.com/huggingface/pytorch-transformers#quick-tour 官网: https://huggingface.co/py ...
《预训练周刊》第25期：HyperCLOVA：数十亿级韩语生成式预训练变换器、GPT-3在生物医学领域不是好的小样本学习器...
No.25 智源社区预训练组预训练研究观点资源活动关于周刊超大规模预训练模型是当前人工智能领域研究的热点,为了帮助研究与工程人员了解这一领域的进展和资讯,智源社区整理了第25期&l ...
预训练BERT学习笔记_Storm*Rage的博客
Transformer的Attention Decoder的结构与Encoder相比多了一个Encoder-Decoder Attention,两个Attention分别用于计算输入和输出的权值: S ...
vue项目解决打开新页面预览PDF，同时修改预览PDF页面的标题和图标
今天遇到一个需求,要就打开的pdf预览页面要,放自己的图标和标题. 问过大神后的知通过以下形式可以实现: 在public目录下新建一个html页面, 在新建的html页面的head中的 <tit ...
PyTorch 加载预训练权重
前言使用PyTorch官方提供的权重或者其他第三方提供的权重对相同模型的参数进行初始化,在数据量较少的前提下,可以帮助模型更快地收敛到最优点,达到更好的效果,即迁移学习. 在大部分的迁移学习场景 ...
论文笔记 | code pretraining（代码预训练系列）
文章目录 Pre-trained contextual embedding of source code CodeBERT: A Pre-trained model for programming a ...

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

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

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

各个模型下载路径

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

最新文章

热门文章