一种基于注意力机制的启发式图像尺度归一化方法

摘要

本发明公开了一种基于注意力机制的启发式图像尺度归一化算法。本方法首先使用一个具有大量数据的数据集训练出一个带有注意力机制的分类网络，该网络能够同时输出注意力特征图以及图像的分类结果。该网络在训练过程中使用分类结果来进行训练。利用注意力特征图具有量化图像中像素的重要程度的能力，本方法使用一个滑动窗口在注意力特征图上进行滑动。当滑动窗口内的值取得最大值时，该滑动窗口对应的位置即为原始图像的待裁剪位置。本发明可以通过这一方法将图像转变到指定的尺寸而不发生形变。当图像需要裁剪时，该方法会自动选择合适的区域进行裁剪。