├── README.md ├── container.jpg └── conv_mlp_transformer.jpg /README.md: -------------------------------------------------------------------------------- 1 | # Container: Context Aggregation Network 2 | 3 | Official Code Release for Container : Context Aggregation Network 4 | 5 | # Comparion between CNN, MLP-Mixer and Transformer 6 | ![Container](conv_mlp_transformer.jpg) 7 | # Best of both worlds 8 | ![Container](container.jpg) 9 | 10 | Code release: 11 | 12 | 1. Imagenet pretrained model for Container V1 : [Container V1](https://github.com/allenai/container) 13 | 2. Imagenet pretrained model for Container V2 : [Container V2](https://github.com/gaopengcuhk/Contianer-V2) 14 | 15 | 16 | 17 | ## Citation 18 | If you find this repository useful, please consider citing our work: 19 | ``` 20 | @article{gao2021container, 21 | title={Container: Context Aggregation Network}, 22 | author={Gao, Peng and Lu, Jiasen and Li, Hongsheng and Mottaghi, Roozbeh and Kembhavi, Aniruddha}, 23 | journal={arXiv preprint arXiv:2106.01401}, 24 | year={2021} 25 | } 26 | ``` 27 | 28 | ## News 29 | Container got accepted by [Neuips 2021](https://nips.cc/Conferences/2021). 30 | 31 | ## Contributor 32 | Peng Gao 33 | 34 | ## Acknowledgement 35 | The codebase are borrowed heavyily from DeiT. 36 | 37 | -------------------------------------------------------------------------------- /container.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/Container/62a162ccea94efc922b0ee7843954cf40c585f2e/container.jpg -------------------------------------------------------------------------------- /conv_mlp_transformer.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/Container/62a162ccea94efc922b0ee7843954cf40c585f2e/conv_mlp_transformer.jpg --------------------------------------------------------------------------------