mldas.explore.mapping.extract_prob_map

mldas.explore.mapping. extract_prob_map ( data , model , img_size = 10 , channel_stride = 1 , sample_stride = 10 , verbose = False , stage_plot = False , single = False , rgb = False , colormap = None )

Calculate probability map using fed raw data and training model.

Parameters

data numpy.ndarray

Input raw measurements extracted from HDF5 file

model torch.nn.Module

Trained model

img_size int

Size of squared image

channel_stride int

Sliding interval in channel space

sample_stride int

Sliding interval in sample space

verbose bool

Do verbose

stage_plot bool

Plot intermediate stages

Returns

prob_array numpy.ndarray

Normalized probability map