MA-CNN
Multi-Attention-CNN
说明
本仓库只是对ICCV 2017 论文《Learning Multi-Attention Convolutional Neural Network for Fine-Grained Image Recognition》中的多注意力卷积神经网络进行模仿和复现
论文作者GitHub地址
本项目的重点是全连接层聚类获得的注意力矩阵和通道损失函数
本项目主要用于图像中细粒度特征的识别和视频流中关键帧的提取
本项目由Pytorch实现,只能运行在GPU上
代码
data_macnn.py
对外提供数据接口,面对不同数据集需要修改继承于Pytorch数据相关模块的三个函数
conv_macnn.py
训练macnn的卷积部分,可以任选卷积网络,注意好输出的通道数量和尺寸就好
part_macnn.py
训练macnn的聚类部分
Dis已实现,D
2021-09-10 19:25:36
19KB
Python
1