使用yolo识别模型对比两张图片并标记不同(2)

上篇文章有漏洞,在这里补充下,比如要识别第二张图相对于第一张图的违建是否拆除了

第一步旋转对其后,图片会有黑色的掩码,如果旋转角度大的话,没识别出来的框可能不是已经拆除了,而是因为黑色掩码遮挡,如下图:

思路:

将第二张图像中计算出的黑色掩码应用到第一张图像上,再拿两张图片用模型识别对象

python 复制代码
# 读取目标图像和计算出的掩码图像
target_img = cv2.imread('overlap1.jpg')
computed_mask_img = cv2.imread('overlap2.jpg', cv2.IMREAD_GRAYSCALE)

# 确保计算出的黑色掩码图像与目标图像具有相同的尺寸
computed_mask_img = cv2.resize(computed_mask_img, (target_img.shape[1], target_img.shape[0]))

# 将计算出的黑色掩码应用到目标图像上
result = cv2.bitwise_and(target_img, target_img, mask=computed_mask_img)

cv2.imwrite('overlap.jpg', result)
相关推荐
YOLO数据集集合14 小时前
天空反无人机检测数据集 | 无人机检测 多旋翼识别 固定翼识别 低空安防 目标检测9063期
人工智能·yolo·目标检测·计算机视觉·无人机·反无人机
深度学习lover15 小时前
<数据集>蚜虫识别<目标检测>
人工智能·深度学习·yolo·目标检测·计算机视觉·蚜虫识别
Together_CZ18 小时前
YOLO-IOD: Towards Real Time Incremental Object Detection——迈向实时增量目标检测
yolo·目标检测·yolo-iod·迈向实时增量目标检测·增量检测·incremental·real time
YOLO数据集集合18 小时前
固定翼无人机目标检测数据集 | 固定翼无人机 低空安防 目标检测 YOLO格式9062期
yolo·目标检测·无人机·固定翼无人机·固定翼·无人机识别·反无人机
深度学习lover21 小时前
<数据集>番茄叶片病害识别<目标检测>
人工智能·yolo·目标检测·计算机视觉·数据集·番茄叶片病害识别
YOLO数据集集合2 天前
福寿螺目标检测数据集 | 福寿螺检测 入侵物种 农业植保 目标检测9062期
人工智能·yolo·目标检测·计算机视觉·福寿螺·福寿螺检测·入侵物种
深度学习lover2 天前
<数据集>电力劳保穿戴识别<目标检测>
人工智能·yolo·目标检测·计算机视觉·数据集·电力劳保穿戴识别
深度学习lover2 天前
<数据集>安全带穿戴识别<目标检测>
人工智能·yolo·目标检测·计算机视觉·安全带穿戴识别
孤狼warrior3 天前
SCTR 五次失败的安全 BN 路由器
人工智能·python·深度学习·算法·安全·yolo
AI云海3 天前
计算机视觉之-YOLO目标检测
yolo·目标检测·计算机视觉