ZF4444 / MMAL-Net

This is a PyTorch implementation of the paper "Multi-branch and Multi-scale Attention Learning for Fine-Grained Visual Categorization (MMAL-Net)" (Fan Zhang, Meng Li, Guisheng Zhai, Yizhao Liu).
249 stars 57 forks source link

原论文指标 #23

Open snant opened 3 years ago

snant commented 3 years ago

你好,我在cub上跑到最好88.85(epoch93),在aircraft上最好94.18(epoch97),总的训练代数都是200,一直跑不到原论文指标,可能是什么原因导致的。

ZF4444 commented 3 years ago

你好,我一般最好的指标是在epoch100以后,还有你是跑的最新的代码吗?我添加了color distort进行数据增强。

snant commented 3 years ago

img = transforms.ColorJitter(brightness=0.2, contrast=0.2)(img)?就是这版代码吧,跟batch有没有关系?我的batch为4,比你的小一些

ZF4444 commented 3 years ago

有关系,目前的batch size是我试过最好的

snant commented 3 years ago

没有设置seed导致每次跑出来结果都不一样,这样是不是可以认为模型最好结果有一定随机性呢

ZF4444 commented 3 years ago

没有设置seed导致每次跑出来结果都不一样,这样是不是可以认为模型最好结果有一定随机性呢

对的

snant commented 3 years ago

能加个你的联系方式吗,还有一些问题想问一下

ZF4444 commented 3 years ago

我的名字里是qq号