Machine learning of hierarchical clustering to segment 2D and 3D images