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)
