目录 模型训练流程 hook 模型训练流程 从 tools/train.py 开始: 一通读取 cfg ,初步设置一些基本参数,log 参数; build 模型,build 数据集 (有多少个 ...
关键机制:Register注册,HOOK机制,Runner机制 构建Hook这个类,里面有很多函数。创建Hook这个类的一个对象 包含所有函数 。构建Runner这个类和它的一个对象。向Runner这个对象的self的 hook添加对象。利用函数的名字hook name去调用这个Hook对象的这个函数。 需要修改model部分,model是得出结果的部分。主要是利用config里面的那些配置去创建 ...
2020-11-23 22:04 0 390 推荐指数:
目录 模型训练流程 hook 模型训练流程 从 tools/train.py 开始: 一通读取 cfg ,初步设置一些基本参数,log 参数; build 模型,build 数据集 (有多少个 ...
1. 首先确保不加apex前模型能训练起来,精度正常 2. 在环境上安装好apex: 3. 修改模型初始化过程,加上apex初始化 修改文件:mmdet/tools/train.py 导入apex包 修改amp初始化: 文件修改diff ...
一、安装 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 ...
。 选用新的backbone来训练 这里选用 res2net_26w_6s 作为backbone; 以c ...
本文将快速引导使用 MMDetection ,记录了实践中需注意的一些问题。 环境准备 基础环境 Nvidia 显卡的主机 Ubuntu 18.04 系统安装,可见 制作 USB 启动盘,及系统安装 Nvidia Driver 驱动安装 ...
(1)查看cuda版本: nvcc -V gcc --version (2) 安装 PyTorch and torchvision from torch. ...