forked from jkjung-avt/tensorrt_demos
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pytrt.pxd
22 lines (19 loc) · 725 Bytes
/
pytrt.pxd
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
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()