首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

在mmdetection中运行train_detector后如何保存模型权重?

在mmdetection中运行train_detector后,可以通过设置相关参数来保存模型权重。具体步骤如下:

  1. 在训练配置文件(通常是一个.py文件)中,找到checkpoint_config部分,设置save_best参数为True,表示只保存最好的模型权重。
代码语言:txt
复制
checkpoint_config = dict(
    # 设置保存最好模型权重的参数
    save_best=True,
    # 设置保存模型权重的路径和文件名
    filepath='path/to/save/model.pth',
    # 设置保存模型权重的间隔epoch数
    interval=1
)
  1. 运行train_detector命令时,通过--work_dir参数指定保存模型权重的路径。
代码语言:txt
复制
python tools/train.py <config_file> --work_dir <save_dir>

其中,<config_file>是训练配置文件的路径,<save_dir>是保存模型权重的路径。

  1. 在训练过程中,每个epoch结束后,mmdetection会自动保存模型权重到指定的路径。
  2. 如果设置了save_best=True,则只会保存在验证集上表现最好的模型权重。

总结起来,通过在训练配置文件中设置checkpoint_config参数,并在运行train_detector命令时指定保存路径,mmdetection会在训练过程中自动保存模型权重。

页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

Keras实现保存和加载权重模型结构

model_from_json(json_string, custom_objects={‘AttentionLayer’: AttentionLayer}) 2019年6月1号更新: 更详细的使用方法: 如何保存...(1)一个HDF5文件即保存模型的结构又保存模型权重 我们不推荐使用pickle或cPickle来保存Keras模型。...你可以使用model.save(filepath)将Keras模型权重保存在一个HDF5文件,该文件将包含: 模型的结构,以便重构该模型 模型权重 训练配置(损失函数,优化器等) 优化器的状态,以便于从上次训练中断的地方开始...如果需要保存模型权重,可通过下面的代码利用HDF5进行保存。...实现保存和加载权重模型结构就是小编分享给大家的全部内容了,希望能给大家一个参考。

3K20

【目标检测】MMDetection的安装与基础使用

MMDetection仅仅是多个Codebases的其中一个,除此之外,还有专用于图像分类的MMClassification,用于目标追踪的MMTracking等。...显示图片 官方仓库的demo文件夹下,提供了三个示例教程,这里就跟着来跑一下。 首先需要将源码的configs文件夹复制到测试工程路径下,这个文件夹包含了各种算法的结构。...首先需要根据官方教程指引,下载faster_rcnn的权重文件faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth,放置checkpoints文件夹下。...init_detector这个函数,注意模型的config_file文件和模型权重必须对应。...第三个示例关于模型训练,使用的是kitti_tiny数据集 训练过程不做详解,各注释已写在代码

1.1K20

一日多技,技能-Linux-Python 006

” 创建镜像如果快速运行镜像,docker run 有很多参数,只有少部分会修改,如果每天创建很多容器,每次都完整输入一次命令就很麻烦,容易出错,这时我们可以写一个 shell 脚本,把会修改的值变成参数...脚本 0 表示脚本的文件名称,$1 开始才是我们指定的参数。...操作起来 用 vim 新建一个run.sh文件,将上面的脚本内容写入,然后保存文件,再给文件赋予可执行权限。...unzip -O cp936 案例2资源-1663747915.zip Python “1、'ConfigDict' object has no attribute 'device' ” 运行 mmdetection...demo的时候出现标题中错误, mmdetection/demo/MMDet_InstanceSeg_Tutorial.ipynb 的Train a new detector的第一个代码块 加入:

77720

ubuntu安装mmdetection

就会按照 py 文件的配置找到相对应的模型配置、训练策略配置以及数据集配置,然后就会开始训练,训练的日志以及模型保存在 work_dirs 这个文件夹。...show_dirs 用来保存处理的图片样本 CUDA_VISIBLE_DEVICES=2 python tools/test.py configs/mask_rcnn/mask_rcnn_r50_fpn...BaseDetector 当中实现了 train_step 方法,这里用了 self(**data),self 指的是自己本身,所以就调用了 __call__ 方法, nn.Module 模型,_...forward_train 函数 SingleStageDetector ,是如下结构,获取特征让 bbox_head 进行 forward_train ,所以后面还得去看 bbox_head...贴一下官网的例子(mmcv 出 bug 了,命令行里运行不了,等他们解决完 bug 再来更新) 跟下面这样子的话呢就可以注册一个 converter 大类,假设是 converters/builder.py

84860

业界 | Facebook开源Mask R-CNN的PyTorch 1.0基准,比mmdetection更快、更省内存

