Skip to content

Points to Grid Schema

Points to Grid Schema¤

neojax.data.schemas.reshape_points_to_grid_schema.ReshapePointsToGridSchema ¤

Schema that reconstructs the spatial grid from a flattened array of point predictions.

transform(bundle: typing.Union[ForwardRef(DataBundle), Inexact[Array, ...], Inexact[ndarray, ...]], /, reference_bundle: neojax.data.bundles.data_bundle.DataBundle | None = None) -> DataBundle ¤

Reshapes flat point outputs into a DataBundle grid.

Parameters:

Name Type Description Default
bundle Union[ForwardRef(DataBundle), Inexact[Array, ...], Inexact[ndarray, ...]]

The input array of point predictions (or existing DataBundle).

required
reference_bundle DataBundle | None

The original bundle, used for coordinates and other metadata.

None

Returns:

Type Description
DataBundle

The reconstructed DataBundle.