Revert "import Tensorrt not in compile time in order to dynamically load tensorrt only if nvidia gpu is present"

This reverts commit 1c4bdabfb5.
This commit is contained in:
Alex Bezdieniezhnykh
2025-04-30 23:32:03 +03:00
parent 1c4bdabfb5
commit cf01e5d952
4 changed files with 20 additions and 12 deletions
+2 -2
View File
@@ -14,11 +14,11 @@ cdef class OnnxEngine(InferenceEngine):
model_meta = self.session.get_modelmeta()
print("Metadata:", model_meta.custom_metadata_map)
cpdef tuple get_input_shape(self):
cdef tuple get_input_shape(self):
shape = self.input_shape
return shape[2], shape[3]
cpdef int get_batch_size(self):
cdef int get_batch_size(self):
return self.batch_size
cpdef run(self, input_data):