使用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)
相关推荐
人工智能教学实践16 小时前
基于 yolov8_pyqt5 自适应界面设计的火灾检测系统 demo:毕业设计参考
qt·yolo·课程设计
paradoxjun20 小时前
YOLOv8源码修改(4)- 实现YOLOv8模型剪枝(任意YOLO模型的简单剪枝)
深度学习·yolo·目标检测·剪枝
EelBarb1 天前
YOLO11/ultralytics:环境搭建
python·yolo·ultralytics
Archie_IT2 天前
基于云计算、大数据与YOLO设计的火灾/火焰目标检测
大数据·人工智能·yolo·目标检测·云计算
红色的山茶花2 天前
YOLOv10-1.1部分代码阅读笔记-val.py
笔记·深度学习·yolo
干啥都是小小白4 天前
yolov5错误更改与相关参数详解(train.py)
人工智能·算法·yolo
码上飞扬6 天前
YOLOv8:目标检测与实时应用的前沿探索
人工智能·yolo·目标检测
红色的山茶花7 天前
YOLOv11-ultralytics-8.3.67部分代码阅读笔记-conv.py
笔记·深度学习·yolo
呆呆珝8 天前
RKNN_C++版本-YOLOV5
c++·人工智能·嵌入式硬件·yolo
FL16238631298 天前
汽车表面划痕刮伤检测数据集VOC+YOLO格式1221张1类别
深度学习·yolo·汽车