一种基于PSA-Swin Transformer的图像分类算法

本发明公开了一种基于PSA‑Swin Transformer的图像分类算法，主要涉及深度学习图像分类领域；包括步骤：S1、首先将图片输入到Patch Partition模块中进行分块，然后在通道方向展平；S2、依次通过四个阶段构建不同大小的特征图；其中，第一个阶段中先通过一个Linear Embeding层后，再通过PSA‑Swin Transformer Block层；PSA‑Swin Transformer Block层包括EPE模块和PSA模块；S3、然后接上一个Layer Norm层、全局池化层以及全连接层得到最终输出；本发明是一种适用于处理小数据集的方法，可以有效地处理较小的数据量，而不需要预训练权值。