Skip to content

Megvii Electric Moped Detector (ONNX based inference)

Notifications You must be signed in to change notification settings

megvii-research/MEMD

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Megvii Electric Moped Detector

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

gif1 gif2 gif3

完整视频: 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  :检测结果文件地址

gif1

About

Megvii Electric Moped Detector (ONNX based inference)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages