2025-06-14【视觉】批量筛选图集中包含某种物体对象的方法

要查找包含的图片

查找结果如下：

其中有一个是错误的。

代码

#
#
# import aircv as ac
#
# # 读取图像
# im_source = ac.imread('large.jpg')
# im_search = ac.imread('s.jpg')
# # 查找模板
# result = ac.find_template(im_source, im_search)
# print(result)import cv2
import numpy as np
import os
from PIL import Image
import cv2def imread_chinese(filepath):"""支持中文路径的图像读取函数"""image = Image.open(filepath).convert('RGB')  # 读取图像并转为RGBreturn cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)  # 转为OpenCV格式def find_template_in_image(big_img, small_img):"""使用 ORB 算法判断 big_img 是否包含 small_img。"""orb = cv2.ORB_create()kp1, des1 = orb.detectAndCompute(small_img, None)kp2, des2 = orb.detectAndCompute(big_img, None)if des1 is None or des2 is None:return False  # 没有提取到特征描述符bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)matches = bf.match(des1, des2)matches = sorted(matches, key=lambda x: x.distance)good_matches = [m for m in matches[:10] if m.distance < 50]  # 可调参数return len(good_matches) > 2  # 设定阈值def find_images_containing_template(image_folder, template_path, output_folder="matched"):"""遍历文件夹，筛选出包含模板的小图，并保存结果。"""small_img = cv2.imread(template_path)assert small_img is not None, "无法加载模板图像"matched_images = []for filename in os.listdir(image_folder):file_path = os.path.join(image_folder, filename)if filename.lower().endswith(('.png', '.jpg', '.jpeg')):big_img = imread_chinese(file_path)if big_img is None:print(f"无法加载图像（可能损坏）：{filename}")continueprint(f"正在处理：{filename}")if find_template_in_image(big_img, small_img):print(f"✅ 找到匹配：{filename}")matched_images.append(filename)# 可选：保存匹配成功的图像到指定目录if not os.path.exists(output_folder):os.makedirs(output_folder)cv2.imwrite(os.path.join(output_folder, filename), big_img)return matched_images# 示例调用
image_folder = 'images'          # 大图所在文件夹
template_path = 'small.jpg'     # 小图模板路径
output_list = find_images_containing_template(image_folder, template_path)print("包含模板的图片列表：")
for img_name in output_list:print(img_name)