NFNet Pytorch实施
此回购包含来自未经归一化的高性能大型图像识别论文的,具有ImageNet高精度的预训练NFNet模型F0-F6。 小型模型的精度与EfficientNet-B7一样,但训练速度快了8.7倍。 大型模型在ImageNet上设置了新的SOTA top-1精度。
NFNet
F0
F1
F2
F3
F4
F5
F6 + SAM
最高为1的准确性Brock等人。
83.6
84.7
85.1
85.7
85.9
86.0
86.5
此实现的Top-1精度
82.82
84.63
84.90
85.46
85.66
85.62
待定
所有学分归的作者所有。 该在很大程度上受到其在出色的JAX实现的启发。 拜访他们的回购网站以作参考。
开始吧
git clone https://github.com/benjs/nfnet
1