mmyolo训练记录

文章介绍了如何使用mmyolo进行实验,包括按照官方安装步骤配置环境,调整数据集路径以适应COCO格式,设置训练参数,以及遇到类别数量不足引发的‘IndexError’时,提出了增大阈值和修改cocoapi的解决方案。
部署运行你感兴趣的模型镜像

最近在使用mmyolo跑一些实验。
1、环境安装,就按照官网的安装步骤来就可以了 。
https://github.com/open-mmlab/mmyolo
在这里插入图片描述
2、数据集,如果是在配置文件里面。将数据集路径加载为自己存放的路径,ps:这里使用的coco格式。
在这里插入图片描述
3、训练参数:
在这里插入图片描述
4、开始训练,mmyolo的训练方法和mmdetection差不多。
在这里插入图片描述
5、我的类别数量比较少,导致报错:IndexError:list index out of range.
在这里插入图片描述
修改方案:
1、增大阈值
在这里插入图片描述
2、修改cocoapi
在这里插入图片描述

您可能感兴趣的与本文相关的镜像

Yolo-v8.3

Yolo-v8.3

Yolo

YOLO(You Only Look Once)是一种流行的物体检测和图像分割模型,由华盛顿大学的Joseph Redmon 和Ali Farhadi 开发。 YOLO 于2015 年推出,因其高速和高精度而广受欢迎

### 使用 MMYOLO 框架训练 VisDrone 数据集 #### 准备环境 安装必要的依赖项并配置开发环境对于顺利运行项目至关重要。确保已正确设置 Python 和 PyTorch 环境之后,可以通过 pip 安装 OpenMMLab 提供的一系列工具包。 ```bash pip install mmcv-full mmdet mmyolo ``` #### 下载和准备数据集 VisDrone 是一个广泛使用的无人机视角下的视觉目标检测与跟踪数据库。下载官方发布的最新版本的数据集文件,并按照如下结构整理: ``` mmyolo/ ├── configs/ │ └── custom_visdrone.py ├── data/ │ ├── visdrone/ │ ├── annotations/ │ ├── train.json │ ├── val.json │ ├── images/ │ ├── train/ │ └── val/ └── tools/ └── ... ``` #### 修改配置文件 创建一个新的配置文件 `custom_visdrone.py` 来定义特定于 VisDrone 数据集的任务参数。此配置继承自默认的 RTMDet 设置[^1],但针对 VisDrone 特征进行了调整。 ```python _base_ = [ '../rtmdet/rtmdet_l_8xb32-300e_coco.py' # 基础配置路径 ] # 数据集相关设定 data_root = 'data/visdrone/' dataset_type = 'CocoDataset' train_pipeline = [...] test_pipeline = [...] data = dict( samples_per_gpu=2, workers_per_gpu=2, train=dict( type='RepeatDataset', times=5, dataset=dict( type=dataset_type, ann_file=data_root + 'annotations/train.json', img_prefix=data_root + 'images/train/', pipeline=train_pipeline)), val=dict( type=dataset_type, ann_file=data_root + 'annotations/val.json', img_prefix=data_root + 'images/val/', pipeline=test_pipeline), test=dict( type=dataset_type, ann_file=data_root + 'annotations/val.json', img_prefix=data_root + 'images/val/', pipeline=test_pipeline)) evaluation = dict(interval=1, metric='bbox') ``` #### 开始训练过程 一切就绪后,在命令行界面执行以下指令启动训练脚本。这里假设已经设置了 CUDA 可用设备来加速计算效率。 ```bash python tools/train.py configs/custom_visdrone.py --gpu-id 0 ``` 通过上述操作,能够基于 MMYOLO 平台有效地利用 RTMDet 架构完成对 VisDrone 数据集的目标检测任务训练工作。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值