zl程序教程

您现在的位置是:首页 >  后端

当前栏目

python工具方法 15 基于opencv的多目标模板匹配

2023-09-14 09:15:04 时间

可以批量匹配文件夹下单图中多个目标

1、准备好模板图和要裁剪的子图

目录中template.png表示用于匹配的模板图,其余jpg后缀的图片是用于匹配的图

 2、运行以下代码,裁剪目标

import cv2
from skimage import transform
from matplotlib import pyplot as plt
import numpy as np
import glob,os
from skimage import io,morphology,color,transform
def calculate_IoU(predicted_bound, ground_truth_bound):
    pxmin, pymin, pxmax, pymax = predicted_bound
    gxmin, gymin, gxmax, gymax = ground_truth_bound
 
    parea = (pxmax - pxmin) * (pymax - pymin)  # 计算P的面积
    garea = (gxmax - gxmin) * (gymax - gymin)  # 计算G的面积
 
    # 求相交矩形的左下和右上顶点坐标(xmin, ymin, xmax, ymax)
    xmin = max(pxmin, gxmin)  # 得到左下顶点的横坐标
    ymin = max(pymin, gymin