栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

mmdetection学习日记

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

mmdetection学习日记

想复现Oriented R-CNN的代码并做一些改动,下载下来发现是使用的mmdetection框架,于是边学习记录一些遇到的问题。

1、训练步骤

(1)在configs中选自己想要用的模型,可以直接修改,我习惯新建一个.py文件,将model、dataset等config文件全部复制过来,方便改动,即开头的_base_中所有文件都拷贝到新.py中。

(2)修改后命令行内运行

python tools/train.py configs/xxx.py
2、无法下载预训练模型,报错"HTTP Error 403: Forbidden"

换成国内源的网址即可正常下载

可以查询:mmcv/model_zoo/open_mmlab.json · master · mirrors / open-mmlab / mmcv · CODE CHINA (gitcode.net)https://gitcode.net/mirrors/open-mmlab/mmcv/-/blob/master/mmcv/model_zoo/open_mmlab.json

选取想要的模型,将地址复制下来,在configs/default_runtime.py中修改load_from为新地址即可。

3、用mmdetection中现成模型训练自己的数据集

如训练COCO格式的自制数据集,类别与COCO不一致

(1)在data文件夹中放入分好train/val/test的图片和标签。

(2)自己的config文件中,将num_classes修改为自己的类别数(加背景),如model ——> roi_head ——> bbox_head ——> num_classes,如果有mask分支则mask_head中也有一项num_classes。

(3)mmdet/datasets/coco.py 、 mmdet/core/evaluation/class_names.py  都将原CLASSES注释掉,换成新的类别名称。

4、ValueError: need at least one array to concatenate

一般都是数据集的标签里出了问题,查看标签文件里是否没有生成bbox坐标或label等

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/530213.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号