README.md

NCE-Net

Code for "Nearest Neighbor Classifier Embedded Network for Active Learning", AAAI 2021

Environments

• Ubuntu 18.04 LTS
• NVIDIA RTX 2080Ti
• Python 3.6
• Pytorch 1.1.0
• Cuda 10.2

Train and Test

To train, test and evaluate NCE-Net, you can simply run:

python main.py

If you find this repository useful for your publications, please consider citing our paper.

@article{wan2021ncenet,
    title={Nearest Neighbor Classifier Embedded Network for Active Learning},
    author={Fang Wan, Tianning Yuan, Mengying Fu, Xiangyang Ji, Qingming Huang and Qixiang Ye}
    booktitle = {AAAI 2021},
    year = {2021}
}