mmdetection-虚拟环境的搭建与代码调试

2024-01-20 20:18

本文主要是介绍mmdetection-虚拟环境的搭建与代码调试,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

1、mmdetection---虚拟环境配置

官方要求:

初始环境:

  • python==3.6.8
  • CUDA==9.0
  • Pytorch==1.0.0
  • GCC==4.8.4
  • mmcv>=0.2.6

遇到的问题:

①、Segmentation fault(core dumped)-段错误

使用‘‘gdb’’追踪错误:

错误原因:gcc和系统版本太低,需要升级系统和gcc版本;

参考网址:https://github.com/open-mmlab/mmdetection/issues/24

②、Cuda版本与RTX2080有冲突:THCudaCheck FAIL:

 

错误原因:显卡使用RTX 2080Ti的时候,CUDA就要装10以上。还没有验证......

③、AttributeError: model 'torch.nn' has no attribute 'SyncBatchNorm';

解决方法:升级Pytorch版本至Pytorch==1.1.0。

(有关Pytorch安装较慢的问题解决,可参考:https://blog.csdn.net/songchunxiao1991/article/details/95192063)

最后,实现的实践环境:

 

  • python==3.6.8
  • CUDA==9.0+
  • Pytorch==1.1.0(如果是Pytorch==1.0.0,会报错:SyncBatchNorm)
  • GCC==5.4.0
  • mmcv>=0.2.6

 

2、mmdetection---代码调试

调试的数据格式是采用VOC的格式进行的。

①、输入数据的修改:./mmdet/datasets:-voc.py(图片文件默认是*.jpg,如若不是,对应源码进行修改)

Note:CLASSES修改为:自己检测的LabelName

②、配置文件的修改:./configs/cascade_rcnn_r50_fpn_x1.py

Note:num_classes修改为目标检测LabelNum+1

Note:batch_size/epochs/model_save/shape都在config配置文件中进行修改

Note:当gpu数量为8时,lr=0.02;当gpu数量为4时,lr=0.01;只要一个gpu,所以设置lr=0.0025

③、评估阶段的修改:./mmdet/core/evaluation/class_names.py

Note:修改cov_classes为自己的检测LabelNames(与第一步操作的LabelName一致)

3、训练与验证的代码操作

①、训练代码操作:

python tools/train.py configs/cascade_rcnn_r50_fpn_1x.py

Note:前提:训练、验证、测试数据已经提前写在模型的配置文件中(见2.2的配置文件的修改环节

 

②、测试代码操作

  • 首先,VOC格式数据需要先生成对应的结果*.pkl
python tools/test.py configs/cascade_rcnn_r50_fpn_1x.py work_dirs/cascade_rcnn_r50_fpn_1x/epoch_50.pth --out Result_file/Res_test.pkl
  • 然后,使用voc_eval.py评估指标
python tools/voc_eval.py Result_file/Res_test.pkl configs/TB_cascade_rcnn_r50_fpn_1x.py

结果展示:

因为项目需要关注真阳和假阳的指标,因此在源码的基础上添加了tp和fp的指标:

 

 

 

 

 

 

 

这篇关于mmdetection-虚拟环境的搭建与代码调试的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/627201

相关文章

Java的栈与队列实现代码解析

《Java的栈与队列实现代码解析》栈是常见的线性数据结构,栈的特点是以先进后出的形式,后进先出,先进后出,分为栈底和栈顶,栈应用于内存的分配,表达式求值,存储临时的数据和方法的调用等,本文给大家介绍J... 目录栈的概念(Stack)栈的实现代码队列(Queue)模拟实现队列(双链表实现)循环队列(循环数组

使用Java将DOCX文档解析为Markdown文档的代码实现

《使用Java将DOCX文档解析为Markdown文档的代码实现》在现代文档处理中,Markdown(MD)因其简洁的语法和良好的可读性,逐渐成为开发者、技术写作者和内容创作者的首选格式,然而,许多文... 目录引言1. 工具和库介绍2. 安装依赖库3. 使用Apache POI解析DOCX文档4. 将解析

C++使用printf语句实现进制转换的示例代码

《C++使用printf语句实现进制转换的示例代码》在C语言中,printf函数可以直接实现部分进制转换功能,通过格式说明符(formatspecifier)快速输出不同进制的数值,下面给大家分享C+... 目录一、printf 原生支持的进制转换1. 十进制、八进制、十六进制转换2. 显示进制前缀3. 指

Python虚拟环境终极(含PyCharm的使用教程)

《Python虚拟环境终极(含PyCharm的使用教程)》:本文主要介绍Python虚拟环境终极(含PyCharm的使用教程),具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,... 目录一、为什么需要虚拟环境?二、虚拟环境创建方式对比三、命令行创建虚拟环境(venv)3.1 基础命令3

使用Python实现全能手机虚拟键盘的示例代码

《使用Python实现全能手机虚拟键盘的示例代码》在数字化办公时代,你是否遇到过这样的场景:会议室投影电脑突然键盘失灵、躺在沙发上想远程控制书房电脑、或者需要给长辈远程协助操作?今天我要分享的Pyth... 目录一、项目概述:不止于键盘的远程控制方案1.1 创新价值1.2 技术栈全景二、需求实现步骤一、需求

Java中Date、LocalDate、LocalDateTime、LocalTime、时间戳之间的相互转换代码

《Java中Date、LocalDate、LocalDateTime、LocalTime、时间戳之间的相互转换代码》:本文主要介绍Java中日期时间转换的多种方法,包括将Date转换为LocalD... 目录一、Date转LocalDateTime二、Date转LocalDate三、LocalDateTim

jupyter代码块没有运行图标的解决方案

《jupyter代码块没有运行图标的解决方案》:本文主要介绍jupyter代码块没有运行图标的解决方案,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录jupyter代码块没有运行图标的解决1.找到Jupyter notebook的系统配置文件2.这时候一般会搜索到

利用Python快速搭建Markdown笔记发布系统

《利用Python快速搭建Markdown笔记发布系统》这篇文章主要为大家详细介绍了使用Python生态的成熟工具,在30分钟内搭建一个支持Markdown渲染、分类标签、全文搜索的私有化知识发布系统... 目录引言:为什么要自建知识博客一、技术选型:极简主义开发栈二、系统架构设计三、核心代码实现(分步解析

Python通过模块化开发优化代码的技巧分享

《Python通过模块化开发优化代码的技巧分享》模块化开发就是把代码拆成一个个“零件”,该封装封装,该拆分拆分,下面小编就来和大家简单聊聊python如何用模块化开发进行代码优化吧... 目录什么是模块化开发如何拆分代码改进版:拆分成模块让模块更强大:使用 __init__.py你一定会遇到的问题模www.

使用Python自建轻量级的HTTP调试工具

《使用Python自建轻量级的HTTP调试工具》这篇文章主要为大家详细介绍了如何使用Python自建一个轻量级的HTTP调试工具,文中的示例代码讲解详细,感兴趣的小伙伴可以参考一下... 目录一、为什么需要自建工具二、核心功能设计三、技术选型四、分步实现五、进阶优化技巧六、使用示例七、性能对比八、扩展方向建