基于深度学习yolov3的安全帽检测以及安全帽检测数据集
数据集和代码下载地址:下载地址
数据可视化:
根目录下运行命令: show_yolo_anno.py (注意脚本内相关参数配置 )
模型训练
模型训练
根目录下运行命令: python train.py (注意脚本内相关参数配置 )
注意:加载训练的配置参数的关键代码段如下,cfg/transport.data为训练coco交通工具数据集的配置参数文件。
if __name__ == \__main__:train(data_cfg =<