使用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)
相关推荐
weixin_377634842 小时前
【YOLO模型】参数全面解读
yolo
武乐乐~5 小时前
论文精读:YOLO-UniOW: Efficient Universal Open-World Object Detection
人工智能·yolo·目标检测
DragonnAi6 小时前
【目标检测标签转换工具】YOLO 格式与 Pascal VOC XML 格式的互转详解(含完整代码)
xml·yolo·目标检测
彭祥.1 天前
大疆无人机搭载树莓派进行目标旋转检测
yolo·目标检测·目标跟踪
武乐乐~2 天前
YOLO-World:基于YOLOv8的开放词汇目标检测
人工智能·yolo·目标检测
小草cys2 天前
查看YOLO版本的三种方法
人工智能·深度学习·yolo
萧霍之2 天前
基于onnxruntime结合PyQt快速搭建视觉原型Demo
pytorch·python·yolo·计算机视觉
多巴胺与内啡肽.2 天前
YOLOv2框架深度解析
yolo
向哆哆3 天前
UniRepLknet助力YOLOv8:高效特征提取与目标检测性能优化
人工智能·yolo·目标检测·yolov8
孤独野指针*P3 天前
深度学习中的目标检测:从 PR 曲线到 AP
python·深度学习·yolo