visualization

Utility functions to visualize and animate DRRs

source

plot_drr

 plot_drr (img:torch.Tensor, title:Optional[str]=None,
           ticks:Optional[bool]=True,
           axs:Optional[matplotlib.axes._axes.Axes]=None)

Plot an image generated by a DRR module.


source

animate

 animate (out:Union[str,pathlib.Path], df:pandas.core.frame.DataFrame,
          sdr:float, drr:diffdrr.drr.DRR,
          ground_truth:Optional[torch.Tensor]=None, verbose:bool=True,
          **kwargs)

Animate the optimization of a DRR.

Type Default Details
out str | pathlib.Path Savepath
df pandas.DataFrame
sdr float
drr DRR
ground_truth torch.Tensor | None None
verbose bool True
kwargs

df is a pandas.DataFrame with columns ["theta", "phi", "gamma", "bx", "by", "bz"]. Each row in df is an iteration of optimization with the updated values for that timestep.