Skip to content

Latest commit

 

History

History
25 lines (18 loc) · 319 Bytes

README.md

File metadata and controls

25 lines (18 loc) · 319 Bytes

custom layer(plugin)

How to Run

  1. build custom layer plugin
cd plugin_src
mkdir build
cd build
cmake ..
make

// a file 'libpreproc_layer.so' will be generated in build directory.
  1. run test_custom_layer.py
pip install cuda-python
pip install tensorrt

python test_custom_layer.py
// Passed