espectre / GCNs_on_imbalanced_datasets

GCN-Based Linkage Prediction for Face Clustering on Imbalanced Datasets: An Empirical Study

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GCNs_on_imbalanced_datasets

This repo provides the code for our IJCAIW'21 oral paper: GCN-Based Linkage Prediction for Face Clustering on Imbalanced Datasets: An Empirical Study, by Huafeng Yang , Xingjian Chen , Fangyi Zhang∗ , Guangyue Hei , Yunjie Wang and Rong Du.

Due to work changes and differences in the company's internal and external network code systems, if there are any problems, please submit an issue.

GCN based face clustering papers collection

  1. Linkage-based Face Clustering via Graph Convolution Network, CVPR 2019 [Code]
  2. Learning to Cluster Faces on an Affinity Graph, CVPR 2019 (Oral) [Project Page]
  3. Learning to Cluster Faces via Confidence and Connectivity Estimation, CVPR 2020 [Project Page]
  4. Density-Aware Feature Embedding for Face Clustering, CVPR 2020

Requirements

  • Python >= 3.6
  • PyTorch >= 0.4.0
  • faiss

Datasets

1. Imbalanced MS1M

Google drive

2. Imbalanced DeepFashion

Google drive

Run

cd imbalance_gcn

1. Imbalanced MS1M

./run_train_test_dist_sup_random_celeb.sh

2. Imbalanced DeepFashion

./run_train_test_dist_sup_random_deepfashion.sh

Citation

Please cite the following paper if you use this repository in your reseach.

@article{yang2021gcn,
  title={GCN-Based Linkage Prediction for Face Clusteringon Imbalanced Datasets: An Empirical Study},
  author={Yang, Huafeng and Chen, Xingjian and Zhang, Fangyi and Hei, Guangyue and Wang, Yunjie and Du, Rong},
  journal={arXiv preprint arXiv:2107.02477},
  year={2021}
}

Ackownledgement

This repo was developed based LTC, many thanks to Lei Yang !

About

GCN-Based Linkage Prediction for Face Clustering on Imbalanced Datasets: An Empirical Study


Languages

Language:Python 91.9%Language:Shell 4.3%Language:Jupyter Notebook 3.8%