SangerY / NPU_CaffeSSD

RK3399 Pro NPU support for Caffe SSD detector

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

NPU_CaffeSSD

RK3399 Pro NPU support for Caffe SSD detector

assume you have installed RKNN toolkit and it's dependencies or you can install it first by

$ pip install rknn_toolkit-0.9.9-cp35-cp35m-linux_x86_64.whl

the package you can find at

  1. [Main Site]
  2. [Download Source Site]

you can find the pedestrian detecttion model of this project at [here].

Usage

extract priorbox (you need change the caffe_root path to yours in priorbox.py, or may need install caffe-ssd first, caffe-ssd you can find at [here])

$ python priorbox.py

convert model and run pedestrian detection (may take some time to run)

$ python3 npu_ssd_det.py

Result

you can use NMS algorithm to remove redundant rects.

About

RK3399 Pro NPU support for Caffe SSD detector

License:MIT License


Languages

Language:Python 100.0%