ptyrad.visualization

Plotting functions for output figures including forward passes, loss curves, positions, etc.

Functions

plot_affine_transformation(scale, asymmetry, ...)
plot_forward_pass(model, indices, dp_power)	Plot the forward pass for the input torch model
plot_loss_curves(loss_iters[, last_n_iters, ...])
plot_obj_tilts(pos, tilts[, figsize, ...])	Plot the obj tilts given the probe position and pos-dependent tilts
plot_obj_tilts_avg(avg_tilt_iters[, ...])
plot_pos_grouping(pos, batches[, ...])
plot_probe_modes([init_probe, opt_probe, ...])
plot_scan_positions(pos[, init_pos, img, ...])	Plot the scan positions given an array of (N,2)
plot_sigmoid_mask(Npix, relative_radius, ...)	Plot a sigmoid mask overlay on img with a line profile
plot_slice_thickness(dz_iters[, ...])
plot_summary(output_path, model, niter, ...)	Wrapper function for most visualization function