Released code for Causal Attention for Unbiased Visual Recognition
This repo contains the codes of training our CaaM on NICO/ImageNet9 dataset. Due to my recent limited bandwidth, this codebase is still messy, which will be further refined and checked recently. 0. Bibtex If you find our codes helpful, please cite our paper: @inproceedings{wang2021causal, title={Causal Attention for Unbiased Visual Recognition}, author={Wang, Tan and Zhou, Chang and Sun, Qianru and Zhang, Hanwang}, booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)}, year={2021} } 1. Preparation Installation: Python3.6, Pytorch1.6, tensorboard, timm(0.3.4), […]
Read more