节省内存:训练过程中使用的 GPU 内存比 mmdetection 少大约 500MB; 使用多 GPU 训练和推理; 批量化推理:可以每 GPU 每批量上使用多张图像进行推理; 支持 CPU 推理...:可以推理时间内于 CPU 上运行。...所有的基线模型都使用了和 Detectron 相同的实验设置,检测模型权重使用 Caffe2 的 ImageNet 权重初始化,这和 Detectron 是一样的。...预训练模型通过下表的 model id 链接获取。 ? 和 Detectron、mmdetection 的性能对比 训练速度 下表的数据单位是秒/迭代,越低越好。...(mmdetection 备注的硬件和 maskrcnn_benchmark 是不同的) ? 训练内存(越低越好) ? 推理准确率(越高越好) ?

1.1K30

轻松搭建主干网络,多种视觉任务一网打尽

这里,以我们最为常见的跨任务研究——搭建主干网络为例,介绍一下如何快速开发一个新的主干网络,并分别使用 MMClassification、MMDetection 和 MMSegmentation 轻松实现分类...这些基础配置文件一般包括模型、数据集、优化策略和运行配置。你也可以配置文件对基础配置文件进行覆盖,或者不使用基础配置文件,全部写到一个配置文件里。...MMDetection 和 MMSegmentation 的配置文件额外指定主干网络注册器的位置,这只需要在 type 添加 mmcls....首先,我们需要将当前仓库的目录添加到 PYTHONPATH 环境变量,这样 Python 才可以找到我们的模型文件,仓库根目录运行如下指令: export PYTHONPATH=`pwd`:$PYTHONPATH...: - $CONFIG:configs/ 文件夹的配置文件 - $WORK_DIR:用于保存日志和权重文件的文件夹 - $CHECKPOINT:权重文件路径 单机多 GPU (以 4 GPU 为例)

95931

业界 | OpenMMLab 第二版发布:吸引业界「目光」的史上最完整的目标检测工具箱

