前提:使用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()
2.效果图