DeepDPM: Deep Clustering With An Unknown Number of Clusters

This repo contains the implementation of our paper:
DeepDPM clustering example on 2D data.
On the left: DeepDPM’s predicted clusters’ assignments, centers and covariances. On the right: Clusters colored by the GT labels, and the net’s decision boundary.
Examples of the clusters found by DeepDPM on the ImageNet Dataset: