pytorch 训练maskrcnn 自定义数据集出错

最新推荐文章于 2024-03-11 11:23:02 发布
原创 最新推荐文章于 2024-03-11 11:23:02 发布 · 517 阅读 · 0 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文解决了使用MaskRcnn进行实例分割时遇到的问题：类别数设置错误及标签编号未从0开始导致的RuntimeError。指出正确的类别数应为目标类别数加一，并确保标签从0开始编号。

问题描述：
RuntimeError: copy_if failed to synchronize: device-side assert triggered

这个问题是我在使用MaskRcnn做实例分割时遇到的，我要检测的目标有5种类别，所以我在data/config.py中的num_classes参数写了5，经过多方查找，发现了一个没注意到的细节，类别应该是5+1，那个1应该是背景。

还有一个原因就是标签的标号没有从0开始。

我的是两个都遇到了.

基于YOLOv12的博物馆文物检测系统设计与实现：OpenHeritage数据集应用
YOLO
06-12 492
OpenHeritage是一个专门为文化遗产保护设计的大规模数据集，包含多种文物类型的标注数据。数据规模：包含超过50,000张高分辨率文物图像类别丰富：涵盖陶瓷、金属器、书画、玉器等15个大类，120个子类标注精细：提供边界框、类别标签及部分文物的语义分割标注多场景采集：包含博物馆展厅、库房、考古现场等多种环境下的文物图像数据集结构示例：text复制下载│ ├── val/├── train/├── val/└── test/
各类深度学习框架详解+深度学习训练环境搭建-GPU版本
master_hunter的博客
10-23 4909
工欲善其事必先利其器，对工具的熟练程度很多时候决定你工程项目的质量和上限。对于人工智能工程师来说，搭建本地深度学习环境来说是比较麻烦的一件事，其中涉及到较多的相关硬件和软件的兼容问题。很多初学者安装的深度学习框架环境仅能够使用CPU运行并没有利用到GPU，关于很多硬件关联的深度学习训练框架并没有清楚的认知，导致配置环境的时候很容易出现各类错误。所以本篇文章主要讲述清楚如何搭建深度学习环境以及框架选择和自己硬件匹配的系统。如果你发现深度学习看似难以掌握，我将尽力简化知识，将其转化为我们更容易理解的内容。
Mask RCNN训练自己的数据集错误记录
peng_hongguang的博客
12-03 589
Mask RCNN训练自己的数据集错误记录 图片标记使用labelme标记及转换时百度labelme批量转换，复制代码运行即可，无需一步步查错 IndexError: boolean index did not match indexed array along dimension 0; dimension is 0 but corresponding boolean dimension is 1.可能是没有修改添加类及标签的代码，使得对应类始终为0从而报错；其他类似报错则可能是24位及8位图片的问题。
pytorch版Maskrcnn benchmark训练自己的数据集（实例分割）
qq_37754309的博客
01-13 5049
之前一直使用的都是tensorflow版maskrcnn,虽然知道pytorch版已经出来很久了，但一直也没想着试试，前几天组里大牛让我试着用pytorch版的maskrcnn来训练一下现在的数据集，毕竟pytorch版在速度上要比tensorflow版的快很多。 先去官网将代码下载下来 https://github.com/facebookresearch/maskrcnn-benc...
facebook Pytorch mask rcnn训练碰到的问题
qq_32799915的博客
04-01 1309
1.每次重新开始训练，都要检查last_checkpoint文件 代码默认从last_checkpoint 开始训练，因此如果要重新开始训练，加载预训练模型，需要将last_checkpoint 文件删除。 2. 维度不匹配问题 当分类数量改变之后，会存在当前模型最后的分类层与预训练参数维度不匹配的问题，pytorch不提供选择load参数的接口，解决方法是：从预训练模型中，移除不需要的那些数，包...
maskrcnn模型加载问题
weixin_44716713的博客
11-03 577
文档外模型加载报错 ![在这里插入代码片](https://img-blog.csdnimg.cn/20201103121149942.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80NDcxNjcxMw==,size_16,color_FFFFFF,t_70#pic_center) ...
打包paddle 运行时出现RuntimeError: (PreconditionNotMet) 错误的解决
最新发布
qq_58168857的博客
03-11 3812
一个自动化操作的项目，使用了paddle，由于第一次用pyinstaller打包paddle项目(为什么不用nuitka?打包时间太长啦，遇到问题解决一次，时间都非常的长，无奈只把自己开发的模块用nuitka进行了打包，主程序用了pyinstaller)，所以遇到了不少坑，前后整了三天，才把这个问题彻底解决，由于我用的是win10系统，最主要的问题在下面这个。
Mask R-CNN 训练自己的数据集（balloon过程+报错解释）
刘大鱼的博客
03-11 1万+
因项目需要，识别带有多边形标注的图像，舍弃了速度快精度也高的yolov3，使用Mask R-CNN网络。在摸爬滚打中查找资料修改代码以及不断地调整训练集，途中踩了不少坑，终于达到预期的效果。 一、配置说明 以下是本人跑Mask R-CNN的配置说明： Python 3.52 tensorflow 1.12.0 keras 2.2.4 Jupyter Notebook 一块gpu 二、标注训练集...
maskRcnn环境配置(anaconda)（win10）及成功运行Mask_RCNN-2.1的demo.ipynb
weixin_43694742的博客
03-06 8040
一、 maskRcnn环境配置（win10） 先看我成功运行的配置： python3.6.13 tensorflow-gpu 1.5.0 scipy1.2.1 Keras2.2.0 maskRcnn环境中所有安装包： absl-py 1.0.0 astor 0.8.1 backcall 0.2.0 bleach 4.1.0 cached-property 1.5.2
【两次】基于win10搭建mask_rcnn环境，并跑通实验————分享经验
qy13913453196的博客
10-15 1万+
一、环境搭建概括 win10+GPU+python3.6+anaconda3.5+VS2015+CUDA8.0+cudnn6.0+tensorflow1.8.0         【啰嗦一下】：由于实验室服务器暂时用不了，本人便在阿里云上开通了ECS的GPU服务器，用起来还不错，于是便把环境搭建起来，然后跑了一下demo；不过在跑Mask_rcnn时系统盘爆满，便想着更换自己购买服务器的系统盘，...
PyTorch版Mask R-CNN图像实例分割实战：训练自己的数据集
10-07
注意：本课程已从maskrcnn-benchmark更新到了Detectron2Mask R-CNN是一种基于深度学习的图像实例分割方法，可对物体进行目标检测和像素级分割。本课程将手把手地教大家使用Labelme图像标注工具制作自己的数据集，并使用PyTorch版本的Mask R-CNN（Facebook 官方Detectron2/maskrcnn-benchmark）训练自己的数据集，从而能开展自己的图像分割应用。本课程的具体项目实战案例是：对汽车行驶场景中的路坑、车、车道线等多类物体进行检测和分割 。本课程在Ubuntu系统上进行项目全过程的演示。本课程提供项目标注的数据集和相关Python程序文件。相关课程：Mask R-CNN图像实例分割实战：训练自己的数据集 （Keras版本 ）https://edu.youkuaiyun.com/course/detail/25128
1only batches of spatial targets supported (non-empty 3D tensors) but got targets of size
zy的博客
01-16 527
1only batches of spatial targets supported (non-empty 3D tensors) but got targets of size
pytorch错误及解决方案
热门推荐
John Kang的博客
04-15 3万+
pytorch错误及解决方案 目录 RuntimeError: CUDA error: device-side assert triggered RuntimeError:invalid argument 5:k not in range for dimension at /pytorch/ate ... ValueError: optimizer got an empty parameter list The size of tensor a (197) must match the..
解决Pytorch转onnx错误：Only tuples, lists and Variables are supported as JIT inputs/outputs！
gaoxueyi551的专栏
07-19 6388
沿着第二种解决思路，重新审视了自己的模型的各个模块，最后发现了问题，在forward函数内部，将推理结果从tensor类型转换为了numpy类型或者是python的list，这都会导致在转onnx时报如上错误，所以最终将forward内部的转换逻辑从内部删除，在模型的外部根据需要再进行转换，问题就迎刃而解。上图是我的代码，没有任何问题。如上图，去掉classify_scores2result转换函数后，直觉返回head的推理结果即可转换成功。...
Pytorch Mask RCNN训练自定义数据集
平凡简单的执着
05-13 1万+
前言（必读） 最近做目标检测，然后记录一下 Faster RCNN、Mask RCNN来做目标检测踩得那些坑。 首先，本文并不是利用Pytorch从头去实现Faster RCNN、Mask RCNN这两个结构的文章。如果有意向去从头实现并了解每一步细节可以看看下面这些视频和博客： 来自B站的两位大佬讲解 大佬一：视频 博客 GitHub 大佬二：视频 博客 GitHub 上面都是利用pytorch从原理到具体实现的步骤。 不过本文主要还是利用Pytorch中的Tor
huggingface：transformers中文文本分词
wu0310zh的博客
09-20 2459
huggingface：transformers中文文本分词
ERROR: 加载预训练模型参数时出现size mismatch错误
pangxing6491的博客
09-23 7524
原来的resnet为： def resnet152(pretrained=False, **kwargs): """Constructs a ResNet-152 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet """ model = ResNet(Bottleneck, [3, 8, 36, 3], **kwargs) if pre.
Mask R-CNN、Pytorch如何训练自己的数据集
qq_26526257的博客
04-21 2981
maskrcnn,windows，pytorch版
Mask R-CNN训练自己的数据集（pytorch）
qq_54448856的博客
10-24 1433
mask rcnn训练自己的数据集
PyTorch实现YOLOV3自定义数据集训练教程
本文档是关于使用PyTorch实现YOLOV3模型训练自定义数据集的教程。主要内容包括数据集准备、训练代码的使用以及数据预处理步骤。 YOLOV3是一种流行的实时目标检测算法，它能够在图像中快速识别多个类别的目标。在...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值