yolov8-seg的后处理问题

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎

GNU General Public License v2.0

1.25k stars 195 forks source link

Open shitianyi opened 1 week ago

shitianyi commented 1 week ago

up,我看了下Yolov8-seg的后处理，发现只有bbox的解码是在cuda上运行的，而mask还是CPU的，后续有考虑基于CUDA实现mask的解码吗