[发明专利]一种基于HOG和视觉词袋的相似图像搜索方法和装置

摘要

本发明提供一种基于HOG和视觉词袋的相似图像搜索方法和装置，所述装置包括预处理单元、输入单元、高斯差分单元、选取单元、HOG特征提取单元、聚类单元、视觉词袋单元和训练单元。本发明的有益效果在于，使用高斯差分算子寻找稳定关键点并确定特征提取区域，减少了传统使用HOG特征提取时对整幅图像进行滑窗扫描操作，既节省了时间，又提高了效率。

主权项

一种基于HOG和视觉词袋的相似图像搜索装置，其特征在于，包括预处理单元、输入单元、高斯差分单元、选取单元、HOG特征提取单元、聚类单元、视觉词袋单元和训练单元；所述预处理单元，用于将训练的图像进行分类作为训练集图像，并对测试图像和所述训练集图像的大小进行归一化；所述输入单元，用于输入归一化后的所述测试图像和所述训练集图像；所述高斯差分单元，用于通过高斯差分算法寻找所述输入单元中的所述测试图像和所述训练集图像中每一幅图像的稳定关键点；所述选取单元，用于选取所述测试图像和所述训练集图像中每一幅图像的稳定关键点的最密集的前N个区域；所述HOG特征提取单元，用于对所述最密集的前N个区域进行HOG特征提取；所述聚类单元，用于通过K‑Means聚类算法对所述训练集图像中的所述HOG特征进行聚类，得到的n个聚类中心为视觉词袋的n个视觉单词；所述视觉词袋单元，用于在所述训练集图像的视觉单词中寻找与所述测试图像中的每一个特征最近邻的视觉单词，并统计所述测试图像中所找到的所有视觉单词出现的频率，得到的频率直方图为所述测试图像的视觉词袋；所述训练单元，用于使用支持向量机分类器对所述训练集图像的视觉词袋进行训练，得到的模型对所述测试图像的视觉词袋进行测试，得到所述测试图像的分类结果，反馈同一类别的图片。