Skip to content

Add support for batched inference, Ultralytics TRT inference, YOLOv6#1055

Draft
aliencaocao wants to merge 3 commits intoobss:mainfrom
aliencaocao:main
Draft

Add support for batched inference, Ultralytics TRT inference, YOLOv6#1055
aliencaocao wants to merge 3 commits intoobss:mainfrom
aliencaocao:main

Conversation

@aliencaocao
Copy link

@aliencaocao aliencaocao commented Jun 17, 2024

  • Add support for batched inference on slices
  • Add support for TensorRT for Ultralytics-based/originally YOLOv8 models (exported using Ultralytics)
  • Add support for YOLOv6 (requires yolov6 repo on sys.path)

This PR is not meant to be merged but rather act as a reference for future more robust integration of these changes, as some of them are hacky patches and makes assumptions about the use case.

However, feel free to cherry pick changes from here and merge them, especially the batched inference part

aliencaocao and others added 3 commits June 16, 2024 00:16
Add support for TensorRT for Ultralytics-based/YOLOv8 models
Add support for YOLOv6 (requires yolov6 repo on sys.path)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants