Skip to content

Latest commit

 

History

History
43 lines (34 loc) · 1.2 KB

README.md

File metadata and controls

43 lines (34 loc) · 1.2 KB

YOLOX-C

This is a repository for deploying YOLOX using onnxruntime.

How to install

install onnxruntime from source with the following script

    # onnxruntime needs newer cmake version to build
    bash ./scripts/install_latest_cmake.bash

    # This place may be very slow, see the bash file for more detail.
    bash ./scripts/install_onnx_runtime.bash

    # dependencies to build apps
    bash ./scripts/install_apps_dependencies.bash

How to build

CPU

make default

# build examples
make apps

GPU with CUDA

make gpu_default

make gpu_apps

How to test

  1. Download onnx model trained on COCO dataset from HERE, and test yolox_l, you can try with other yolox models also.
wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_l.onnx -O ./data/yolox_l.onnx
  1. Test inference examples
    put inference model in models/, test images in datas/, and the results images will be put in results/.
./build/examples/yolox ./models/yolox_s.onnx ./data