Skip to content

AIT-Assistive-Autonomous-Systems/mmdeploy_tensorrt_ops

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Builds a selection of TensorRT operators of mmdeploy, currently only the deformable_conv2d pytorch operator.

To use the operators in python, load the dll from the package path at runtime:

import ctypes
from ament_index_python import get_package_prefix
plugins_path = os.path.join(get_package_prefix('mmdeploy_tensorrt_ops'), 'lib', 'libmmdeploy_tensorrt_ops.so')
plugins_handle = ctypes.CDLL(plugins_path, mode=ctypes.RTLD_GLOBAL)
assert(plugins_handle)

About

TensorRT operators of mmdeploy for ROS

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages