DETR训练自己的数据集AP=0

最新推荐文章于 2024-03-14 00:31:19 发布
最新推荐文章于 2024-03-14 00:31:19 发布
阅读量776 收藏 12
点赞数 7
文章标签： 人工智能 机器学习 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Qingxingxx/article/details/136026203
版权

自己的数据集约6k训练集，3k测试集。
直接跑detr的代码会出现MAP全是0的情况。

有效的方法：
1、main.py里--eos_coef的default改小，改成0.01
2、修改--lr_drop和lr_scheduler里的衰减参数，调整学习率衰减策略

总体感觉detr训练比较慢，精度上升得也慢。

Brownie_pancake
关注
  • 7
    点赞
  • 踩
  • 12
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
detr上板训练编译结果
04-10
detr上板训练编译结果
BETR预训练模型（num-classes = 8）
07-11
BETR预训练模型（num-classes = 8）
关于DETR模型训练自定义数据集AP=0问题
pengxiang1998的博客
03-01 2019
刚开始时，博主还以为是初始训练的问题，后来进行了6个轮次后结果依旧没有改变，此时以为是数据集的类别与编号没有对应造成的，但经过排查却并非是这个原因。这时博主就很奇怪了，先前不是已经给了ResNet预训练权重了吗，怎么还会要求这个呢，查看了一下服务器的源码也没有发现异常，而在查看GPU使用情况时（如下图），博主此时的batch-size=32，但GPU利用率与显存占用率并不高，这与论文中提到的很吃配置明显是不符的。修改后再次运行，不再报错，也可以运行了，但缺依旧没有解决AP=0的问题。
深度学习：利用Detectron2训练自己的数据集之模型评估——解决AP=0问题
weixin_42565090的博客
03-10 4252
本文主要阐述模型评估相关内容，遇到的坑及解决办法 在进行模型评估的时候遇到有些自定义的类别的AP=0。经过
DETR源码笔记（一）
在努力的松鼠的博客
05-18 7705
源码获取：https://gitee.com/fgy120/DETR 首先对DETR做个简单介绍 上图即为DETR的流程pipeline，相比以前的RCNN系列、YOLO系列等，最特别的在于加入了Transformer。 直接看源码，从train.py的main函数开始。 if __name__ == '__main__': parser = argparse.ArgumentParser('DETR training and evaluation script', parents..
MMdetection3.0 问题：DETR验证集上AP值全为0.0000
MZYYZT的博客
12-02 845
2、在Faster-rcnn、YOLOv3模型上均能训练成功，但是在DETR训练时，出现验证指标全为0.0000的问题。1、选择的是NWPU-VHR-10数据集，其中训练集455张，验证、测试相同均为195张。是因为数据集量太小的原因吗？还是需要特别关注某些具体参数的调整呢？1、DETR验证集上AP值全为0.0000。请各位路过的大佬指教一下！
【已解决】RT-DETR /YOLOX用自己的数据集训练时，遇到AP一直为0的情况
m0_49687072的博客
03-14 505
1.这里的id和image_id应该从几开始。2. categories里的id应该从几开始。此外 我还有几点疑惑。
DETR类模型COCO训练结果异常，评测AP=0解决方案记录
qq_36396406的博客
12-13 1239
接着我想到了我在coco_minitrain上只实验过1enc+4dec与3enc+6dec的缩水版模型（为了快速验证，hidden_dim与num_query均也有相应调小），于是猜测是不是完整的6enc+6dec难收敛的问题，于是我将1enc+4dec与3enc+6dec在完整COCO上实验，发现能得到正常的结果。因为之前的1enc+4dec与3enc+6dec的缩水版模型的参数量相对较小，我将其bs分别调整为8与4，对于3080TI 12G，这刚好能卡到显存上限。由此看来，bs似乎就是那个罪魁祸首。
【YOLO】解决YOLOX训练时AP为0
MangoloD的博客
08-12 1万+
前段时间被YOLOX刷屏了，各个公众号都在强推，号称：性能超Yolov5，吊打一切YOLO 于是，我满怀期待，下载下源码，准备试一试．．． 一、问题描述 嗯，没错，遇到了一堆bug，不过好在都解决了，最后训练的时候 好家伙，AP一直为0，上GIthub上一搜索，看来遇到同样问题的不止我一个 而且根据有些答复修改，发现也不对，最后陷入沉思．．． 直到昨天群组里偶然有人说他用YOLOX成功还部署了，我就呆了，告诉了他我训练时AP为0，并询问了他具体使用方法，当然得到的答复并没有真正解决问题．．． 二、发现问题
DETR训练自己的数据集
孟孟单单的博客
03-28 2632
DETR训练自己的数据集
mmdetection算法之DETR（0）
weixin_41311686的博客
06-09 704
基本网络结构 backbone:resnet50 neck:无 head:transformer
py-faster-rcnn AP = 0 Mean AP = 0
dream2017seek的博客
07-04 1700
python3下，py-faster-rcnn训练自己数据集时mean ap =0一、实验环境二、问题三、问题分析问题一：考虑标记错误(修改后没有解决问题)问题二：源码需要针对kitti进行修改(修改后没有解决问题)问题三：训练的次数不够，不能检测出来(修改后没有解决问题)附Git查找：Mean AP = 0.0000（无结果）问题四：进行召回率计算时，参数计算错误(修改后成功解决问题)四、解决思路五、计算召回率文件代码voc_eval.py六、展望 一、实验环境 python3 ubuntu16.04 c
全景分割-基于FAIR-DETR对Cityscapes数据集进行微调实现全景分割-附项目源码-优质项目实战.zip
最新发布
05-12
全景分割_基于FAIR-DETR对Cityscapes数据集进行微调实现全景分割_附项目源码_优质项目实战
深度学习领域detr算法在小麦头目标检测（带数据集）-2、end-to-end-object-detection-with-t
06-18
深度学习领域detr算法在小麦头目标检测（带数据集）--2、end-to-end-object-detection-with-transformers-detr 语言：python 内容包括：源码、数据集、数据集描述 目的：使用detr算法在小麦头中目标检测。 带数据集...
rt-detr目标检测+python+tensorRT推理代码
06-28
该代码实现rt-detr的部署，使用python，调用tensorRT转换后的模型，实现单张图片或者图片文件夹批量推理，代码输入图片/文件夹路径，模型路径，输出图片保存路径，运行命令python ./infer_tensorrt.py --infer_dir=....
MMDetection实验记录踩坑记录
pengxiang1998的博客
09-08 722
因为在此之前已经运行过YOLOX,Faster-RCNN等模型，所以数据集的设置肯定是没有问题的，而博主也只是修改了DAB-DETR的类别参数而已，一想到这，才猛然想起，DETR类模型的类别数还需要加上一类，即背景类，先前在修改参数时，看到原始参数为80，还以为不用加1了呢。MMDetection针对模型在训练过程中意外中断，避免重头训练，设计了断点训练参数 resume，只需要将该参数设置为最后一次训练的权重即可。在实验MMDetection的DAB-DETR模型进行实验时，AP值始终上不去。
深度学习之目标检测（十一）--DETR详解
热门推荐
木卯_THU的博客
09-26 9万+
深度学习之图像分类（十一）DETR详解 深度学习之目标检测（十一）-- DETR详解深度学习之图像分类（十一）DETR详解1. 前言2. DETR 框架2.1 CNN Backbone2.2 Transformer Encoder2.3 Transformer Decoder2.4 FFN3. 二分图匹配和损失函数4. 代码5. 总结 继 Transformer 应用于图像分类后，本章学习 Transformer 应用于图像目标检测的开山之作 – DEtection TRansformer，其大大简化了目标
关于eval.py中MAP，AP计算为0的问题，已经解决！！！（若不是类别名字大小写问题，可尝试参考本文）
lgt3402788288的博客
06-14 4432
本篇文章主要记录 eval.py 中遇到的问题！！！ 导致MAP，AP 计算为0！！！ 本文始于学习SSD时，运用训练的权重做检测发现可行，但是运用eval.py去计算MAP结果只为0. 博主在这一个问题上犯了两个错，本文提供的方法不仅仅适用于SSD的eval.py 同样适用于解决其他项目中的voc_eval.py。 博主思考查询了一晚上，若对你有帮助，给个赞！！！ 两个错误： 1、 修改voc_eval函数，出现AP，MAP为零，是由于npos，rec, tp，fp等这几个参数计算出了问题。也就是rec
DETR训练自己数据集
10-11
要训练自己的数据集，需要进行以下步骤： 1. 准备数据集，包括图像和对应的标注文件。 2. 将标注文件转换为COCO格式，即包括图像信息和目标检测信息的JSON文件。 3. 修改DETR代码中的数据加载部分，使其能够读取自己的数据集。 4. 进行训练并调整超参数，如学习率、batch size等。 5. 在训练过程中进行模型评估，并根据评估结果进行调整和优化。

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值