process_prediction

radionets.evaluation.utils.process_prediction(conf, img_test, img_true, norm_dict, model, model_2)

Applies the normalization, gets and rescales a prediction and performs the inverse Fourier transformation.

Parameters:

confdictionary

config files containing the settings

img_testtorch.Tensor

input file for the network

img_truetorch.tensor

true image

norm_dictdictionary

dict containing the normalization factors

modelarchitecture

model initialized with save file

model_2

model initialized with save file

Returns:

ifft_predndarray

predicted source in image space

ifft_truthndarray

true source in image space