NVIDIA-AI-IOT / face-mask-detection

Face Mask Detection using NVIDIA Transfer Learning Toolkit (TLT) and DeepStream for COVID-19
MIT License
241 stars 94 forks source link

How to use the model generated by pipeline in Deepstream python? #16

Closed lakshaychhabra closed 3 years ago

lakshaychhabra commented 3 years ago

Hi, I have successfully generated a tlt and an etlt file but how to run these in the python deepstream pipeline? I m following the Nvidia python apps deepstream repo but Whenever I gave the path of etlt model it shows uff error.

pp:619:gst_nvinfer_logger: NvDsInferContext[UID 1]: Info from NvDsInferContextImpl::buildModel() [UID = 1]: Trying to create engine from model files ERROR: [TRT]: UffParser: Could not read buffer. parseModel: Failed to parse UFF model ERROR: failed to build network since parsing model errors. ERROR: Failed to create network using custom network creation function ERROR: Failed to get cuda engine from custom library API 0:00:04.676202639 24497 0x1f46aa30 ERROR nvinfer gstnvinfer.cpp:613:gst_nvinfer_logger: NvDsInferContext[UID 1]: Error in NvDsInferContextImpl::buildModel() [UID = 1]: build engine file failed

lakshaychhabra commented 3 years ago

I had this error because of the KEY issue, Advice for fellow devs, Please save your key somewhere in your system and in your notebook assign the value of key directly in place of tlt_encode.