Implemented 3D connected components as wrapper class for scipy.ndimage.label

Ticket: https://www.notion.so/qim-dtu/3D-connected-components-ca5c410fef49444eac7c6efd0b56ead4?pvs=4

I have made a simple plot functions that takes the ConnectedComponents object and extracts the connected_components volume.

However I don't know if it is necessary as the user could do this manually.

def plot_connected_components(connected_components: ConnectedComponents, **kwargs):
    """ Plots connected components from the ConnectedComponents class as 2d slices.

    Args:
        connected_components (ConnectedComponents): The connected components class from the qim3d.utils.connected_components module.
        **kwargs: Additional keyword arguments to pass to qim3d.viz.img.slice_viz.

    Returns:
        matplotlib.pyplot: the 3D plot of the connected components.
    """
    fig = slice_viz(connected_components.connected_components, **kwargs)
    return fig

qim3d/utils/connected_components.py

@@ -88,10 +88,5 @@ def get_3d_connected_components(image: np.ndarray | torch.Tensor):
     Returns:
         class: Returns class object of the connected components.
     """
-    if image.ndim != 3:
-        raise ValueError(
-            f"Given array is not a volume! Current dimension: {image.ndim}"
-        )
-
     connected_components, num_connected_components = label(image)
     return ConnectedComponents(connected_components, num_connected_components)