Name		Name	Last commit message	Last commit date
parent directory ..
configs		configs
datasets		datasets
models		models
runner		runner
README.md		README.md
__init__.py		__init__.py
confidence.py		confidence.py
deduce.py		deduce.py
extract.py		extract.py
main.py		main.py
test_gcn_e.py		test_gcn_e.py
test_gcn_v.py		test_gcn_v.py
train_gcn_e.py		train_gcn_e.py
train_gcn_v.py		train_gcn_v.py

README.md

Learning to Cluster Faces via Confidence and Connectivity Estimation

Paper

Learning to Cluster Faces via Confidence and Connectivity Estimation, CVPR 2020 [Project Page]

Test

Download the pretrained models in the model zoo.

Test GCN-V

sh scripts/vegcn/test_gcn_v_ms1m.sh

Test GCN-E

sh scripts/vegcn/test_gcn_e_ms1m.sh

Train

Train GCN-V

sh scripts/vegcn/train_gcn_v_ms1m.sh

Train GCN-E

sh scripts/vegcn/train_gcn_e_ms1m.sh