TensorRT-Demo/pytrt.pxd

23 lines
725 B
Cython
Raw Normal View History

2023-03-06 20:44:29 +08:00
from libcpp.string cimport string
cdef extern from 'trtNet.cpp' namespace 'trtnet':
pass
cdef extern from 'trtNet.h' namespace 'trtnet':
cdef cppclass TrtGooglenet:
TrtGooglenet() except +
void initEngine(string, int *, int *)
void forward(float *, float *)
void destroy()
cdef cppclass TrtMtcnnDet:
TrtMtcnnDet() except +
void initDet1(string, int *, int *, int *)
void initDet2(string, int *, int *, int *)
void initDet3(string, int *, int *, int *, int *)
void setBatchSize(int)
int getBatchSize()
void forward(float *, float *, float *)
void forward(float *, float *, float *, float *)
void destroy()