使用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)
相关推荐
贤小二AI14 小时前
贤小二c#版Yolov5 yolov8 yolov10 yolov11自动标注工具 + 免python环境 GPU一键训练包
人工智能·深度学习·yolo
zy_destiny1 天前
【工业场景】用YOLOv12实现饮料类别识别
人工智能·python·深度学习·yolo·机器学习·计算机视觉·目标跟踪
卧式纯绿1 天前
每日文献(八)——Part one
人工智能·yolo·目标检测·计算机视觉·目标跟踪·cnn
HABuo1 天前
【YOLOv8】YOLOv8改进系列(12)----替换主干网络之StarNet
人工智能·深度学习·yolo·目标检测·计算机视觉
AdaCoding2 天前
YOLOv8架构详解
yolo·网络结构图
云卷云舒___________2 天前
【Ultralytics YOLO COCO 评估脚本 | 获得COCO评价指标】
yolo·coco·ultralytics
plmm烟酒僧3 天前
在 RK3588 多线程推理 YOLO 时,同时开启硬件解码和 RGA 加速的性能分析
yolo·rkmpp·瑞芯微·硬件加速·rga·色彩空间转换
HABuo3 天前
【YOLOv8】YOLOv8改进系列(11)----替换主干网络之MobileNetV4
人工智能·深度学习·yolo·目标检测·计算机视觉
Kai HVZ3 天前
《深度学习》——yolov4详解
人工智能·深度学习·yolo