WZMIAOMIAO / deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.
GNU General Public License v3.0
22.75k stars 7.96k forks source link

Efficientnetv2在Imagenet上的推理精度验证? #705

Open qingtian5 opened 1 year ago

qingtian5 commented 1 year ago

大佬有没有验证过efficienctnetv2-s在imagenet1K验证集上的精度,我怎么推理出来的精度很低,只有12%,好奇怪呀?

WZMIAOMIAO commented 1 year ago

没有哦,我只是转了下权重,验证的话可以参考下官方原项目代码

qingtian5 commented 1 year ago

stride = 2 的时候,conv层错了,tf 和 pytorch 的 conv2D 不一样

应该改成 https://github.com/rwightman/pytorch-image-models/blob/29fda20e6d428bf636090ab207bbcf60617570ca/timm/layers/cond_conv2d.py#L34 这里的 conv2DSame