【mask】根据bbox提示同一张图片生成多个矩形框掩码

最新推荐文章于 2025-03-17 20:00:00 发布

YEAH-Han

最新推荐文章于 2025-03-17 20:00:00 发布

阅读量660

点赞数 9

文章标签： python mask labelimg

本文链接：https://blog.youkuaiyun.com/qq_48559526/article/details/136699999

版权

本文介绍了如何使用LabelImg获取图片中的多个目标（ROI）并创建对应的掩码，通过OpenCV的函数实现对图像进行区域填充，展示了从读取图片到保存掩码的完整代码过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前提：使用labelimg得到bbox

1.代码

import cv2
import numpy as np

# 读取图片
image = cv2.imread("D:\Desktop\mult_test\images\SL03509990_1694761223500.jpg")

# 假设我们有多个目标的ROI（感兴趣区域）
rois = [
    (565,635,1006,859),  # 目标1的坐标
    (1515,655,1607,733),  # 目标2的坐标
    (1872,610,2011,682)
]

# 创建一个全白的掩码，大小与原图一致
mask = np.ones(image.shape, dtype=np.uint8) * 255

# 为每个ROI创建掩码
for idx, roi in enumerate(rois, start=1):
    x1, y1, x2, y2 = roi
    # 在掩码上绘制黑色矩形
    mask = cv2.rectangle(mask, (x1, y1), (x2, y2), color=(0, 0, 0), thickness=-1)

    # 保存或展示结果
cv2.imwrite(f'mask_{idx}.jpg', mask)
    # cv2.imshow(f'Mask {idx}', mask)
    # cv2.waitKey(0)

# 关闭所有窗口
# cv2.destroyAllWindows()