摘要：# 注意力模块 ## 空间注意力模块 对特征图每个位置进行attention调整，(x,y)二维调整，使模型关注到值得更多关注的区域上。 对于卷积神经网络，CNN每一层都会输出一个C x H x W的特征图，C就是通道，同时也代表卷积核的数量，亦为特征的数量，H 和W就是原始图片经过压缩后的图的高度 阅读全文

摘要：主要参数说明：dataset: 加载的数据集；batch_size: 批大小；shuffle：是否将数据打乱；sampler：样本抽样num_workers：使用多进程加载的进程数，0代表不使用多进程；collate_fn：如何将多个样本数据拼接成一个batch，一般使用默认的拼接方式即可；pin_ 阅读全文