此项目只用于进行安全帽目标探测,不包括配套的推流和前端项目,如需要整个项目请自行下载并按照技术文档中的部署说明部署
要求环境
- Ubuntu18.04
- CUDA == 10.1
- CUDNN
- gcc and g++ >= 7.5
下载权重并放置到根目录
安装运行环境
# 安装 virtualenv
sudo pip3 install -U virtualenv
# 创建 virtual env
virtualenv --system-site-packages -p python3 ./venv
# 激活 virtual env
source venv/bin/activate
# 开始安装与编译依赖
python detection/setup.py develop测试视频
python tools/process_video.py detection/myconfigs/baseline/faster_rcnn_r50_fpn_1x.py fastr50_963.pth --input_path [video path] --output_path [save_path]引用