├── .gitignore ├── DATASET.md ├── LICENSE ├── MODEL_ZOO.md ├── README.md ├── baseline ├── __init__.py ├── aro.py ├── chinese_whispers.py └── sklearn_cluster.py ├── dsgcn ├── README.md ├── __init__.py ├── configs │ ├── cfg_test_det_fashion_20_prpsls.py │ ├── cfg_test_det_fashion_2_prpsls.py │ ├── cfg_test_det_fashion_8_prpsls.py │ ├── cfg_test_det_ms1m_20_prpsls.py │ ├── cfg_test_det_ms1m_2_prpsls.py │ ├── cfg_test_det_ms1m_5_prpsls.py │ ├── cfg_test_det_ms1m_8_prpsls.py │ ├── cfg_test_det_ytb_4_prpsls.py │ ├── cfg_test_seg_ms1m_20_prpsls.py │ ├── cfg_test_seg_ms1m_2_prpsls.py │ ├── cfg_test_seg_ms1m_5_prpsls.py │ ├── cfg_test_seg_ms1m_8_prpsls.py │ ├── cfg_train_det_fashion_4_prpsls.py │ ├── cfg_train_det_fashion_84_prpsls.py │ ├── cfg_train_det_fashion_8_prpsls.py │ ├── cfg_train_det_ms1m_4_prpsls.py │ ├── cfg_train_det_ms1m_84_prpsls.py │ ├── cfg_train_det_ms1m_8_prpsls.py │ ├── cfg_train_seg_ms1m_4_prpsls.py │ ├── cfg_train_seg_ms1m_84_prpsls.py │ ├── cfg_train_seg_ms1m_8_prpsls.py │ └── yaml │ │ ├── cfg_test_0.7_0.75.yaml │ │ ├── cfg_test_hnsw_2_i0_18_i1.yaml │ │ ├── cfg_test_hnsw_2_i0_6_i1.yaml │ │ ├── cfg_train_0.7_0.75.yaml │ │ └── cfg_train_8_prpsl.yaml ├── datasets │ ├── __init__.py │ ├── build_dataloader.py │ ├── cluster_dataset.py │ ├── cluster_det_processor.py │ ├── cluster_processor.py │ ├── cluster_seg_processor.py │ └── sampler.py ├── main.py ├── models │ ├── __init__.py │ └── dsgcn.py ├── runner │ ├── __init__.py │ └── runner.py ├── test_cluster_det.py ├── test_cluster_seg.py ├── train.py ├── train_cluster_det.py └── train_cluster_seg.py ├── evaluation ├── __init__.py ├── evaluate.py └── metrics.py ├── lgcn ├── README.md ├── __init__.py ├── configs │ ├── cfg_test_lgcn_fashion.py │ ├── cfg_test_lgcn_ms1m.py │ ├── cfg_train_lgcn_fashion.py │ └── cfg_train_lgcn_ms1m.py ├── datasets │ ├── __init__.py │ ├── build_dataloader.py │ └── cluster_dataset.py ├── main.py ├── models │ ├── __init__.py │ └── lgcn.py ├── online_evaluation.py ├── test_lgcn.py └── train_lgcn.py ├── post_process ├── __init__.py ├── deoverlap.py └── nms.py ├── proposals ├── __init__.py ├── generate_basic_proposals.py ├── generate_iter_proposals.py ├── generate_proposals.py ├── graph.py ├── metrics.py ├── stat_cluster.py └── super_vertex.py ├── requirements.txt ├── scripts ├── baseline │ ├── aro.sh │ ├── aro_fashion.sh │ ├── aro_ytb.sh │ ├── cw_fashion.sh │ ├── cw_ms1m.sh │ ├── cw_ytb.sh │ ├── dbscan.sh │ ├── dbscan_fashion.sh │ ├── dbscan_ytb.sh │ ├── fast_hac.sh │ ├── fast_hac_fashion.sh │ ├── fast_hac_ytb.sh │ ├── kmeans_fashion.sh │ ├── kmeans_ytb.sh │ ├── meanshift_fashion.sh │ ├── mini_kmeans.sh │ ├── spectral.sh │ └── spectral_fashion.sh ├── dsgcn │ ├── step_by_step │ │ ├── deoverlap.sh │ │ ├── evaluate.sh │ │ ├── gcn_d_upper_bound.sh │ │ ├── generate_basic_proposals.sh │ │ ├── generate_iter_proposals.sh │ │ └── pipeline.sh │ ├── test_cluster_det_fashion.sh │ ├── test_cluster_det_iop_ms1m.sh │ ├── test_cluster_det_ms1m.sh │ ├── test_cluster_det_ytb.sh │ ├── test_cluster_seg_ms1m.sh │ ├── train_cluster_det_fashion.sh │ ├── train_cluster_det_iop_ms1m.sh │ ├── train_cluster_det_ms1m.sh │ └── train_cluster_seg_ms1m.sh ├── lgcn │ ├── test_lgcn_fashion.sh │ ├── test_lgcn_ms1m.sh │ ├── train_lgcn_fashion.sh │ └── train_lgcn_ms1m.sh ├── tools │ └── test_knn.sh └── vegcn │ ├── test_gcn_e_fashion.sh │ ├── test_gcn_e_ms1m.sh │ ├── test_gcn_v_fashion.sh │ ├── test_gcn_v_ms1m.sh │ ├── train_gcn_e_fashion.sh │ ├── train_gcn_e_ms1m.sh │ ├── train_gcn_v_fashion.sh │ └── train_gcn_v_ms1m.sh ├── tools ├── analyze_proposals.py ├── baseline_cluster.py ├── download_data.py ├── dsgcn_upper_bound.py └── test_knn.py ├── utils ├── __init__.py ├── adjacency.py ├── dataset.py ├── dist.py ├── draw.py ├── faiss_gpu.py ├── faiss_search.py ├── knn.py ├── logger.py ├── misc.py └── misc_cluster.py └── vegcn ├── README.md ├── __init__.py ├── confidence.py ├── configs ├── cfg_test_gcne_fashion.py ├── cfg_test_gcne_ms1m.py ├── cfg_test_gcnv_fashion.py ├── cfg_test_gcnv_ms1m.py ├── cfg_train_gcne_fashion.py ├── cfg_train_gcne_ms1m.py ├── cfg_train_gcnv_fashion.py └── cfg_train_gcnv_ms1m.py ├── datasets ├── __init__.py ├── gcn_e_dataset.py └── gcn_v_dataset.py ├── deduce.py ├── extract.py ├── main.py ├── models ├── __init__.py ├── gcn_e.py ├── gcn_v.py └── utils.py ├── runner ├── __init__.py └── runner.py ├── test_gcn_e.py ├── test_gcn_v.py ├── train_gcn_e.py └── train_gcn_v.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/.gitignore -------------------------------------------------------------------------------- /DATASET.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/DATASET.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_ZOO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/MODEL_ZOO.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/README.md -------------------------------------------------------------------------------- /baseline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/baseline/__init__.py -------------------------------------------------------------------------------- /baseline/aro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/baseline/aro.py -------------------------------------------------------------------------------- /baseline/chinese_whispers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/baseline/chinese_whispers.py -------------------------------------------------------------------------------- /baseline/sklearn_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/baseline/sklearn_cluster.py -------------------------------------------------------------------------------- /dsgcn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/README.md -------------------------------------------------------------------------------- /dsgcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/__init__.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_det_fashion_20_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_det_fashion_20_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_det_fashion_2_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_det_fashion_2_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_det_fashion_8_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_det_fashion_8_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_det_ms1m_20_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_det_ms1m_20_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_det_ms1m_2_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_det_ms1m_2_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_det_ms1m_5_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_det_ms1m_5_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_det_ms1m_8_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_det_ms1m_8_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_det_ytb_4_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_det_ytb_4_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_seg_ms1m_20_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_seg_ms1m_20_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_seg_ms1m_2_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_seg_ms1m_2_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_seg_ms1m_5_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_seg_ms1m_5_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_test_seg_ms1m_8_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_test_seg_ms1m_8_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_train_det_fashion_4_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_train_det_fashion_4_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_train_det_fashion_84_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_train_det_fashion_84_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_train_det_fashion_8_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_train_det_fashion_8_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_train_det_ms1m_4_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_train_det_ms1m_4_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_train_det_ms1m_84_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_train_det_ms1m_84_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_train_det_ms1m_8_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_train_det_ms1m_8_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_train_seg_ms1m_4_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_train_seg_ms1m_4_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_train_seg_ms1m_84_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_train_seg_ms1m_84_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/cfg_train_seg_ms1m_8_prpsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/cfg_train_seg_ms1m_8_prpsls.py -------------------------------------------------------------------------------- /dsgcn/configs/yaml/cfg_test_0.7_0.75.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/yaml/cfg_test_0.7_0.75.yaml -------------------------------------------------------------------------------- /dsgcn/configs/yaml/cfg_test_hnsw_2_i0_18_i1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/yaml/cfg_test_hnsw_2_i0_18_i1.yaml -------------------------------------------------------------------------------- /dsgcn/configs/yaml/cfg_test_hnsw_2_i0_6_i1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/yaml/cfg_test_hnsw_2_i0_6_i1.yaml -------------------------------------------------------------------------------- /dsgcn/configs/yaml/cfg_train_0.7_0.75.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/yaml/cfg_train_0.7_0.75.yaml -------------------------------------------------------------------------------- /dsgcn/configs/yaml/cfg_train_8_prpsl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/configs/yaml/cfg_train_8_prpsl.yaml -------------------------------------------------------------------------------- /dsgcn/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/datasets/__init__.py -------------------------------------------------------------------------------- /dsgcn/datasets/build_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/datasets/build_dataloader.py -------------------------------------------------------------------------------- /dsgcn/datasets/cluster_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/datasets/cluster_dataset.py -------------------------------------------------------------------------------- /dsgcn/datasets/cluster_det_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/datasets/cluster_det_processor.py -------------------------------------------------------------------------------- /dsgcn/datasets/cluster_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/datasets/cluster_processor.py -------------------------------------------------------------------------------- /dsgcn/datasets/cluster_seg_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/datasets/cluster_seg_processor.py -------------------------------------------------------------------------------- /dsgcn/datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/datasets/sampler.py -------------------------------------------------------------------------------- /dsgcn/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/main.py -------------------------------------------------------------------------------- /dsgcn/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/models/__init__.py -------------------------------------------------------------------------------- /dsgcn/models/dsgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/models/dsgcn.py -------------------------------------------------------------------------------- /dsgcn/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/runner/__init__.py -------------------------------------------------------------------------------- /dsgcn/runner/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/runner/runner.py -------------------------------------------------------------------------------- /dsgcn/test_cluster_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/test_cluster_det.py -------------------------------------------------------------------------------- /dsgcn/test_cluster_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/test_cluster_seg.py -------------------------------------------------------------------------------- /dsgcn/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/train.py -------------------------------------------------------------------------------- /dsgcn/train_cluster_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/train_cluster_det.py -------------------------------------------------------------------------------- /dsgcn/train_cluster_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/dsgcn/train_cluster_seg.py -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/evaluation/__init__.py -------------------------------------------------------------------------------- /evaluation/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/evaluation/evaluate.py -------------------------------------------------------------------------------- /evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/evaluation/metrics.py -------------------------------------------------------------------------------- /lgcn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/README.md -------------------------------------------------------------------------------- /lgcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/__init__.py -------------------------------------------------------------------------------- /lgcn/configs/cfg_test_lgcn_fashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/configs/cfg_test_lgcn_fashion.py -------------------------------------------------------------------------------- /lgcn/configs/cfg_test_lgcn_ms1m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/configs/cfg_test_lgcn_ms1m.py -------------------------------------------------------------------------------- /lgcn/configs/cfg_train_lgcn_fashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/configs/cfg_train_lgcn_fashion.py -------------------------------------------------------------------------------- /lgcn/configs/cfg_train_lgcn_ms1m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/configs/cfg_train_lgcn_ms1m.py -------------------------------------------------------------------------------- /lgcn/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/datasets/__init__.py -------------------------------------------------------------------------------- /lgcn/datasets/build_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/datasets/build_dataloader.py -------------------------------------------------------------------------------- /lgcn/datasets/cluster_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/datasets/cluster_dataset.py -------------------------------------------------------------------------------- /lgcn/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/main.py -------------------------------------------------------------------------------- /lgcn/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/models/__init__.py -------------------------------------------------------------------------------- /lgcn/models/lgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/models/lgcn.py -------------------------------------------------------------------------------- /lgcn/online_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/online_evaluation.py -------------------------------------------------------------------------------- /lgcn/test_lgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/test_lgcn.py -------------------------------------------------------------------------------- /lgcn/train_lgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/lgcn/train_lgcn.py -------------------------------------------------------------------------------- /post_process/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/post_process/__init__.py -------------------------------------------------------------------------------- /post_process/deoverlap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/post_process/deoverlap.py -------------------------------------------------------------------------------- /post_process/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/post_process/nms.py -------------------------------------------------------------------------------- /proposals/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/proposals/__init__.py -------------------------------------------------------------------------------- /proposals/generate_basic_proposals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/proposals/generate_basic_proposals.py -------------------------------------------------------------------------------- /proposals/generate_iter_proposals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/proposals/generate_iter_proposals.py -------------------------------------------------------------------------------- /proposals/generate_proposals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/proposals/generate_proposals.py -------------------------------------------------------------------------------- /proposals/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/proposals/graph.py -------------------------------------------------------------------------------- /proposals/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/proposals/metrics.py -------------------------------------------------------------------------------- /proposals/stat_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/proposals/stat_cluster.py -------------------------------------------------------------------------------- /proposals/super_vertex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/proposals/super_vertex.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/baseline/aro.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/aro.sh -------------------------------------------------------------------------------- /scripts/baseline/aro_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/aro_fashion.sh -------------------------------------------------------------------------------- /scripts/baseline/aro_ytb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/aro_ytb.sh -------------------------------------------------------------------------------- /scripts/baseline/cw_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/cw_fashion.sh -------------------------------------------------------------------------------- /scripts/baseline/cw_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/cw_ms1m.sh -------------------------------------------------------------------------------- /scripts/baseline/cw_ytb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/cw_ytb.sh -------------------------------------------------------------------------------- /scripts/baseline/dbscan.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/dbscan.sh -------------------------------------------------------------------------------- /scripts/baseline/dbscan_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/dbscan_fashion.sh -------------------------------------------------------------------------------- /scripts/baseline/dbscan_ytb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/dbscan_ytb.sh -------------------------------------------------------------------------------- /scripts/baseline/fast_hac.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/fast_hac.sh -------------------------------------------------------------------------------- /scripts/baseline/fast_hac_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/fast_hac_fashion.sh -------------------------------------------------------------------------------- /scripts/baseline/fast_hac_ytb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/fast_hac_ytb.sh -------------------------------------------------------------------------------- /scripts/baseline/kmeans_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/kmeans_fashion.sh -------------------------------------------------------------------------------- /scripts/baseline/kmeans_ytb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/kmeans_ytb.sh -------------------------------------------------------------------------------- /scripts/baseline/meanshift_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/meanshift_fashion.sh -------------------------------------------------------------------------------- /scripts/baseline/mini_kmeans.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/mini_kmeans.sh -------------------------------------------------------------------------------- /scripts/baseline/spectral.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/spectral.sh -------------------------------------------------------------------------------- /scripts/baseline/spectral_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/baseline/spectral_fashion.sh -------------------------------------------------------------------------------- /scripts/dsgcn/step_by_step/deoverlap.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/step_by_step/deoverlap.sh -------------------------------------------------------------------------------- /scripts/dsgcn/step_by_step/evaluate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/step_by_step/evaluate.sh -------------------------------------------------------------------------------- /scripts/dsgcn/step_by_step/gcn_d_upper_bound.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/step_by_step/gcn_d_upper_bound.sh -------------------------------------------------------------------------------- /scripts/dsgcn/step_by_step/generate_basic_proposals.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/step_by_step/generate_basic_proposals.sh -------------------------------------------------------------------------------- /scripts/dsgcn/step_by_step/generate_iter_proposals.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/step_by_step/generate_iter_proposals.sh -------------------------------------------------------------------------------- /scripts/dsgcn/step_by_step/pipeline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/step_by_step/pipeline.sh -------------------------------------------------------------------------------- /scripts/dsgcn/test_cluster_det_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/test_cluster_det_fashion.sh -------------------------------------------------------------------------------- /scripts/dsgcn/test_cluster_det_iop_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/test_cluster_det_iop_ms1m.sh -------------------------------------------------------------------------------- /scripts/dsgcn/test_cluster_det_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/test_cluster_det_ms1m.sh -------------------------------------------------------------------------------- /scripts/dsgcn/test_cluster_det_ytb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/test_cluster_det_ytb.sh -------------------------------------------------------------------------------- /scripts/dsgcn/test_cluster_seg_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/test_cluster_seg_ms1m.sh -------------------------------------------------------------------------------- /scripts/dsgcn/train_cluster_det_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/train_cluster_det_fashion.sh -------------------------------------------------------------------------------- /scripts/dsgcn/train_cluster_det_iop_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/train_cluster_det_iop_ms1m.sh -------------------------------------------------------------------------------- /scripts/dsgcn/train_cluster_det_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/train_cluster_det_ms1m.sh -------------------------------------------------------------------------------- /scripts/dsgcn/train_cluster_seg_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/dsgcn/train_cluster_seg_ms1m.sh -------------------------------------------------------------------------------- /scripts/lgcn/test_lgcn_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/lgcn/test_lgcn_fashion.sh -------------------------------------------------------------------------------- /scripts/lgcn/test_lgcn_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/lgcn/test_lgcn_ms1m.sh -------------------------------------------------------------------------------- /scripts/lgcn/train_lgcn_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/lgcn/train_lgcn_fashion.sh -------------------------------------------------------------------------------- /scripts/lgcn/train_lgcn_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/lgcn/train_lgcn_ms1m.sh -------------------------------------------------------------------------------- /scripts/tools/test_knn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/tools/test_knn.sh -------------------------------------------------------------------------------- /scripts/vegcn/test_gcn_e_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/vegcn/test_gcn_e_fashion.sh -------------------------------------------------------------------------------- /scripts/vegcn/test_gcn_e_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/vegcn/test_gcn_e_ms1m.sh -------------------------------------------------------------------------------- /scripts/vegcn/test_gcn_v_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/vegcn/test_gcn_v_fashion.sh -------------------------------------------------------------------------------- /scripts/vegcn/test_gcn_v_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/vegcn/test_gcn_v_ms1m.sh -------------------------------------------------------------------------------- /scripts/vegcn/train_gcn_e_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/vegcn/train_gcn_e_fashion.sh -------------------------------------------------------------------------------- /scripts/vegcn/train_gcn_e_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/vegcn/train_gcn_e_ms1m.sh -------------------------------------------------------------------------------- /scripts/vegcn/train_gcn_v_fashion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/vegcn/train_gcn_v_fashion.sh -------------------------------------------------------------------------------- /scripts/vegcn/train_gcn_v_ms1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/scripts/vegcn/train_gcn_v_ms1m.sh -------------------------------------------------------------------------------- /tools/analyze_proposals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/tools/analyze_proposals.py -------------------------------------------------------------------------------- /tools/baseline_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/tools/baseline_cluster.py -------------------------------------------------------------------------------- /tools/download_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/tools/download_data.py -------------------------------------------------------------------------------- /tools/dsgcn_upper_bound.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/tools/dsgcn_upper_bound.py -------------------------------------------------------------------------------- /tools/test_knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/tools/test_knn.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/adjacency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/adjacency.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/dist.py -------------------------------------------------------------------------------- /utils/draw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/draw.py -------------------------------------------------------------------------------- /utils/faiss_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/faiss_gpu.py -------------------------------------------------------------------------------- /utils/faiss_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/faiss_search.py -------------------------------------------------------------------------------- /utils/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/knn.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/misc_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/utils/misc_cluster.py -------------------------------------------------------------------------------- /vegcn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/README.md -------------------------------------------------------------------------------- /vegcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/__init__.py -------------------------------------------------------------------------------- /vegcn/confidence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/confidence.py -------------------------------------------------------------------------------- /vegcn/configs/cfg_test_gcne_fashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/configs/cfg_test_gcne_fashion.py -------------------------------------------------------------------------------- /vegcn/configs/cfg_test_gcne_ms1m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/configs/cfg_test_gcne_ms1m.py -------------------------------------------------------------------------------- /vegcn/configs/cfg_test_gcnv_fashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/configs/cfg_test_gcnv_fashion.py -------------------------------------------------------------------------------- /vegcn/configs/cfg_test_gcnv_ms1m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/configs/cfg_test_gcnv_ms1m.py -------------------------------------------------------------------------------- /vegcn/configs/cfg_train_gcne_fashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/configs/cfg_train_gcne_fashion.py -------------------------------------------------------------------------------- /vegcn/configs/cfg_train_gcne_ms1m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/configs/cfg_train_gcne_ms1m.py -------------------------------------------------------------------------------- /vegcn/configs/cfg_train_gcnv_fashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/configs/cfg_train_gcnv_fashion.py -------------------------------------------------------------------------------- /vegcn/configs/cfg_train_gcnv_ms1m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/configs/cfg_train_gcnv_ms1m.py -------------------------------------------------------------------------------- /vegcn/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/datasets/__init__.py -------------------------------------------------------------------------------- /vegcn/datasets/gcn_e_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/datasets/gcn_e_dataset.py -------------------------------------------------------------------------------- /vegcn/datasets/gcn_v_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/datasets/gcn_v_dataset.py -------------------------------------------------------------------------------- /vegcn/deduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/deduce.py -------------------------------------------------------------------------------- /vegcn/extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/extract.py -------------------------------------------------------------------------------- /vegcn/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/main.py -------------------------------------------------------------------------------- /vegcn/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/models/__init__.py -------------------------------------------------------------------------------- /vegcn/models/gcn_e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/models/gcn_e.py -------------------------------------------------------------------------------- /vegcn/models/gcn_v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/models/gcn_v.py -------------------------------------------------------------------------------- /vegcn/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/models/utils.py -------------------------------------------------------------------------------- /vegcn/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/runner/__init__.py -------------------------------------------------------------------------------- /vegcn/runner/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/runner/runner.py -------------------------------------------------------------------------------- /vegcn/test_gcn_e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/test_gcn_e.py -------------------------------------------------------------------------------- /vegcn/test_gcn_v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/test_gcn_v.py -------------------------------------------------------------------------------- /vegcn/train_gcn_e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/train_gcn_e.py -------------------------------------------------------------------------------- /vegcn/train_gcn_v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yl-1993/learn-to-cluster/HEAD/vegcn/train_gcn_v.py --------------------------------------------------------------------------------