-
Notifications
You must be signed in to change notification settings - Fork 606
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added yolov8engine.py for TensorRT-quantized YOLOv8 models #1046
base: main
Are you sure you want to change the base?
Conversation
@alsozatch, thanks for your contribution! Can you please check the errors in the tests and try to fix them? |
It looks like you are using 'isort' and 'black' to enforce some order for imports. I've not used these packages so I'm not sure what I'm looking at. Is the required import format specified somewhere? |
@alsozatch you can check this section, formatting is very easy and automated by a script 👍🏻 https://github.com/obss/sahi?tab=readme-ov-file#contributing |
Hey @alsozatch, @fcakyon, Will the integration and maintenance of the use of the engine format be available? Or should we implement it on our side? Thx. |
Model file to allow for SAHI inference on yolov8 TensorRT quantized .engine models. Used exactly the same as the base Yolov8 model, except
Here's a function to grab the config file information from the .pt file that you quantized from initially.
Still need to manually put in imgsz and half flag equivalent to the imgsz and half parameters you use to quantize it.