摘要
本发明公开的一种基于改进的语义分割网络的实时图像检索方法,包括以下步骤:S1、收集检索图像组成需要训练的数据集,并对数据集进行扩充,将扩充后的数据集分为训练集、测试集、验证集;S2、构建改进的实时计算的语义分割网络;S3、将训练集的图片输入改进的实时计算的语义分割网络,得到图片的语义向量,按照图片的类别信息存储语义向量,得到图像检索的语义向量数据库;S4、将测试集中待检测图片送入改进的实时计算的语义分割网络得到语义特征向量,比较待检索的图片的语义类别和图像检索语义向量数据库中向量的图像得到包含带检索图片语义类别的候选图片组;S5、将待检测图片的语义特征向量和候选图片组语义特征向量的进行匹配。
- 单位