- 博客(3)
- 收藏
- 关注
原创 093自动生成xml目标检测框数据集-不用手动标注
这个是通过轮廓查找,自动将检测的到的轮廓外接矩形保存成xml文件,可以供目标检测数据集训练使用。首先打开02auto_xml.py文件,将里面第71行的阈值修改成上面获取得到的最合适的阈值,该生成的xml文件可以用labelimg打开检查生成效果,可用于目标检测数据集训练模型。就会将jpg文件夹里面的所有图片都进行轮廓查找,将外接矩形框保存成xml文件。回到代码部分,整个代码分为两个部分,(一个文件夹存放图片和两个py代码文件),弹出可视化界面,然后我们加载需要处理的图片,再选择合适的轮廓查找阈值。
2023-11-19 14:06:44
151
原创 085马路坑洼积水检测
运行03pyqt界面.py可以展示一个pyqt的可视化交互界面,通过点击按钮加载感兴趣的图片进行识别。运行01makeTxt.py会将data文件下的图片路径及标签保存在txt文本内,运行02train.py会对图片进行读取并训练模型保存在runs文件下,
2023-10-17 20:36:22
667
原创 086基于深度学习的安全带是否佩戴检测
在02中可以选择的模型有10多种,可以都训练进行对比、包括alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNeXt、ShuffleNet、VGG、EfficientNet和Swin transformer等10多种模型。需要大量标注好的含有安全带佩戴情况的图像数据集来训练模型,并进行适当的调参和评估,以获得较好的检测效果。传统的目标检测方法:可以使用一些经典的目标检测算法,如Faster R-CNN、YOLO等,训练一个模型来检测图像中的人和安全带。
2023-10-17 20:29:32
421
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人