wulele2的博客

文章目录前言1、Registry介绍2、构建ResNet50为例2.1. 构造ResNet类2.2 利用Registry注册ResNet3、SparseRCNN总结前言  本文主要介绍detectron2如何构建模型。本文将首先介绍注册器，然后介绍如何利用注册器注册模型，最后介绍下构建流程即可。感兴趣可以看下mmdetection中注册器,你会发现这两种优秀框架所用到的设计思想一致。1、Registry介绍  Registry你可以理解成一个能够存储类的字典。比如{‘BackBone’, resne

文章目录前言1、DatasetCatalog和MetadataCatalog对象2、dataset构建1. 通过DatasetCatalog读取数据集2. mapper封装dataset_dict3. 构建dataset总结前言  detectron2中对读入的数据集格式有特定要求，比如说需要将voc和coco格式数据集转成dataset_dict的形式。这么干说可能你也听不明白，还是一步一步来吧。1、DatasetCatalog和MetadataCatalog对象  本次分析代码主要在detect

文章目录前言1、从train.py文件debug开始2、函数装饰器configurable3、合并3.1 from_config函数3.2 总的程序流程总结前言 本篇主要讲解detectron2是如何读取数据集并用dataloader进行包装的。一个目标检测模型往往包含众多参数，那么如何提取出对应数据集的参数呢？detectron2设计了configuable装饰器。因此，本文主要分析下读取过程。细节后续有空在写。1、从train.py文件debug开始  在介绍detectron2的engine中

文章目录前言1、Hook的创建2、train_loop.py2.1 TrainerBase类2.2 SimpleTrainer类3、defaults.py总结前言 本文主要介绍detectron2的engine目录下内容，该目录主要介绍了训练一个模型的思路。1、Hook的创建  上来讲hook可能不太友好，但hook是理解detectron2训练流程的一个关键。首先这是hook的父类。没什么可说的，关键是实现了四个方法。即hook的发生作用的时间节点发生在“训练前，训练后，iter前和iter后”