mirror of
https://github.com/azaion/detections.git
synced 2026-06-23 14:21:07 +00:00
Roman Meshko
6ad4b700dd
ci/woodpecker/push/02-build-push Pipeline was successful
* Run tests * Run tests * Run tests * Run tests * Added rebuild * Added files for e2e tests * Added rebuild * Added rebuild * Added biuld TensorRT flag * Changed to use NumPy 1.x for Jetson * Make universal invocation * Make Cython constans * Changed to prepare onnx * Changed smoke-test to wait AI conversion * Added step for model conversion * Changed to not run step in parallel * Push model to docker registry * Push model to docker registry * Push model to docker registry
23 lines
472 B
Cython
23 lines
472 B
Cython
from engines.inference_engine cimport InferenceEngine
|
|
|
|
|
|
cdef class TensorRTEngine(InferenceEngine):
|
|
|
|
cdef public object context
|
|
cdef object cuda_context
|
|
cdef object cuda_lock
|
|
|
|
cdef public object d_input
|
|
cdef public object d_output
|
|
cdef str input_name
|
|
cdef list input_shape
|
|
|
|
cdef object h_output
|
|
cdef str output_name
|
|
cdef list output_shape
|
|
|
|
cdef object stream
|
|
|
|
cdef tuple get_input_shape(self)
|
|
cdef run(self, input_data)
|