megvii-research / MEMD

Megvii Electric Moped Detector (ONNX based inference)

Megvii Electric Moped Detector

包含电动车检测 ONNX 模型及推理代码

完整视频: demo_1, demo_2, demo_3

准备工作

安装 cuda 10.1 以及 cudnn 7.6.3

安装

git clone https://github.com/megvii-research/MEMD.git
cd MEMD
pip3 install -r requirements.txt
cd memd/utils/lib_nms && bash compile.sh && cd ../../..

推理

python3 scripts/inference.py  --detector ./models/model.onnx --input-img ./demo/input.jpg --model-json ./models/config.json --output-path ./demo/output.jpg

--detector     ：onnx模型
--input-img    ：输入图片地址
--model-json   ：模型的配置文件地址
--output-path  ：检测结果文件地址