关于新版 MMDetection AI 科技评论按:去年 10 月,中国香港中文大学-商汤联合实验室 OpenMMLab 的首版计划,开放了 MMCV 和 MMDetection(初期版本...此后,MMDetection 逐渐发展成为一个统一的平台,涵盖了许多主流检测方法和现代模块。它不仅包括训练和推理代码,而且还为 200 多个网络模型提供了权重参数。...许多任务的训练过程可通用类似的工作流程,其中训练和验证流程可以循环的运行,并且验证流程可选。每个周期,我们都会在模型运行多次前传和反传操作。...由于这些代码库也开发,因此在其模型仓库中报告的结果可能已过时,并且这些结果是不同的硬件上所进行的测试。为了公平比较,我们提取最新的代码并在相同的环境中进行测试,结果显示在下图中。... MMDetection ,我们采用 1333×800 作为默认训练数据规模。

75620

OpenMMLab 第二版发布:吸引业界「目光」的史上最完整的目标检测工具箱

AI 科技评论按:去年 10 月,中国香港中文大学-商汤联合实验室 OpenMMLab 的首版计划,开放了 MMCV 和 MMDetection(初期版本)。...此后,MMDetection 逐渐发展成为一个统一的平台,涵盖了许多主流检测方法和现代模块。它不仅包括训练和推理代码,而且还为 200 多个网络模型提供了权重参数。...许多任务的训练过程可通用类似的工作流程,其中训练和验证流程可以循环的运行,并且验证流程可选。每个周期,我们都会在模型运行多次前传和反传操作。...由于这些代码库也开发,因此在其模型仓库中报告的结果可能已过时,并且这些结果是不同的硬件上所进行的测试。为了公平比较,我们提取最新的代码并在相同的环境中进行测试,结果显示在下图中。... MMDetection ,我们采用 1333×800 作为默认训练数据规模。

1.2K20

MMDetection全流程实战指南:手把手带你构建目标检测模型

假如安装成功将会显示以下界面: 安装完,官网还提供一个代码来测试是否成功安装。 #步骤 1. 我们需要下载配置文件和模型权重文件。...您可以将上述代码保存为 .py 文件,并在包含 data/archive 的目录运行它。完成上述操作,应会在控制台中看到"已完成"的消息。...模型结果 当训练完成,训练好的结果就能够 work_dir 里查看到,里面有模型的配置文件,权重文件以及训练过程的记录。...但是这个还是一个相对较高的 loss,因此模型的准确度也仅在 30 %左右。我们可以在网上实际找几张图片来测试一下模型实际图片上的效果如何。...模型测试 MMDetection 源码里提供了调用模型的配置和权重文件来直接对图像进行测试的 demo 工具,我们可以直接在终端调用 demo/image_demo.py 这个文件来实现模型的效果测试。

4.6K24

重点已划好!OpenMMLab 9 月动态一览

主分支 v0.29.0 支持 PoolFormer (CVPR'2022),发布了预训练模型 MMOCR 新分支 v1.0.0rc1 1.0.0rc1 发布,公开了重构的文本识别模型权重和训练日志...个轻量化姿态估计算法:SimCC (ECCV'22), Debias-IPR(ICCV'21), IPR(ECCV'18) 和 DSNT(ArXiv'18) 优化了热力图(heatmap)可视化功能,支持模型训练同时保存可视化结果...日志中统计的损失不会被平滑的问题 修复启用 EMAHook 时,无法加载维度不匹配的权重 修复启用 WandbVisBackend 无法正常上传保存在本地的图片 感谢 @nijkah @okotaku...MMPose 1.0 的支持 代码改进 config 文件夹增加了模型浏览导引页面,方便寻找和比较不同模型 优化了部分配置文件的格式 Bug 修复 修复了 MMEngine 依赖版本问题 修复了...修复 torch <= 1.12 版本 LayerNorm 导出时的shape问题 修复应用 TensrRT 量化时遇到的 calibration 错误 使用 onnxruntime-gpu 推理

1K10

5 分钟掌握 Python 的 Hook 钩子函数

content_stash.register_input_filter_hook(input_filter_hook) # 执行流程 content_stash.input_pipeline(content) 3. hook开源框架的应用...3.1 keras 深度学习训练流程,hook函数体现的淋漓尽致。...流程先后拆解成: 开始训练 训练一个epoch前 训练一个batch前 训练一个batch 训练一个epoch 评估验证集 结束训练 这些步骤是穿插在训练一个batch数据的过程,这些可以理解成是钩子函数...,我们可能需要在这些钩子函数实现一些定制化的东西,比如在训练一个epoch我们要保存下训练的模型结束训练时用最好的模型执行下测试集的效果等等。...self.on_train_batch_begin) or not generic_utils.is_default(self.on_train_batch_end)) 这些钩子的原始程序是模型训练流程

11K31

MMYOLO一站式上手指南,从安装、训练到模型优化全面覆盖

注:本脚本需要使用 config 文件,按本文档行文顺序,目前并未讲如何写对应数据集的 config 文件,所以该脚本可以在下一小节有了 config 文件 再尝试。...,并且最多保存 2 个权重 # 模型评估时候自动保存最佳模型 checkpoint=dict(interval=10, max_keep_ckpts=2, save_best='auto..._s-v61_fast_1xb12-40e_cat.py --save-path configs_full/yolov5/yolov5_s-v61_fast_1xb12-40e_cat.py 也可以开始训练的输出路径获得...python tools/train.py configs/yolov5/yolov5_s-v61_fast_1xb12-40e_cat.py --amp 运行训练命令,Tensorboard 文件会生成可视化文件夹...我们训练完成,可以通过以下命令测试得到的模型的表现: python tools/test.py configs/yolov5/yolov5_s-v61_fast_1xb12-40e_cat.py

93630

mmdetection最小复刻版(一):整体概览

计划新增的新特性例如可视化分析;核心细节加入tensorboard;darknet权重mmdetection权重转换;新loss实现以及新增算法等等。如果各位有新的想法,可以和我交流。...,图片字节码作为value高效存储,保存为一个文件,有点类似tfrecord。...这个库比较庞大,功能非常强,需要提前开启缓存服务器,然后客户端运行(服务器程序和客户端可以同一个机器),可能在一些复杂场景会用到 LmdbBackend 前面已经说过了。...train/val和test的运行逻辑有区别。 train或者val模式下,运行逻辑是: 1....也就是说如果你想直接用tf的权重,然后迁移到pytorch,那么一定要注意same的实现,必须要手动先算出pad参数,利用F.pad函数先实现same功能,然后进行conv,这样复现的结果才是完全一致的

1.1K10

mmdetection使用目标检测工具箱训练,测试

技术细节—手把手教你如何实现一个新的网络 https://zhuanlan.zhihu.com/p/101222759 mmdetection的configs的各项参数具体解释 https://zhuanlan.zhihu.com...官方安装说明地址:https://github.com/open-mmlab/mmdetection/blob/master/INSTALL.md 简单来说,该仓库目前只支持linux系统上运行,不支持...但是运行官方代码的前提是你已经下载了相关模型的checkpoint的pth文件,并放在mmdetection文件夹目录下的checkpoints文件夹下。...官方提供的所有训练好的pth模型文件下载地址都在MODEL_ZOO.md。另外随便照一张图片重命名为test.jpg放到mmdetection目录下就可以了。...使用这一步的test命令的时候会报错,程序中断,但是其实问题是一致的,应该是训练中保存下来的pth文件没有CLASSES信息,所以show不了图片结果。

1.8K30

一招带你轻松实现MM系列模型的一键转换与高效部署!

本文入选【技术写作训练营】优秀结营作品,作者:李剑锋 模型转换一直以来都是实际应用很让人头疼的问题。使用 OpenMMLab 等算法框架训练完模型,我们通常得到模型的配置和权重文件。...接下来,就让我来详细介绍一下模型部署的一些基础知识以及如何通过 OpenMMLab 官方网站,一键转换并部署 MM 系列模型。...选定,我们要载入自己已经使用的配置文件(正常情况下是存放在 work_dirs 下的 .py 文件)以及模型权重文件(也 work_dir s的 .pth 文件)。...如图所示,“查看参数”里我们可以看到这个是什么算法的模型以及对应转换完的框架是什么。由于看不到具体的配置和权重文件的名字,所以建议大家模型转化第一时间下载模型到本地并备注好相对应的信息。...大概等待几分钟,如图所示,我们就能够获得最终的测试结果。我们可以看到各个部分的运行速度和结果。

49920

手把手带你轻松掌握 MMDetection 整体构建流程(二)

轻松掌握 MMDetection 整体构建流程(一) 一文,我们重点分析了 MMDetection 框架 Model 整体构建流程,但仅对 Model 算法组件方面进行深入分析,并未涉及整个框架训练和测试流程...Model 运行后会输出 loss 以及其他一些信息,会通过 logger 进行保存或者可视化。 7....第二层模块抽象 总体把握了整个 MMDetection 框架训练和测试流程,下个层次是每个模块内部抽象流程,主要包括 Pipeline、DataParallel、Model、Runner 和 Hooks...上图是一个非常典型的训练流程 Pipeline,每个类都接收字典输入,输出也是字典,顺序执行,其中绿色表示该类运行新增字段,橙色表示对该字段可能会进行修改。...下面列出了 MMDetection 几个非常重要的 hook 以及其作用的生命周期: 例如 CheckpointHook 每个训练 epoch 完成后会被调用,从而实现保存权重功能。

2.7K20

Kaggle实战目标检测奇淫技巧合集

预训练模型 数据处理完以后,基本上就是要冲击前排了,这里就是要考虑如何选用预训练模型了,一般的检测都是使用 ImageNet 预训练的 backbone,这是基本配置,高级一点的就是针对数据集做一次预训练...再一点就是使用 coco 预训练的完整检测模型权重,这样的效果就是模型收敛速度贼快,而且效果一般都比较好,也是大家最常用的方法,这里给出 mmdetection 修改 coco 预训练权重类别数的脚本:...我自己也 mmdetection 实现了 SeNet 系列的 backbone,无奈没卡,没法训一个 coco 出来。。。。 4. 训练技巧 深度炼丹名不虚传。。。。。..., bbox_head 中使用,mmdetection 的相应配置 (如果要正确使用,需要做点改动,自行修改源码吧,不难): #1. rpn 处更改 rpn_head=dict( type=...实现细节移步:NMS 与 soft NMS mmdetection 的设置如下: test_cfg = dict( rpn=dict( nms_across_levels=

3K21

YOLOX | MMDetection 复现保姆级解析

近期,MMDetection 开源团队成员也组织进行了相关复现。通过与社区成员的协同开发,不仅让复现过程更加高效,而且社区成员参与过程可以不断熟悉算法,熟悉 MMDetection 开发模式。...#(偷偷注释掉)为了方便将官方开源权重迁移到 MMDetection 推理精度对齐过程,我们没有修改任何模型代码,而且简单的复制开源代码,分别插入 MMDetection 的 backbone...2.1 推理精度对齐 为了方便将官方开源权重迁移到 MMDetection 推理精度对齐过程,我们没有修改任何模型代码,而且简单的复制开源代码,分别插入 MMDetection 的 backbone... MMDetection 可以直接通过修改配置文件来支持上述功能: test_pipeline = [ dict(type='LoadImageFromFile'), dict(...MMDetection 中进行评估,验证是否对齐,结果如下: 注意:由于官方开源代码一直处于更新,现在下载的最新权重,可能 mAP 不是上表的值。

78410
领券