def __init__(self, engine): """__init__ # Arguments engine: path to the TensorRT engine file """ self.trtnet = pytrt.PyTrtMtcnn(engine, shape1, shape2, shape3) self.trtnet.set_batchsize(1)
def __init__(self, engine): """__init__ # Arguments engine: path to the TensorRT engine file """ self.trtnet = pytrt.PyTrtMtcnn(engine, (3, 710, 384), (2, 350, 187), (4, 350, 187)) self.trtnet.set_batchsize(1)
def __init__(self, engine): """__init__ # Arguments engine: path to the TensorRT engine file """ self.trtnet = pytrt.PyTrtMtcnn(engine, (3, 1064, 756), (2, 527, 373), (4, 527, 373)) self.trtnet.set_batchsize(1)
def __init__(self, engine): self.trtnet = pytrt.PyTrtMtcnn(engine, (3, 48, 48), (2, 1, 1), (4, 1, 1), (10, 1, 1))
def __init__(self, engine): self.trtnet = pytrt.PyTrtMtcnn(engine, (3, 216, 384), (2, 103, 187), (4, 103, 187)) self.trtnet.set_batchsize(1)