The pipeline basically is a custom license plate recognition,
- read image
2… load model 1 (Vehicle model, that detects license plate along with vehicles) (Tiny-yolo converted using trtexec)
- load model 2 (Character model, that detects two classes) (Tiny-yolo converted using trtexec)
- infer in model 1 (detects license plate and extracts it to send model)
- infer in model 2 (detects two classes in the license plate image)
We have been using onnx models for these pipelines and those work fine. Since we want a jetson tx2 board for deployment we wanted to use trt models. I cannot share the whole pipeline code, but I have attached two files. First containing the TRT_model class (reference form Class) which is used to read the engine file, and has a function for detection and returning output, and the second file has a snippet of code.
I can run the two models separately, but can’t run them one after another.
If these files are not enough i can send some extra files, with details. Thank you for your help.
trt_error.zip (35.0 KB)