Detectron2日积月累
文章平均质量分 52
一些Detectron2使用过程中的笔记和代码分析
蛊惑one
坚持输出!!!
展开
-
手把手教你在Detectron2中搭建自己的Backbone
准备一个Bockbone,我们以最新的RepVGG为例:查看:yaml文件:OUT_FEATURES: [“res2”, “res3”, “res4”, “res5”]修改主干网络,返回一个字典。原创 2022-11-21 16:18:49 · 1933 阅读 · 6 评论 -
DCT-Mask(CVPR2021)核心代码解析
代码主要由两个重要组成部分,分别为dct_mask_head.py和mask_encoding.pydct_mask_head.py# dec_mask_head.py"""类:MaskRCNNDCTHaed方法:__init__():初始化方法from_config():从配置文件中获取一些基本的配置信息layers():构建DCT-Mask的卷积层forward(): 训练和测试内容mask_rcnn_dct_loss():dct损失mask_rcnn_dct_inference(原创 2022-05-11 11:07:32 · 1080 阅读 · 0 评论 -
Pytorch常用代码(不时更新)
主要参考:Link1. 导入torch并查看其版本import torchprint(torch.__version__)2. 随机种子def set_up(seed): torch.manual_seed(seed) torch.cuda.manual_seed_all(seed) np.random.seed(seed) random.seed(seed) torch.backends.cudnn.deterministic = True d原创 2022-05-04 15:35:10 · 1056 阅读 · 0 评论 -
Detectron2错误:KeyError:‘model‘
我是在使用预训练模型是碰到的该问题:问题所在:加载的预训练模型是pth格式的,但Detectron2因为你加载了pkl格式的预训练模型。而pkl格式与pth格式的预训练模型不同在于:pkl是weights['model']['roi_haed.mask_head.predictor.weight']格式的,pth是weights['roi_haed.mask_head.predictor.weight']格式的。使用最简单的方法是修改官方的库文件:点击到checkpoint.py文件原创 2022-04-27 10:25:49 · 7309 阅读 · 0 评论 -
安装指定版本的detectron2-Linux版
我们想要复现的代码往往不是在最新的detectron2上复现,需要我们使用先前版本的detectron2才能复现网络,因此需要重新下载detectron2并配置环境。1、来到detectron2官网地址:https://github.com/facebookresearch/detectron2寻找需要的版本,并下载,这里我们以v0.3为例:2、创建虚拟环境detectron2-v0.3conda create -n detectron2-v0.3 python=3.7创建成功:3、激活原创 2022-04-05 20:31:40 · 3501 阅读 · 2 评论 -
查看detectron2项目的版本号并安装其依赖
1. 查看detectron2版本号我们从github上下载下来的detectron2,经常不知道其版本号,这就导致无法安装正确其正确的torhch和torchvision版本,那么在哪里可以看到detectron2的版本号呢?其位置在:detectron2/__init__.py我们可以看到其版本为0.1.3.2. 来到detectron2官方项目detectron2官方项目链接: link.在INSTALL.md中可以看到所需要的依赖。了解其依赖后,我们可以按照这些规定按照dete原创 2022-04-19 15:15:40 · 1381 阅读 · 0 评论