我是靠谱客的博主 忧心荷花,最近开发中收集的这篇文章主要介绍mmdetection踩坑经验使用mmdetection框架训练自己的数据集修改数据集的类别后发现报错mmdetection 配置文件解读关于图片尺寸等问题可视化结果的自定义(锚框,文字等),觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

使用mmdetection框架训练自己的数据集

根据以下教程即可:

https://blog.csdn.net/qq_35077107/article/details/124768392

修改数据集的类别后发现报错

AssertionError: The `num_classes` (1) in CenterNetHead of MMDataParallel doe

检查了发现类别数目其实都已经修改好了,为什么还会报错呢,其实是因为你没有重新编译环境,具体方法是;python setup.py install。 详见:https://blog.csdn.net/sunnyblogs/article/details/121409301


mmdetection 配置文件解读

https://www.cnblogs.com/Allen-rg/p/12196888.html

关于图片尺寸等问题

https://zhuanlan.zhihu.com/p/141176981
值得注意的是:是把你自己数据集里面的图片放缩并自适应(默认keep_ratio = TRUE)到(1333,800)


可视化结果的自定义(锚框,文字等)

mmdet/core/visualization/image.py在这里面,重新定义了:

def imshow_det_bboxes:
			bbox_color = None
			text_color = None
			font_size = 0
			thickness = 0

最后

以上就是忧心荷花为你收集整理的mmdetection踩坑经验使用mmdetection框架训练自己的数据集修改数据集的类别后发现报错mmdetection 配置文件解读关于图片尺寸等问题可视化结果的自定义(锚框,文字等)的全部内容,希望文章能够帮你解决mmdetection踩坑经验使用mmdetection框架训练自己的数据集修改数据集的类别后发现报错mmdetection 配置文件解读关于图片尺寸等问题可视化结果的自定义(锚框,文字等)所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(42)

评论列表共有 0 条评论

立即
投稿
返回
顶部