Paddle2.0中的RAN注意力卷积网络如何优化性能？

本文共计982个文字，预计阅读时间需要4分钟。

相关专题

论文原文：Residual Attention Network for Image Classification

参考代码：

PyTorch的实现

数据集介绍

本项目使用10分类的动物数据集进行训练和测试.

该十分类动物数据集，包含dog,horse,elephant,butterfly,chicken,cat,cow,sheep,spider和squirrel。每一分类的图片数量为2k-5k。

文件结构

文件名或文件夹名	功能
ran.py	ran网络定义文件
attention_module.py	注意力模块定义文件
basic_layers.py	残差模块定义文件
animal_dataset.py	数据集定义文件
config.py	配置文件
train_val_split.py	训练验证划分文件
train.py	模型训练
eval.py	模型验证