search

laugh12321 / TensorRT-YOLO

🚀 你的YOLO部署神器。TensorRT Plugin、CUDA Kernel、CUDA Graphs三管齐下，享受闪电般的推理速度。| Your YOLO Deployment Powerhouse. With the synergy of TensorRT Plugins, CUDA Kernels, and CUDA Graphs, experience lightning-fast inference speeds.
https://github.com/laugh12321/TensorRT-YOLO
GNU General Public License v3.0
720 stars 81 forks source link

[Feature]: Hope to support YOLO Segmentation #12

Closed zhouyang986 closed 1 week ago

zhouyang986 commented 7 months ago

seg~

laugh12321 commented 7 months ago

会有这个计划，但是可能得很久以后了。可以看看别的现成的YOLOv8-TensorRT

zhouyang986 commented 7 months ago

好滴~是的！

laugh12321 commented 1 week ago

@zhouyang986 YOLO Segmentation 现已支持！trtyolo 提供对 yolov3-seg、yolov5-seg yolov8-seg yolo11-seg 的导出，YOLOv6,YOLOv7,YOLOv9 的 Segmentation 可参考 UltralyticsSegment 自行实现带插件的ONNX导出。C++ Python 推理模块均适用。