使用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)
相关推荐
stsdddd1 天前
【YOLO算法包裹背包行李箱塑料袋包装纸盒快递盒带目标检测数据集】
算法·yolo·目标检测
YOLO数据集集合2 天前
无人机航拍河道垂钓检测数据集|水域禁钓智能识别|YOLO目标检测实战 河道垂钓识别数据集|无人机水域巡检|非法垂钓检测|深度学习目标检测
yolo·目标检测·无人机
ZPC82102 天前
双目相机 + 点云 + YOLO 是机械臂抓取最标准、最精准的方案!
数码相机·yolo
动物园猫2 天前
水面5种垃圾目标检测数据集分享(适用于YOLO系列深度学习分类检测任务)
深度学习·yolo·目标检测
紫_龙2 天前
yolov8标注数据训练数据验证数据
yolo
YOLO数据集集合2 天前
滑坡智能识别|遥感卫星无人机多源影像数据集|深度学习语义分割开源基准
人工智能·深度学习·yolo·目标检测·视觉检测·无人机
白日做梦Q2 天前
Label Studio 安装与使用完整文档(可直接复制部署)
深度学习·yolo·计算机视觉
断眉的派大星2 天前
YOLO 算法为何使用 BCE(二元交叉熵)损失
算法·yolo
前网易架构师-高司机3 天前
带标注的跌倒检测数据集,识别率88.6%,10793张图,支持yolo,coco json,voc xml,文末有模型训练代码
yolo·数据集·摔倒·跌倒
全球通史3 天前
Jetson Nano YOLO 检测四尺度热力图实时显示
嵌入式硬件·yolo·ubuntu