当前位置: 首页 > news >正文

【BUG】‘DetDataSample‘ object has no attribute ‘_gt_sem_seg‘

问题: 使用mmdetection框架使用COCO格式训练自定义数据集时,其中模型使用HTC模型时出现如下问题:

AttributeError: ‘DetDataSample’ object has no attribute ‘_gt_sem_seg’. Did you mean: ‘gt_sem_seg’?
results = self(**data, mode=mode)

阅读Hybrid Task Cascade for Instance Segmentation和MMDetection的指导文档中数据集准备中发现,需要我们在COCO格式数据集基础上,提供一个stuffthingmaps文件夹,包含原始图像对应的语义分割标签。

解决方法:

  1. 新建文件夹存储原始图像对应的语义分割标签;
  2. train_dataloaderval_dataloaderdata_prefix=dict(img='train_img/', seg='train_seg/')处补充seg=‘your_path’
    • train_img是图像地址
    • train_seg是图像对应语义分割标签地址

mmdet/configs/_base_/datasets/coco_instance.py文件中修改train_dataloaderval_dataloader部分,具体如下:

train_dataloader = dict(batch_size=8,num_workers=2,persistent_workers=True,sampler=dict(type='DefaultSampler', shuffle=True),batch_sampler=dict(type='AspectRatioBatchSampler'),dataset=dict(type=dataset_type,data_root=data_root,ann_file='annotations/annotation_train.json',data_prefix=dict(img='train_img/', seg='train_seg/'),filter_cfg=dict(filter_empty_gt=True, min_size=32),pipeline=train_pipeline,backend_args=backend_args))val_dataloader = dict(batch_size=1,num_workers=2,persistent_workers=True,drop_last=False,sampler=dict(type='DefaultSampler', shuffle=False),dataset=dict(type=dataset_type,data_root=data_root,ann_file='annotations/annotation_valid.json',data_prefix=dict(img='val_img/', seg='val_seg/'),test_mode=True,pipeline=test_pipeline,backend_args=backend_args))

相关文章:

  • BRAFAR: Bidirectional Refactoring, Alignment, Fault Localization, and Repair...
  • Listremove数据时报错:Caused by: java.lang.UnsupportedOperationException
  • Win11/Win10无法保存ip设置提示请检查一个或多个设置并重试怎么办?
  • [人机交互]协作与通信的设计
  • 二叉树—中序遍历—非递归
  • centos的根目录占了大量空间怎么办
  • 大语言模型(LLM)领域,有几项显著的进展和技术突破
  • 如何用Java读取PDF
  • 自然语言处理之情感分析:ALBERT在社交媒体的应用技术教程
  • 家庭宽带IP与IDC机房IP
  • ϵ-prediction和z0-prediction是什么意思
  • Day17 聚类算法(K-Means、DBSCAN、层次聚类)
  • HarmonyOS 5.0 低时延音视频开发​​
  • 【硬核数学】0. 序章:万丈高楼平地起,AI数学筑基之旅《从零构建机器学习、深度学习到LLM的数学认知》
  • 数字图像相关法在薄板变形测量中的实践
  • Apache Velocity代码生成简要介绍
  • 单例模式都有哪些?有什么区别?
  • 求组合数【递推+快速幂+卢卡斯+线性筛】
  • 单例模式的实现方法
  • Android数据库全栈开发实战:Room+SQLCipher+Hilt企业级应用构建
  • 上海乐高乐园度假区将于7月5日开园
  • 躺着玩手机真有意思,我“瞎”之前最喜欢了
  • 山东滕州一车辆撞向公交站台撞倒多人,肇事者被控制,案件已移交刑警
  • 山大齐鲁医院护士论文现“男性确诊子宫肌瘤”,院方称将核实
  • 李学明谈笔墨返乡:既耕春圃,念兹乡土
  • 想要“逆转”糖尿病,减少这两处脂肪是关键