def count_voxels(masked_in, masked_out):
    """Count voxels in and outside the mask"""
    count_in = len(masked_in[masked_in != 0])
    count_out = len(masked_out[masked_out != 0])
    return count_in, count_out
