Files

custom_layer

Name		Name	Last commit message	Last commit date
parent directory ..
data		data
plugin_src		plugin_src
README.md		README.md
load_plugin_lib.py		load_plugin_lib.py
test_custom_layer.py		test_custom_layer.py

README.md

custom layer(plugin)

How to Run

build custom layer plugin

cd plugin_src
mkdir build
cd build
cmake ..
make

// a file 'libpreproc_layer.so' will be generated in build directory.

run test_custom_layer.py

pip install cuda-python
pip install tensorrt

python test_custom_layer.py
// Passed