Helmet-Detection-Track

安全帽检测+追踪

- Python 3.5.2
- Keras 2.1.5
- tensorflow 1.6.0

检测部分yolov3,追踪部分KCF算法

使用训练好的权重trained_weights_stage_1.h5,置于models/trained_weights_stage_1.h5

python traffic_main.py

其中：

config,py 中更改相关设置

保存的图片存放于vis文件夹中，路径可在config.py中更改

make_video.py将vis中图片合成视频

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
.idea		.idea
models		models
tracker		tracker
yolo3		yolo3
.DS_Store		.DS_Store
1582436108570269.mp4		1582436108570269.mp4
README.md		README.md
config.py		config.py
instance.py		instance.py
kcf.py		kcf.py
make_video.py		make_video.py
multiple_object_controller.py		multiple_object_controller.py
traffic_main.py		traffic_main.py
util.py		util.py
video_helper.py		video_helper.py
visualizer.py		visualizer.py
yolo.py		yolo.py