Yolov4-deepsort头盔检测
使用DarknetYOLOv4模型训练的头盔(安全帽)检测器。
测试环境
Windows 10 x64 2020 (build 19041.388)
NVIDIA RTX 2070 Super
CUDA 10.1 / CuDNN 7.6.5
Python 3.7.7 x64
tensorflow 2.2.0 GPU
训练体重
将重量文件放在./configs
要使用自己的数据集进行训练,您应该使用 。 并需要更改一些参数
使用的数据集
+约100张图片
依存关系
Python
opencv-python,numpy,scikit图像在图像上画框和文字
张量流2.2.0 使用DeepSORT模型跟踪对象
matplotlib 创建颜色图
CUDA 10.1 / CuDNN 7.6.5
暗网
用于yolov4对象检测
dark.dll,
1