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

最新推荐文章于 2024-10-31 14:31:24 发布
最新推荐文章于 2024-10-31 14:31:24 发布
阅读量506 收藏
点赞数 1
CC 4.0 BY-SA版权
分类专栏： 机器视觉 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/phker/article/details/109576341
本文解决了使用MaskRcnn进行实例分割时遇到的问题：类别数设置错误及标签编号未从0开始导致的RuntimeError。指出正确的类别数应为目标类别数加一，并确保标签从0开始编号。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

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

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

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

我的是两个都遇到了.

各类深度学习框架详解+深度学习训练环境搭建-GPU版本
master_hunter的博客
10-23 4462
工欲善其事必先利其器，对工具的熟练程度很多时候决定你工程项目的质量和上限。对于人工智能工程师来说，搭建本地深度学习环境来说是比较麻烦的一件事，其中涉及到较多的相关硬件和软件的兼容问题。很多初学者安装的深度学习框架环境仅能够使用CPU运行并没有利用到GPU，关于很多硬件关联的深度学习训练框架并没有清楚的认知，导致配置环境的时候很容易出现各类错误。所以本篇文章主要讲述清楚如何搭建深度学习环境以及框架选择和自己硬件匹配的系统。如果你发现深度学习看似难以掌握，我将尽力简化知识，将其转化为我们更容易理解的内容。
解密 Faster R-CNN：高效目标检测的两阶段框架
专注于深入研究多种编程语言，以实战为导向，逐步拓展开发技能，提升工程化编码和思维能力，展现无敌技术实力。
03-03 227
Faster R-CNN 是一种高效的目标检测框架，通过结合区域提议网络（RPN）和 Fast R-CNN，实现快速生成候选区域并精准分类，从而在图像中高效地定位和识别目标。
Mask RCNN训练自己的数据集错误记录
peng_hongguang的博客
12-03 575
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 5001
之前一直使用的都是tensorflow版maskrcnn,虽然知道pytorch版已经出来很久了，但一直也没想着试试，前几天组里大牛让我试着用pytorch版的maskrcnn来训练一下现在的数据集，毕竟pytorch版在速度上要比tensorflow版的快很多。 先去官网将代码下载下来 https://github.com/facebookresearch/maskrcnn-benc...
facebook Pytorch mask rcnn训练碰到的问题
qq_32799915的博客
04-01 1280
1.每次重新开始训练，都要检查last_checkpoint文件 代码默认从last_checkpoint 开始训练，因此如果要重新开始训练，加载预训练模型，需要将last_checkpoint 文件删除。 2. 维度不匹配问题 当分类数量改变之后，会存在当前模型最后的分类层与预训练参数维度不匹配的问题，pytorch不提供选择load参数的接口，解决方法是：从预训练模型中，移除不需要的那些数，包...
maskrcnn模型加载问题
weixin_44716713的博客
11-03 561
文档外模型加载报错 ![在这里插入代码片](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 3480
一个自动化操作的项目，使用了paddle，由于第一次用pyinstaller打包paddle项目(为什么不用nuitka?打包时间太长啦，遇到问题解决一次，时间都非常的长，无奈只把自己开发的模块用nuitka进行了打包，主程序用了pyinstaller)，所以遇到了不少坑，前后整了三天，才把这个问题彻底解决，由于我用的是win10系统，最主要的问题在下面这个。
【两次】基于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时系统盘爆满，便想着更换自己购买服务器的系统盘，...
使用YOLO V5训练自动驾驶目标检测网络
williamhyin的博客
07-31 6343
YOLO V5 模型结构及迁移学习 YOLO 是一种快速紧凑的开源对象检测模型，与其它网络相比，同等尺寸下性能更强，并且具有很不错的稳定性，是第一个可以预测对象的类别和边界框的端对端神经网络。YOLO 家族一直有着旺盛的生命力，从YOLO V1一直到”V5“，如今已经延续五代，凭借着不断的创新和完善，一直被计算机视觉工程师作为对象检测的首选框架之一。 Ultralytics于5月27日发布了YOLOv5 的第一个正式版本，其性能与YOLO V4不相伯仲，是现今最先进的对象检测技术之一，并在推理速度上是目前
那我不使用yolo系列的模型，我使用fast rcnn的模型，那我现在有了xml文件了，下一步是什么？
最新发布
03-16
如果用户使用的框架有现成的VOC数据集支持，比如PyTorch的torchvision.datasets.VOCDetection，那么可能只需要将数据集放在指定目录，并调整配置即可。否则，需要自己编写数据解析和加载的代码。这时候可能需要指导...
MASK_RCNN解读3-原始训练模型加载和构建
01-06
MASK_RCNN解读3-原始训练FPN模型加载 现在开始进行网络的训练，脚本为： python tools/train_net.py –config-file configs/my_e2e_mask_rcnn_R_50_FPN_1x.yaml MODEL.ROI_BOX_HEAD.NUM_CLASSES 6 SOLVER.IMS_PER_BATCH 4 SOLVER.BASE_LR 0.001 SOLVER.MAX_ITER 36000 SOLVER.STEPS ” (24000, 32000)” TEST.IMS_PER_BATCH 1 MODEL.RPN.FPN_POST_NMS_TOP
解决Pytorch转onnx错误：Only tuples, lists and Variables are supported as JIT inputs/outputs！
gaoxueyi551的专栏
07-19 6143
沿着第二种解决思路，重新审视了自己的模型的各个模块，最后发现了问题，在forward函数内部，将推理结果从tensor类型转换为了numpy类型或者是python的list，这都会导致在转onnx时报如上错误，所以最终将forward内部的转换逻辑从内部删除，在模型的外部根据需要再进行转换，问题就迎刃而解。上图是我的代码，没有任何问题。如上图，去掉classify_scores2result转换函数后，直觉返回head的推理结果即可转换成功。...
2.1【Pytorch版(torch-cuda版 ) Mask-RCNN 训练自己的数据集】
木易进
05-15 1万+
写在前面：由于课程试验要求，需要基于pytorch实现maskrcnn，so最近又跑了一下pytorch版的maskrcnn，官方已经给出了详细的教程，虽然说支持cpu推理，但是不支持cpu训练啊，奈何手头上只有一个cpu本，也没有nvida显卡，只有intel的集显，so整理一波本次训练maskrcnn的过程。 这篇blog是按照常规的torch版的maskrcnn的训练教程写的，理论上，所依赖的是torch-cuda版本，后续还有一篇是基于torch-cpu版本的。因为在试验过程中先踩了这部分的坑，.
RuntimeError: copy_if failed to synchronize: device-side assert triggered
weixin_36670529的博客
04-13 667
在使用pytorch训练的时候提示RuntimeError: copy_if failed to synchronize: device-side assert triggered错误 有两个方法可以尝试去解决一下： 1.尝试减少学习率试试看能不能解决这个问题，如果不能，请看第二种方法 2.看看config文件中设置的类别数目与数据中实际的类别数目是否一致，注意对于目标检测需要根据实际的类别数+1(背景) ...
detectron2训练自己的数据集_keras版MaskRCNN来训练自己的目标检测数据集
weixin_39631263的博客
11-21 437
向AI转型的程序员都关注了这个号????????????机器学习AI算法工程 公众号：datayx一、运行环境的安装：1、下载好cuda9跟cudnn7，然后在安装好后，cuda其会自动添加到环境变量里，所以使用keras进行GPU加速的时候会自动使用这些库。2、TensorFlow-gpu版本的安装，这个安装方法有三种，第一种是直接在pycharm里的安装库里安装。第二种就是使用pip来安装，这个在...
跑Yolo v3（pytorch）时遇到的问题
qq_43332629的博客
11-21 521
（1）RuntimeError: copy_if failed to synchronize: device-side assert triggered 在yolo v3的代码中，没有背景类，去检查自己的数据集是否存在标签从1开始的情况，因此需要将起始类别标签设置为0 （2）RuntimeError: CUDA error: device-side assert triggered 检查数据集里面...
pytorch报RuntimeError: copy_if failed to synchronize: device-side assert triggered
热门推荐
修炼之路
07-11 2万+
在使用pytorch训练的时候提示RuntimeError: copy_if failed to synchronize: device-side assert triggered错误 有两个方法可以尝试去解决一下： 1.尝试减少学习率试试看能不能解决这个问题，如果不能，请看第二种方法 2.看看config文件中设置的类别数目与数据中实际的类别数目是否一致，注意对于目标检测需要根据实际的类别...
【亲测免费】 PyTorch版Mask R-CNN图像实例分割实战：训练自己的数据集
gitblog_06658的博客
10-31 543
PyTorch版Mask R-CNN图像实例分割实战：训练自己的数据集 【下载地址】PyTorch版MaskR-CNN图像实例分割实战训练自己的数据集331003安装项目1分享 PyTorch版Mask R-CNN图像实例分割实战：训练自己的数据集【331003】安装项目1本仓库提供了一个资源文件，详细介绍了如何在PyT...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值