
Yolov4-DeepSort-Headgear-Detection: Yolov4检测安全帽
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:本项目采用YOLOv4算法结合DeepSort跟踪技术,专注于检测施工现场工人佩戴的安全帽情况,保障作业安全。
Yolov4-deepsort头盔检测使用DarknetYOLOv4模型训练的头盔(安全帽)检测器,在Windows 10 x64 2020 (build 19041.388) 系统上进行测试,配备NVIDIA RTX 2070 Super GPU和CUDA 10.1/CuDNN 7.6.5。Python版本为3.7.7 x64,并使用TensorFlow 2.2.0 进行GPU训练。重量文件应放置在./configs目录下。
若要利用自定义数据集进行训练,需要调整一些参数并提供约100张图片的数据集。此项目依赖于Python库如opencv-python、numpy和scikit-image来绘制图像中的框与文字,并使用matplotlib创建颜色图。CUDA 10.1/CuDNN 7.6.5版本用于支持DarknetYOLOv4对象检测,相关文件包括dark.dll等。
全部评论 (0)
还没有任何评论哟~


