针对有一点mmdetction基础的,然后想根据自己的数据集,熟练训练自己的模型。需要改成自己配置的地方,我会在代码中做好标记,方便修改。 我们先了解一下mmdetection的基本流程,你想训练一个模型,你只需要准备的是:数据集,mmdetection的配置文件。 下面我分为两部分,分别 ...
. 首先确保不加apex前模型能训练起来,精度正常 . 在环境上安装好apex: . 修改模型初始化过程,加上apex初始化 修改文件:mmdet tools train.py 导入apex包 修改amp初始化: 文件修改diff如下: . 修改梯度反向写法 首先弄清楚mmdetection反向的过程,这部分是调用的mmcv实现 runner中实现了训练的过程,以最基本的runner为例,文件: ...
2021-05-16 21:39 4 761 推荐指数:
针对有一点mmdetction基础的,然后想根据自己的数据集,熟练训练自己的模型。需要改成自己配置的地方,我会在代码中做好标记,方便修改。 我们先了解一下mmdetection的基本流程,你想训练一个模型,你只需要准备的是:数据集,mmdetection的配置文件。 下面我分为两部分,分别 ...
v100)。于是开始怀念我熟悉的pytorch+apex,又断断续续花了两天多的时间把整个训练框架用 ...
关键机制:Register注册,HOOK机制,Runner机制 构建Hook这个类,里面有很多函数。创建Hook这个类的一个对象(包含所有函数)。构建Runner这个类和它的一个对象。向Runner这个对象的self的_hook添加对象。利用函数 ...
由于网络优化器使用的是目前各个激光网络开源库Second、Openpcdet等使用的adam_onecycle 无法使用apex.amp进行初始化,应该是无法识别优化器的问题 怎么都无法解决这个问题,最终决定放弃 后面会尝试将torch代码转成pytorch-lightning试试 ...
mmdetection中的模型checkpoints是需要自己手动下载的,下载步骤如下; 打开mmdetection, 进入configs目录,可以看到这里面有很多以目标检测模型命名的文件夹,选择你想使用的模型,打开目录,下面以faster_rcnn为例 进入faster_rcnn目录 ...
一、安装 https://github.com/open-mmlab/mmdetection/blob/master/docs/INSTALL.md 二、训练自己的数据 1、数据 mmdet的默认格式是coco的,这里就以voc格式为例,data下文 ...
📖 首先需要准备好数据集,这里有xml标签数据转voc数据集格式的说明以及免费分享的数据集:xml转voc数据集 - 一届书生 - 博客园 (cnblogs.com) 1. 准备工作目录 我们的工作目录,也就是mmdetection目录,如下所示: configs ...
使用mmdetection训练自己的VOC数据集 首先,是按照官方文档组织VOC的数据集结构; 1.目录结构 mmdetection ├── mmdet ├── tools ├── configs ├── data #手动创建data、VOCdevkit、VOC2007 ...