MMdetection用voc数据集训练:The `num_classes` (x) in Shared2FCBBoxHead of MMDataParallel does not matches..

这主要是由于你的类别不匹配导致,将mmdetection-master\mmdet\datasets\voc.py中的类别换成你自己的类别名称即可,如:

 运行不再报错:

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Transformer发轫于NLP(自然语言处理),并跨界应用到CV(计算机视觉)领域。 Swin Transformer是基于Transformer的计算机视觉骨干网,在图像分类、目标检测、实例分割、语义分割等多项下游CV应用中取得了SOTA的性能。该项工作也获得了ICCV 2021顶会最佳论文奖。本课程将手把手地教大家使用labelme标注和使用Swin Transformer训练自己的数据集进行图片和视频的实例分割。  本课程将介绍Transformer及在CV领域的应用、Swin Transformer的原理。 本课程以汽车驾驶场景图片和视频开展项目实践:对汽车行驶场景中的路坑、车、车道线进行物体标注和实例分割。  课程在Windows和Ubuntu系统上分别做项目演示。包括:安装软件环境、安装Pytorch、安装Swin-Transformer-Object-Detection、标注自己的数据集、准备自己的数据集数据集格式转换(Python脚本完成)、修改配置文件、训练自己的数据集、测试训练出的网络模型、性能统计、日志分析。  本课程提供项目的数据集和相关Python程序文件。相关课程: 《Transformer原理与代码精讲(PyTorch)》https://edu.csdn.net/course/detail/36697《Transformer原理与代码精讲(TensorFlow)》https://edu.csdn.net/course/detail/36699《ViT(Vision Transformer)原理与代码精讲》https://edu.csdn.net/course/detail/36719《DETR原理与代码精讲》https://edu.csdn.net/course/detail/36768《Swin Transformer实战目标检测训练自己的数据集》https://edu.csdn.net/course/detail/36585《Swin Transformer实战实例分割:训练自己的数据集》https://edu.csdn.net/course/detail/36586《Swin Transformer原理与代码精讲》 https://download.csdn.net/course/detail/37045
### 回答1: 我可以尝试帮你解决这个问题。首先,你需要下载VOC数据集,然后将其转换为MMDetection可以识别的格式。接着,您可以使用预训练的模型进行训练,或者按照MMDetection框架中的步骤手动训练模型。 ### 回答2: mmdetection是一个开源的基于PyTorch的目标检测工具箱,用于训练和推理各种目标检测模型。它支持多种数据集格式,包括Pascal VOC数据集。 要使用mmdetection训练VOC数据集,首先需要准备VOC数据集的标注文件和图像。VOC数据集通常包含JPEG格式的图像文件和对应的XML格式的标注文件,其中包含目标类别、边界框坐标等信息。 接下来,需要在mmdetection的配置文件中对训练参数进行设置。配置文件是一个Python格式的脚本,包含了模型结构、数据增强、训练参数等信息。可以根据需求选择合适的模型结构,例如Faster R-CNN、SSD等。还可以设置训练时的超参数,如学习率、优化器类型等。 在配置文件中,需要设置数据集的相关参数。mmdetection提供了相应的数据集类,可以根据VOC数据集的特点进行配置。需要指定数据集的路径、类别列表、图像和标注文件的文件名等。 完成配置文件的设置后,可以使用mmdetection提供的命令行工具进行训练。通过运行相应的命令,mmdetection会根据配置文件中的设置加载数据集并进行训练训练过程中,mmdetection会逐渐优化模型参数,使其能够更好地检测目标。 训练完成后,可以使用训练得到的模型进行目标检测。可以通过加载模型和输入图像,调用mmdetection提供的API进行目标检测操作。这样就可以利用训练好的模型对VOC数据集中的图像进行目标检测了。 总之,使用mmdetection训练VOC数据集需要准备数据集和标注文件,设置好模型和训练参数的配置文件,然后使用mmdetection提供的命令行工具进行训练训练完成后,可以使用训练好的模型进行目标检测

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值