MMYOLO调试RTMDet--小数据集split_ss_dota_200

2024-04-11 23:44

本文主要是介绍MMYOLO调试RTMDet--小数据集split_ss_dota_200,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

背景

用MMYOLO调试旋转目标检测时需要用到dota数据集,根据MMYOLO的官方教程,dota数据集经过处理后变为split_ss_dota,但是该数据集还是很大,对于一些配置比较低的机器要调试比较麻烦,所以这里针对该数据集,挑选了训练集200张图片,测试集30张,便于调试算法。

小数据集在本篇博客配套资源包💐

1. MMYOLO及依赖库的版本

版本号
mmyolo0.6.0
mmengine0.10.3
mmcv2.0.1
mmdet3.0.0
mmcls1.0.0.rc6

2. 一键软链接上面的包:

首先将上面的包,gitclone到mmyolo同级目录:
在这里插入图片描述

cd mmyolo

文件夹的名字可能不同,需要根据自己的名字修改👇。

ln -s ../mmengine/mmengine/ mmengine && ln -s ../mmcv/mmcv/ mmcv && ln -s ../mmdetection-3.0.0/mmdet/ mmdet && ln -s ../mmrotate-1.0.0rc1/mmrotate/ mmrotate

配置文件更改

博主用到的配置文件是rtmdet-r_s_fast_1xb8-36e_dota.py,以这个为例,下面只给出需要更改的部分。

1. mmyolo/configs/rtmdet/rotated/rtmdet-r_l_syncbn_fast_2xb4-36e_dota.py

# ========================Frequently modified parameters======================
# -----data related-----
# data_root = 'data/split_ss_dota/'
data_root = '/home/hhh/dataset/ss_dota_200/'
train_batch_size_per_gpu = 2 # 8  这个其实可以不改,在下面哪个文件改就行
# Worker to pre-fetch data for each single GPU during training
train_num_workers = 0 # 8
# persistent_workers must be False if num_workers is 0.
persistent_workers = False # True

2. mmyolo/configs/rtmdet/rotated/rtmdet-r_s_fast_1xb8-36e_dota.py

# Batch size of a single GPU during training
train_batch_size_per_gpu = 2

调试指令

conda activate mmyolo
cd mmyolo
python tools/train.py /home/hhh/pyproject/mmyolo/configs/rtmdet/rotated/rtmdet-r_s_fast_1xb8-36e_dota.py

查看配置文件是否修改成功

python tools/misc/print_config.py configs/rtmdet/rotated/rtmdet-r_s_fast_1xb8-36e_dota.py

在这里插入图片描述

自己编译MMCV

注意:如果机器配置低,建议关闭其他软件,只开一个终端进行编译

# 我是直接去mmcv官网,releases找到自己想要的版本直接下载的
git clone -b 2.x git@github.com:open-mmlab/mmcv.git cd mmcvconda activate mmyolo# 安装 ninja 和 psutil 以加快编译速度
pip install -r requirements/optional.txt -i https://pypi.tuna.tsinghua.edu.cn/simple# 仅仅编译,不安装
MMCV_WITH_OPS=1 python setup.py build_ext --inplace# 建立软链接
cd ../mmyolo
ln -s ../mmcv/mmcv mmcv

编译成功后会生成一个_ext文件,这个文件很重要,如果没有的话调试时会报错:
在这里插入图片描述

参考

MMYOLO 实用类第一期:OpenMMLab 2.0 源码阅读和调试「必备」技巧

这篇关于MMYOLO调试RTMDet--小数据集split_ss_dota_200的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



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

相关文章

Pandas统计每行数据中的空值的方法示例

《Pandas统计每行数据中的空值的方法示例》处理缺失数据(NaN值)是一个非常常见的问题,本文主要介绍了Pandas统计每行数据中的空值的方法示例,具有一定的参考价值,感兴趣的可以了解一下... 目录什么是空值?为什么要统计空值?准备工作创建示例数据统计每行空值数量进一步分析www.chinasem.cn处

如何使用 Python 读取 Excel 数据

《如何使用Python读取Excel数据》:本文主要介绍使用Python读取Excel数据的详细教程,通过pandas和openpyxl,你可以轻松读取Excel文件,并进行各种数据处理操... 目录使用 python 读取 Excel 数据的详细教程1. 安装必要的依赖2. 读取 Excel 文件3. 读

利用Python调试串口的示例代码

《利用Python调试串口的示例代码》在嵌入式开发、物联网设备调试过程中,串口通信是最基础的调试手段本文将带你用Python+ttkbootstrap打造一款高颜值、多功能的串口调试助手,需要的可以了... 目录概述:为什么需要专业的串口调试工具项目架构设计1.1 技术栈选型1.2 关键类说明1.3 线程模

Spring 请求之传递 JSON 数据的操作方法

《Spring请求之传递JSON数据的操作方法》JSON就是一种数据格式,有自己的格式和语法,使用文本表示一个对象或数组的信息,因此JSON本质是字符串,主要负责在不同的语言中数据传递和交换,这... 目录jsON 概念JSON 语法JSON 的语法JSON 的两种结构JSON 字符串和 Java 对象互转

C++如何通过Qt反射机制实现数据类序列化

《C++如何通过Qt反射机制实现数据类序列化》在C++工程中经常需要使用数据类,并对数据类进行存储、打印、调试等操作,所以本文就来聊聊C++如何通过Qt反射机制实现数据类序列化吧... 目录设计预期设计思路代码实现使用方法在 C++ 工程中经常需要使用数据类,并对数据类进行存储、打印、调试等操作。由于数据类

SpringBoot使用GZIP压缩反回数据问题

《SpringBoot使用GZIP压缩反回数据问题》:本文主要介绍SpringBoot使用GZIP压缩反回数据问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录SpringBoot使用GZIP压缩反回数据1、初识gzip2、gzip是什么,可以干什么?3、Spr

SpringBoot集成Milvus实现数据增删改查功能

《SpringBoot集成Milvus实现数据增删改查功能》milvus支持的语言比较多,支持python,Java,Go,node等开发语言,本文主要介绍如何使用Java语言,采用springboo... 目录1、Milvus基本概念2、添加maven依赖3、配置yml文件4、创建MilvusClient

SpringValidation数据校验之约束注解与分组校验方式

《SpringValidation数据校验之约束注解与分组校验方式》本文将深入探讨SpringValidation的核心功能,帮助开发者掌握约束注解的使用技巧和分组校验的高级应用,从而构建更加健壮和可... 目录引言一、Spring Validation基础架构1.1 jsR-380标准与Spring整合1

MySQL 中查询 VARCHAR 类型 JSON 数据的问题记录

《MySQL中查询VARCHAR类型JSON数据的问题记录》在数据库设计中,有时我们会将JSON数据存储在VARCHAR或TEXT类型字段中,本文将详细介绍如何在MySQL中有效查询存储为V... 目录一、问题背景二、mysql jsON 函数2.1 常用 JSON 函数三、查询示例3.1 基本查询3.2

SpringBatch数据写入实现

《SpringBatch数据写入实现》SpringBatch通过ItemWriter接口及其丰富的实现,提供了强大的数据写入能力,本文主要介绍了SpringBatch数据写入实现,具有一定的参考价值,... 目录python引言一、ItemWriter核心概念二、数据库写入实现三、文件写入实现四、多目标写入