zqt321的博客

原创 未穿戴安全帽反光衣的人脸识别

未穿戴安全帽反光衣的人脸识别

原创 yolov5-6.0/6.1加入SE、CBAM、CA注意力机制（理论及代码）

yolov5-6.0/6.1加入SE、CBAM、CA注意力机制（理论及代码）

原创 yolov5-v6.1发布

yolov5-v6.1发布！更加完整的工程项目！

原创 yolov5中使用denselayer替换focus，fpn结构改为bi-fpn

原创 python批量修改txt文件里的类别数，批量修改文件名

做工程项目时，经常会遇到需要将两个数据集融合到一个数据集里的情况，但是两个数据集里的类别数又要能够对应上，比如两个数据集都是二分类的，那类别标签就是0和1，这种情况下就需要将另一个数据集的标签改为2和3，这样才能将两个数据集融合到一个里面。直接上代码，更换成自己的路径和自己想要修改的类别就可以直接用。def modify_txt(oldtxt_path): for oldtxt in os.listdir(oldtxt_path): b = [] txt_r

原创 yolov5安全帽检测、反光衣检测、抽烟检测

yolov5模型改进、改进后的各项数据以及安全帽检测、反光衣检测、抽烟检测

原创 关于yolov5融合wbf算法的一些事情

最近发现很多私信我的小伙伴会纠结于一件事情：将yolov5网络中的nms算法用wbf算法替换掉，以此来提升网络准确率。那关于wbf算法是什么呢？weighted boxes fusion（边界框融合算法）既然已经到了要考虑替换nms的地步了，那我这里就默认大家已经熟悉了nms的流程，所以我就不多费口舌了，直接切入正题。（不知道nms如何工作的同学可以去c站搜nms，结果一箩筐）wbf算法对于冗余预测框的处理，是将高于设定阈值的框全部利用起来。利用冗余检测框的坐标、类别、置信度，求平均数，得到

原创 关于yolov5加注意力之后的yaml文件讲解（5.0版本和6.0版本通用），文末有一个大部分人都没注意到的细节

关于yolov5加注意力之后的yaml文件讲解（5.0版本和6.0版本通用），文末有一个大部分人都没注意到的细节

原创 关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比

yolov5-5.0版本加入注意力与yolov5-6.0版本的使用效果，项目为安全帽检测

原创 yolov5加入CBAM,SE,CA,ECA注意力机制，纯代码（22.3.1还更新）

CBAM,SE,ECA,CA注意力添加到yolov5网络中，5.0版本