85 lines
2.9 KiB
YAML
85 lines
2.9 KiB
YAML
Collections:
|
|
- Name: Visual-Attention-Network
|
|
Metadata:
|
|
Training Data: ImageNet-1k
|
|
Training Techniques:
|
|
- AdamW
|
|
- Weight Decay
|
|
Architecture:
|
|
- Visual Attention Network
|
|
- LKA
|
|
Paper:
|
|
URL: https://arxiv.org/pdf/2202.09741v2.pdf
|
|
Title: "Visual Attention Network"
|
|
README: configs/van/README.md
|
|
Code:
|
|
URL: https://github.com/open-mmlab/mmclassification/blob/v0.23.0/mmcls/models/backbones/van.py
|
|
Version: v0.23.0
|
|
|
|
Models:
|
|
- Name: van-b0_3rdparty_in1k
|
|
Metadata:
|
|
FLOPs: 880000000 # 0.88G
|
|
Parameters: 4110000 # 4.11M
|
|
In Collection: Visual-Attention-Network
|
|
Results:
|
|
- Dataset: ImageNet-1k
|
|
Metrics:
|
|
Top 1 Accuracy: 75.41
|
|
Top 5 Accuracy: 93.02
|
|
Task: Image Classification
|
|
Weights: https://download.openmmlab.com/mmclassification/v0/van/van-tiny_8xb128_in1k_20220501-385941af.pth
|
|
Config: configs/van/van-b0_8xb128_in1k.py
|
|
- Name: van-b1_3rdparty_in1k
|
|
Metadata:
|
|
FLOPs: 2520000000 # 2.52G
|
|
Parameters: 13860000 # 13.86M
|
|
In Collection: Visual-Attention-Network
|
|
Results:
|
|
- Dataset: ImageNet-1k
|
|
Metrics:
|
|
Top 1 Accuracy: 81.01
|
|
Top 5 Accuracy: 95.63
|
|
Task: Image Classification
|
|
Weights: https://download.openmmlab.com/mmclassification/v0/van/van-small_8xb128_in1k_20220501-17bc91aa.pth
|
|
Config: configs/van/van-b1_8xb128_in1k.py
|
|
- Name: van-b2_3rdparty_in1k
|
|
Metadata:
|
|
FLOPs: 5030000000 # 5.03G
|
|
Parameters: 26580000 # 26.58M
|
|
In Collection: Visual-Attention-Network
|
|
Results:
|
|
- Dataset: ImageNet-1k
|
|
Metrics:
|
|
Top 1 Accuracy: 82.80
|
|
Top 5 Accuracy: 96.21
|
|
Task: Image Classification
|
|
Weights: https://download.openmmlab.com/mmclassification/v0/van/van-base_8xb128_in1k_20220501-6a4cc31b.pth
|
|
Config: configs/van/van-b2_8xb128_in1k.py
|
|
- Name: van-b3_3rdparty_in1k
|
|
Metadata:
|
|
FLOPs: 8990000000 # 8.99G
|
|
Parameters: 44770000 # 44.77M
|
|
In Collection: Visual-Attention-Network
|
|
Results:
|
|
- Dataset: ImageNet-1k
|
|
Metrics:
|
|
Top 1 Accuracy: 83.86
|
|
Top 5 Accuracy: 96.73
|
|
Task: Image Classification
|
|
Weights: https://download.openmmlab.com/mmclassification/v0/van/van-large_8xb128_in1k_20220501-f212ba21.pth
|
|
Config: configs/van/van-b3_8xb128_in1k.py
|
|
- Name: van-b4_3rdparty_in1k
|
|
Metadata:
|
|
FLOPs: 12220000000 # 12.22G
|
|
Parameters: 60280000 # 60.28M
|
|
In Collection: Visual-Attention-Network
|
|
Results:
|
|
- Dataset: ImageNet-1k
|
|
Metrics:
|
|
Top 1 Accuracy: 84.13
|
|
Top 5 Accuracy: 96.86
|
|
Task: Image Classification
|
|
Weights: https://download.openmmlab.com/mmclassification/v0/van/van-b4_3rdparty_in1k_20220909-f4665b92.pth
|
|
Config: configs/van/van-b4_8xb128_in1k.py
|