MMAL网
这是论文用于细粒度的PyTorch实施(张帆,李萌,翟桂生,刘亦钊)由第27届国际多媒体建模国际会议(MMM2021)提供。 欢迎与我们讨论问题!
目录
要求
的Python 3.7
pytorch 1.3.1
numpy的1.17.3
scikit图像0.16.2
Tensorboard 1.15.0
TensorboardX 2.0
tqdm 4.41.1
图像2.6.1
枕头6.1.0
数据集
下载数据集,并将提取的图像文件夹的内容复制到datasets / CUB 200-2011 / images中。
下载数据集并将提取的data / images文件夹的内容复制到datasets / FGVC_Aircraft / data / images )
您也可以尝试其他细粒度的数据集。
培训TBMSL-Net
如果要训练MMAL-Net,请在运行py
1