- 博客(10)
- 资源 (1)
- 收藏
- 关注
原创 yolov5-6.0/6.1加入SE、CBAM、CA注意力机制(理论及代码)
yolov5-6.0/6.1加入SE、CBAM、CA注意力机制(理论及代码)
2022-03-21 22:39:53
28845
81
原创 python批量修改txt文件里的类别数,批量修改文件名
做工程项目时,经常会遇到需要将两个数据集融合到一个数据集里的情况,但是两个数据集里的类别数又要能够对应上,比如两个数据集都是二分类的,那类别标签就是0和1,这种情况下就需要将另一个数据集的标签改为2和3,这样才能将两个数据集融合到一个里面。直接上代码,更换成自己的路径和自己想要修改的类别就可以直接用。def modify_txt(oldtxt_path): for oldtxt in os.listdir(oldtxt_path): b = [] txt_r
2022-01-17 17:00:00
1572
2
原创 关于yolov5融合wbf算法的一些事情
最近发现很多私信我的小伙伴会纠结于一件事情:将yolov5网络中的nms算法用wbf算法替换掉,以此来提升网络准确率。那关于wbf算法是什么呢?weighted boxes fusion(边界框融合算法)既然已经到了要考虑替换nms的地步了,那我这里就默认大家已经熟悉了nms的流程,所以我就不多费口舌了,直接切入正题。(不知道nms如何工作的同学可以去c站搜nms,结果一箩筐)wbf算法对于冗余预测框的处理,是将高于设定阈值的框全部利用起来。利用冗余检测框的坐标、类别、置信度,求平均数,得到
2022-01-02 19:44:13
7570
25
原创 关于yolov5加注意力之后的yaml文件讲解(5.0版本和6.0版本通用),文末有一个大部分人都没注意到的细节
关于yolov5加注意力之后的yaml文件讲解(5.0版本和6.0版本通用),文末有一个大部分人都没注意到的细节
2021-12-12 15:23:02
5054
39
原创 关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
yolov5-5.0版本加入注意力与yolov5-6.0版本的使用效果,项目为安全帽检测
2021-12-08 17:46:17
9719
12
原创 yolov5加入CBAM,SE,CA,ECA注意力机制,纯代码(22.3.1还更新)
CBAM,SE,ECA,CA注意力添加到yolov5网络中,5.0版本
2021-12-07 17:33:47
73331
220
diou-diounms.txt
2021-12-26
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人