You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
15 lines
615 B
15 lines
615 B
2 years ago
|
# distutils: language = c++
|
||
|
|
# cython: c_string_encoding=ascii
|
||
|
|
|
||
|
|
from libcpp cimport bool
|
||
|
|
from libcpp.string cimport string
|
||
|
|
|
||
|
|
from .onnxmodel cimport ONNXModel as cppONNXModel
|
||
|
|
from selfdrive.modeld.models.commonmodel_pyx cimport CLContext
|
||
|
|
from selfdrive.modeld.runners.runmodel_pyx cimport RunModel
|
||
|
|
from selfdrive.modeld.runners.runmodel cimport RunModel as cppRunModel
|
||
|
|
|
||
|
|
cdef class ONNXModel(RunModel):
|
||
|
|
def __cinit__(self, string path, float[:] output, int runtime, bool use_tf8, CLContext context):
|
||
|
|
self.model = <cppRunModel *> new cppONNXModel(path, &output[0], len(output), runtime, use_tf8, context.context)
|