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