PredictionImageGradient

class radionets.core.PredictionImageGradient(validation_data, model, amp_phase, arch_name)

Bases: Callback

Callback for computing spatial gradients of model predictions.

Parameters:

validation_datastr or Path

Path to validation dataset.

modelstr or Path

Path to pretrained model.

amp_phasebool

Whether to use amplitude-phase representation.

arch_namestr

Name of the architecture used for the model.

Methods Summary

save_output_pred()

Methods Documentation

save_output_pred()