/SmartHelmetSystem

Primary LanguageJupyter Notebook

SmartHelmetSystem

此项目只用于进行安全帽目标探测,不包括配套的推流和前端项目,如需要整个项目请自行下载并按照技术文档中的部署说明部署

本检测项目部署

要求环境

  • 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]

引用