目标检测模型训练的邪修方法

总结

本次使用虫子数据集。由于数据集中虫子的分类我无法确定标注是否正确,因此感觉我的方法没有改进。只体验一下方法的效果。

一、数据集

使用insects数据集。

参考https://aistudio.baidu.com/projectdetail/4041847

二、别人训练的结果

参考https://developer.aliyun.com/article/1672458

三、我的第一轮训练的结果

3.1训练环境

使用paddledetection 2.7版本

3.1.1主配置文件

3.1.2 coco_detection_insects.yml

数据集7类数据,使用了coco格式进行训练,目录改为数据集的存储目录

3.1.3 runtime.yml

除路径外没有改变配置

3.1.4 optimizer_6x.yml

配置没有进行修改

3.1.5 rtdetr_r50vd.yml

这个文件完全没有修改,后续也不会进行修改,因此不完全展示

3.1.6 rtdetr_reader.yml

因为我的显存只有8G,因此这里的batchsize设置成3,其它的配置未做修改。

3.2训练结果

训练batch只有72,这是batch=72时的验证集的评估结果。

四、第二轮训练

4.1数据集的情况

由于对数据集不熟悉,对什么虫子是否有误标无法确定,只能相信标注的结果,另外对漏标的虫子也无法处理。

数据集采用随机生成,数据集的情况为

4.2 训练的结果

这个结果跟上一次的差别很大,原因是将图像的大小改成了640X640。

4.3 数据集标注存在的问题

图中置信度0.93的实际类别是acuminatus还是coleoptera无法确定,因此无法修改标签。

图中置信度0.83的虫子,实际类别是acuminatus还是coleoptera无法确定,因此无法修改标签。

图中置信度0.84的虫子,实际类别是acuminatus还是coleoptera无法确定,因此无法修改标签。

以下三张图中检出的标注感觉类别应该一致。

图中这个虫子没有标注框

相关推荐
毕竟是shy哥2 小时前
TSDD-UB:UB:一种基于纹理简化的去噪扩散模型, 用于超声 B 扫信号下的无监督缺陷检测
目标检测·缺陷检测·扩散模型·工业缺陷检测·无损检测·超声检测·无监督缺陷检测
春日见6 小时前
5分钟入门强化学习之动态规划算法与实现
大数据·人工智能·python·算法·机器学习·计算机视觉
yubo05097 小时前
计算机视觉第三课:在图像上画框、画文字、画圆点
opencv·计算机视觉·目标跟踪
迁移科技12 小时前
AI+3D视觉赋能汽车箱体智能上下料
人工智能·3d·自动化·视觉检测
yubo050913 小时前
计算机视觉第五课:给每个物体画 bounding box
人工智能·opencv·计算机视觉
AI人工智能+14 小时前
赋能智慧监管:基于深度学习的特种行业许可证全要素精准识别系统
计算机视觉·自然语言处理·ocr·特种行业许可证识别
weixin_4684668515 小时前
机器学习与深度学习新手区分指南
人工智能·python·深度学习·机器学习·计算机视觉·ai·机器视觉
yubo050915 小时前
计算机视觉第四课:寻找轮廓(自动框出所有物体)
人工智能·opencv·计算机视觉
fie888915 小时前
近红外与可见光图像融合的ICA变换:原理、实现与应用
图像处理·人工智能·计算机视觉
sali-tec16 小时前
C# 基于OpenCv的视觉工作流-章79-单位转换
图像处理·人工智能·opencv·算法·计算机视觉