Closed Terizian closed 2 years ago
model1_trt = TrtYOLO(model1, (model1Height, model1Width), model1_category_num)
In my latest code, you don't need to specify "height" and "width" of input to the model. Those would be automatically determined by the code. Please pull the latest code from this repo and modify your code accordingly.
Source code for reference: https://github.com/jkjung-avt/tensorrt_demos/blob/67c4eb0f5e8a46656eb65073e5685dc7654498d8/utils/yolo_with_plugins.py#L275
Thanks for pointing it out! It seems I was using the latest version of everything else except the inference code. I can confirm that it works after modifying the code to make trt_yolo.py.
Thanks again
I've followed the tutorial in the README with regards to YOLOv3. Here is a list of the package versions:
I converted custom trained yolov3 models using the code in the yolo folder. I then try to load the model to use for inference as follows:
model1_trt = TrtYOLO(model1, (model1Height, model1Width), model1_category_num)
This is the output from my code: