create_jet

radiosim.jets.jet.create_jet(grid, conf)

Creates the clean jets with all its components written in a list. Dependend on the ‘train_type’ the components will be seperated or summed up.

Parameters:

grid: ndarray

input grid of shape [n, 1, img_size, img_size] or [1, img_size, img_size]

conf:

loaded config file

Returns:

jets: ndarray

image of the full jet, sum over all components, shape: [n, 1, img_size, img_size]

jet_comps: ndarray

images of each component and background, shape: [n, c*2, img_size, img_size] with c being the max number of components. A jet without counter jet has c components. A jet with counter jet has c*2-1 components, since the center appears only once. Adding one channel for the backgound gives c*2 channels.

source_lists: ndarray

array which stores all (seven) properties of each component, shape: [n, c*2-1, 7]