[发明专利]图像定位方法、装置、系统和存储介质

权利要求书

1.一种图像定位方法，其特征在于，所述图像定位方法包括：

获取待处理图像；以及

利用训练好的生成对抗网络对所述待处理图像进行图像定位；

其中，所述利用训练好的生成对抗网络对所述待处理图像进行图像定位包括：

基于输入的所述待处理图像和噪声，利用所述训练好的生成对抗网络的生成器得到生成包围盒，以作为所述图像定位的结果。

2.根据权利要求1所述的图像定位方法，其特征在于，所述图像定位方法还包括：

计算所述生成包围盒与真实包围盒的交并比，所述真实包围盒为与所述待处理图像相对应的真实包围盒；以及

如果所述交并比大于或等于预设阈值，则记作定位准确，反之则记作定位失败。

3.根据权利要求2所述的图像定位方法，其特征在于，所述图像定位方法还包括：计算所述图像定位的准确率。

4.根据权利要求1-3中的任一项所述的图像定位方法，其特征在于，所述生成对抗网络包括生成器、映射管道和判别器，其中：

所述生成器接收输入的图像和噪声，输出生成包围盒；

所述映射管道将输入的包围盒映射为包围盒热点图；以及

所述判别器判断输入的包围盒热点图是真实包围盒热点图还是生成包围盒热点图。

5.根据权利要求4所述的图像定位方法，其特征在于，所述生成器包括卷积层、池化层、加载预训练权重的残差单元、加载随机权重的残差单元、全局池化层以及多输出全连接层。

6.根据权利要求4所述的图像定位方法，其特征在于，所述判别器包括卷积层、池化层、加载预训练权重的残差单元、加载随机权重的残差单元、全局池化层以及单输出全连接层。

7.根据权利要求4所述的图像定位方法，其特征在于，所述生成对抗网络的训练包括：

将样本图像和噪声输入所述生成器，经由所述生成器输出生成包围盒；

将与所述样本图像相对应的真实包围盒和所述生成器输出的生成包围盒输入所述映射管道，经由所述映射管道输出真实包围盒热点图和生成包围盒热点图；

将所述样本图像与所述映射管道输出的真实包围盒热点图组合成对输入所述判别器，经由所述判别器输出针对所述真实包围盒热点图的判别结果；

将所述样本图像与所述映射管道输出的生成包围盒热点图组合成对输入所述判别器，经由所述判别器输出针对所述生成包围盒热点图的判别结果；

基于所述判别器输出的针对所述真实包围盒热点图的判别结果和针对所述生成包围盒热点图的判别结果，计算所述判别器的损失值；

基于所述判别器输出的针对所述生成包围盒热点图的判别结果，计算所述生成器的损失值；以及

基于所述生成器的损失值和所述判别器的损失值，优化所述生成器和所述判别器。

8.根据权利要求7所述的图像定位方法，其特征在于，所述基于所述判别器输出的针对所述真实包围盒热点图的判别结果和针对所述生成包围盒热点图的判别结果，计算所述判别器的损失值包括：

基于所述判别器输出的针对所述真实包围盒热点图的判别结果与预定义的全1向量，计算第一交叉熵；

基于所述判别器输出的针对所述生成包围盒热点图的判别结果与预定义的全0向量，计算第二交叉熵；以及

计算所述第一交叉熵和所述第二交叉熵的平均值，以作为所述判别器的损失值。

9.根据权利要求7所述的图像定位方法，其特征在于，所述基于所述判别器输出的针对所述生成包围盒热点图的判别结果，计算所述生成器的损失值包括：

基于所述判别器输出的针对所述生成包围盒热点图的判别结果与预定义的全1向量，计算第三交叉熵，以作为所述生成器的损失值。