search

spacewalk01 / tensorrt-yolov9

Cpp and python implementation of YOLOv9 using TensorRT API
https://github.com/WongKinYiu/yolov9
MIT License
109 stars 18 forks source link

YOLO V9 Seg support #9

Closed toystorynova closed 6 months ago

toystorynova commented 6 months ago

Seems segmentation was added? Would you be able to add it here? Would be more than appreciated, thanks!

spacewalk01 commented 6 months ago

Hi it seems there is a bug in segmentation code: https://github.com/WongKinYiu/yolov9/issues/233 I will try to work on it as soon as bug is fixed. Thanks for asking.

spacewalk01 commented 6 months ago

Implemented yolov9 pan/instance segmentation using onnx python: https://github.com/spacewalk01/yolov9-onnx-segmentation

spacewalk01 commented 6 months ago

I will try to extend it.

toystorynova commented 6 months ago

Was hoping for a C++ TensorRT example, is that possible? or is that bug you linked stopping you from that? Nevertheless, appreciate your work!

spacewalk01 commented 6 months ago

Yes i am very interested in working on c++ yolov9 segmentation using trt. Currently i am busy. I will try to work on it as soon as possible.

spacewalk01 commented 6 months ago

Thanks for asking