diff --git a/models/common.py b/models/common.py index 781f999445db..374d691d72f0 100644 --- a/models/common.py +++ b/models/common.py @@ -521,6 +521,7 @@ def __init__(self, weights="yolov5s.pt", device=torch.device("cpu"), dnn=False, Binding = namedtuple("Binding", ("name", "dtype", "shape", "data", "ptr")) logger = trt.Logger(trt.Logger.INFO) with open(w, "rb") as f, trt.Runtime(logger) as runtime: + trt.init_libnvinfer_plugins(None, "") model = runtime.deserialize_cuda_engine(f.read()) context = model.create_execution_context() bindings = OrderedDict()