Source code for locpix.img_processing.render
"""Module for rendering point clouds of various forms"""
import numpy as np
import polars as pl
[docs]
def visualise_seg_item(item, column):
"""Function for visualising the segmentation of
a point cloud and saving
Where column is the column the histogram values are
in
Note that as not every pixel will contain localisations
there may be pixels which get set to zero.
Args:
item (datastruc.item) : This contains the
df and the labels to visualise.
This must be true!
column (string) : Column containing the
segmention labels
"""
labels = item.df.select(pl.col(column)).to_numpy()
x_pixels = item.df.select(pl.col("x_pixel")).to_numpy()
y_pixels = item.df.select(pl.col("y_pixel")).to_numpy()
# note this assumes all histos have same shape
if item.histo is not None and len(item.histo) != 0:
assert item.histo[0].shape[0] == np.max(x_pixels) + 1
assert item.histo[0].shape[1] == np.max(y_pixels) + 1
histo_width = np.max(x_pixels) + 1
histo_height = np.max(y_pixels) + 1
histo = np.zeros((histo_width, histo_height))
histo[x_pixels, y_pixels] = labels
return histo