yolov8进行实例分割提取分割掩码

Question

目前我遇到了问题，运行YOLOv11分割模型后的预测结果是将蓝色蒙版覆盖在原图上，但我想要根据预测出来的蓝色蒙版在原图上对目标区域进行分割，而且使其余部分为黑色,应该如何操作？

Answer 1

阿里嘎多学长整理AIGC生成，因移动端显示问题导致当前答案未能完全显示，请使用PC端查看更加详细的解答过程

Yolov8实例分割提取分割掩码

你遇到的问题是，YOLOv8模型预测结果是将蓝色蒙版覆盖在原图上，但你想要根据预测出来的蓝色蒙版在原图上对目标区域进行分割，并将其余部分设置为黑色。

解决方案：

1. 使用 OpenCV 库中的 cv2.bitwise_and() 函数将预测结果与原图进行位运算，生成目标区域的掩码。
2. 使用 cv2.fillPoly() 函数将掩码填充到目标区域中。
3. 使用 cv2.bitwise_not() 函数将掩码与原图进行位运算，生成黑色的背景。

核心代码：

import cv2

# 预测结果
prediction = cv2.imread('prediction.png')

# 原图
original_image = cv2.imread('original_image.png')

# 将预测结果与原图进行位运算，生成目标区域的掩码
mask = cv2.bitwise_and(prediction, original_image)

# 将掩码填充到目标区域中
cv2.fillPoly(mask, [mask], (0, 0, 0))

# 将掩码与原图进行位运算，生成黑色的背景
result = cv2.bitwise_not(mask, original_image)

# 保存结果
cv2.imwrite('result.png', result)

这个解决方案将生成一个黑色的背景图像，并将目标区域填充到其中。