Add mask for instance segmentation #900
-
Hello, I saw in your code that in the yolo v8 model, on line 62, you save only the output boxes from the model, it would be great if you also save the masks and display them in the output. Line 62 in 0414e47 prediction_result = [ |
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 1 reply
This comment has been hidden.
This comment has been hidden.
-
Segmentation mask support for yolov8, mmdetection, detectron2, torchvision models is live with the latest release! @FilipeMendesVMD: https://github.com/obss/sahi/discussions/1051 There is also 5x memory and speed improvement when working with masks! Check the demo notebook on how it works: https://github.com/obss/sahi/blob/main/demo/inference_for_yolov8.ipynb |
Beta Was this translation helpful? Give feedback.
Segmentation mask support for yolov8, mmdetection, detectron2, torchvision models is live with the latest release! @FilipeMendesVMD: https://github.com/obss/sahi/discussions/1051
There is also 5x memory and speed improvement when working with masks!
Check the demo notebook on how it works: https://github.com/obss/sahi/blob/main/demo/inference_for_yolov8.ipynb