Skip to content

Cory-M/DCCM

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

DCCM

This repository is a PyTorch implementation for Deep Comprehensive Correlation Mining for Image Clustering (accepted to ICCV 2019) at https://arxiv.org/abs/1904.06925?context=cs.CV

by Jianlong Wu*, Keyu Long*, Fei Wang, Chen Qian, Cheng Li, Zhouchen Lin and Hongbin Zha.

citation

If you find DCCM useful in your research, please consider citing:

@inproceedings{DCCM,
    author={Wu, Jianlong and Long, Keyu and Wang, Fei and Qian, Chen and Li, Cheng and Lin, Zhouchen and Zha, Hongbin},
    title={Deep Comprehensive Correlation Mining for Image Clustering},
    booktitle={International Conference on Computer Vision},   
    year={2019},   
}

Table of contents

Introduction

DCCM Figure 1. The pipeline of the proposed DCCM.

Usage

To train with CIFAR10/100 datasets, try:

$ python main.py --config cfgs/cifar10.yaml
$ python main.py --config cfgs/cifar100.yaml

To resume with a certain checkpoint , try:

$ python main.py --config cfgs/xx.yaml --resume xxx.ckpt

Parameters and datapaths can be modified in the config files.

Note that we use meta-files (examples could be found in the folder 'meta') to load data.

Requirments

  • a Python installation version 3.6.5
  • a Pytorch installation version 0.4.1
  • a Keras installation version 2.0.2
  • download the image dataset and stored according to the meta-files

Please note that all reported performance are tested under this environment.

Comparisons with SOTAs

Table 1. Clustering performance of different methods on six challenging datasets. Results

Reference Github Repos

Our group at SenseTime Research is looking for algorithm researchers and engineers. Our research interests include object detection, tracking, classification, and segmentation, auto network search, network compression and quantization on mobile terminals, 3d gaze tracking, computer vision related SDK, and product platform development. Our group aims to pioneer the computer vision based IOT industry. We have a lot of NOI & ACM gold medal winners, and thousands of GPU Cards. Our team has win the world champions of MegaFace in face recognition and VOT challenge in object tracking, and has published many research papers in top conferences, such as CVPR, ICCV, ECCV, and NeurIPS. Please feel free to contact us with Wechat#: 18810636695 or Email: wangfei@sensetime.com if you are interested in our group.

About

Deep Comprehensive Correlation Mining

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages