文章目录 前言 1、Registry介绍 2、构建ResNet50为例 2.1. 构造ResNet类 2.2 利用Registry注册ResNet 3、SparseRCNN 总结 前言 本文主要介绍detectron2如何构建模型。本文将首先介绍注册器,然后介绍如何利用注册器注册模型,最后介绍下构建流程即可。感兴趣可以看下mmdetection中注册器,你会发现这两种优秀框架
文章目录前言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后”
文章目录前言1、从train.py文件debug开始2、函数装饰器configurable3、合并3.1 from_config函数3.2 总的程序流程总结前言 本篇主要讲解detectron2是如何读取数据集并用dataloader进行包装的。一个目标检测模型往往包含众多参数,那么如何提取出对应数据集的参数呢?detectron2设计了configuable装饰器。因此,本文主要分析下读取过程。细节后续有空在写。1、从train.py文件debug开始 在介绍detectron2的engine中
文章目录前言1、DatasetCatalog和MetadataCatalog对象2、dataset构建1. 通过DatasetCatalog读取数据集2. mapper封装dataset_dict3. 构建dataset总结前言 detectron2中对读入的数据集格式有特定要求,比如说需要将voc和coco格式数据集转成dataset_dict的形式。这么干说可能你也听不明白,还是一步一步来吧。1、DatasetCatalog和MetadataCatalog对象 本次分析代码主要在detect
Copyright © 2005-2024 51CTO.COM 版权所有 京ICP证060544号