Skip to content

This is a repository for deploying YOLOX using onnxruntime.

Notifications You must be signed in to change notification settings

iamstarlee/YOLOX-C

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

37 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

This is a repository for deploying YOLOX using onnxruntime.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published