Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# Reversible Column Networks
|
| 2 |
This repo is the official implementation of:
|
| 3 |
|
|
@@ -10,8 +13,11 @@ International Conference on Learning Representations (ICLR) 2023\
|
|
| 10 |
|
| 11 |
|
| 12 |
## Updates
|
| 13 |
-
|
| 14 |
-
RevCol
|
|
|
|
|
|
|
|
|
|
| 15 |
|
| 16 |
***2/10/2023***\
|
| 17 |
RevCol model weights released.
|
|
@@ -36,6 +42,7 @@ Initial commits: codes for ImageNet-1k and ImageNet-22k classification are relea
|
|
| 36 |
RevCol is composed of multiple copies of subnetworks, named columns respectively, between which multi-level reversible connections are employed. RevCol coud serves as a foundation model backbone for various tasks in computer vision including classification, detection and segmentation.
|
| 37 |
|
| 38 |
|
|
|
|
| 39 |
## Main Results on ImageNet with Pre-trained Models
|
| 40 |
|
| 41 |
| name | pretrain | resolution | #params |FLOPs | acc@1 | pretrained model | finetuned model |
|
|
@@ -47,7 +54,9 @@ RevCol is composed of multiple copies of subnetworks, named columns respectively
|
|
| 47 |
| RevCol-B<sup>\*</sup> | ImageNet-22K | 384x384 | 138M | 48.9G | 86.7 |[baidu](https://pan.baidu.com/s/1l8zOFifgC8fZtBpHK2ZQHg?pwd=rh58)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_base_22k.pth)| [baidu](https://pan.baidu.com/s/18G0zAUygKgu58s2AjCBpsw?pwd=rv86)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_base_22k_1kft_384.pth)|
|
| 48 |
| RevCol-L<sup>\*</sup> | ImageNet-22K | 224x224 | 273M | 39G | 86.6 |[baidu](https://pan.baidu.com/s/1ueKqh3lFAAgC-vVU34ChYA?pwd=qv5m)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_large_22k.pth)| [baidu](https://pan.baidu.com/s/1CsWmcPcwieMzXE8pVmHh7w?pwd=qd9n)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_large_22k_1kft_224.pth)|
|
| 49 |
| RevCol-L<sup>\*</sup> | ImageNet-22K | 384x384 | 273M | 116G | 87.6 |[baidu](https://pan.baidu.com/s/1ueKqh3lFAAgC-vVU34ChYA?pwd=qv5m)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_large_22k.pth)| [baidu](https://pan.baidu.com/s/1VmCE3W3Xw6-Lo4rWrj9Xzg?pwd=x69r)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_large_22k_1kft_384.pth)|
|
|
|
|
| 50 |
|
|
|
|
| 51 |
## Getting Started
|
| 52 |
Please refer to [INSTRUCTIONS.md](INSTRUCTIONS.md) for setting up, training and evaluation details.
|
| 53 |
|
|
@@ -75,4 +84,4 @@ If you have any questions about this repo or the original paper, please contact
|
|
| 75 |
year={2023},
|
| 76 |
url={https://openreview.net/forum?id=Oc2vlWU0jFY}
|
| 77 |
}
|
| 78 |
-
```
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
---
|
| 4 |
# Reversible Column Networks
|
| 5 |
This repo is the official implementation of:
|
| 6 |
|
|
|
|
| 13 |
|
| 14 |
|
| 15 |
## Updates
|
| 16 |
+
**3/15/2023***\
|
| 17 |
+
RevCol Huge checkpoint for segmentation released! Add visualization tools.
|
| 18 |
+
|
| 19 |
+
**3/9/2023***\
|
| 20 |
+
Detection, Segmentation Code and Model Weights Released.
|
| 21 |
|
| 22 |
***2/10/2023***\
|
| 23 |
RevCol model weights released.
|
|
|
|
| 42 |
RevCol is composed of multiple copies of subnetworks, named columns respectively, between which multi-level reversible connections are employed. RevCol coud serves as a foundation model backbone for various tasks in computer vision including classification, detection and segmentation.
|
| 43 |
|
| 44 |
|
| 45 |
+
|
| 46 |
## Main Results on ImageNet with Pre-trained Models
|
| 47 |
|
| 48 |
| name | pretrain | resolution | #params |FLOPs | acc@1 | pretrained model | finetuned model |
|
|
|
|
| 54 |
| RevCol-B<sup>\*</sup> | ImageNet-22K | 384x384 | 138M | 48.9G | 86.7 |[baidu](https://pan.baidu.com/s/1l8zOFifgC8fZtBpHK2ZQHg?pwd=rh58)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_base_22k.pth)| [baidu](https://pan.baidu.com/s/18G0zAUygKgu58s2AjCBpsw?pwd=rv86)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_base_22k_1kft_384.pth)|
|
| 55 |
| RevCol-L<sup>\*</sup> | ImageNet-22K | 224x224 | 273M | 39G | 86.6 |[baidu](https://pan.baidu.com/s/1ueKqh3lFAAgC-vVU34ChYA?pwd=qv5m)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_large_22k.pth)| [baidu](https://pan.baidu.com/s/1CsWmcPcwieMzXE8pVmHh7w?pwd=qd9n)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_large_22k_1kft_224.pth)|
|
| 56 |
| RevCol-L<sup>\*</sup> | ImageNet-22K | 384x384 | 273M | 116G | 87.6 |[baidu](https://pan.baidu.com/s/1ueKqh3lFAAgC-vVU34ChYA?pwd=qv5m)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_large_22k.pth)| [baidu](https://pan.baidu.com/s/1VmCE3W3Xw6-Lo4rWrj9Xzg?pwd=x69r)/[github](https://github.com/megvii-research/RevCol/releases/download/checkpoint/revcol_large_22k_1kft_384.pth)|
|
| 57 |
+
| RevCol-H<sup>\*+</sup> | Megdata-168M | pretrain 224 / finetune 640 | 2.1B | 2537 | 90.0 |[huggingface](https://huggingface.co/LarryTsai/RevCol/blob/main/revcol_models/classification/revcol_huge_megdata.pth)|[huggingface](https://huggingface.co/LarryTsai/RevCol/blob/main/revcol_models/classification/revcol_huge_megdata_in1k.pth)|
|
| 58 |
|
| 59 |
+
[+]: Note that we use a slightly different model on RevCol-H with one more branch from the bottom level to the top one. Later experiments prove that this connection is unnecessary, however, consider RevCol-H's training cost, we do not retrain it.
|
| 60 |
## Getting Started
|
| 61 |
Please refer to [INSTRUCTIONS.md](INSTRUCTIONS.md) for setting up, training and evaluation details.
|
| 62 |
|
|
|
|
| 84 |
year={2023},
|
| 85 |
url={https://openreview.net/forum?id=Oc2vlWU0jFY}
|
| 86 |
}
|
| 87 |
+
```
|