/SeCu

PyTorch Implementation for SeCu

Primary LanguagePythonApache License 2.0Apache-2.0

SeCu

PyTorch Implementation for Our ICCV'23 Paper: "Stable Cluster Discrimination for Deep Clustering"

Requirements

  • Python 3.8
  • PyTorch 1.6

Usage:

SeCu with size constraint for CIFAR-10

sh run_cifar10.sh 0

SeCu with entropy constraint for CIFAR-10

sh run_cifar10_entropy.sh 0

Citation

If you use the package in your research, please cite our paper:

@inproceedings{qian2023secu,
  author    = {Qi Qian},
  title     = {Stable Cluster Discrimination for Deep Clustering},
  booktitle = {{IEEE/CVF} International Conference on Computer Vision, {ICCV} 2023},
  year      = {2023}
}