armed.callbacks.aec_callbacks

Custom callbacks for autoencoder-classifiers.

Functions

compute_image_metrics(epoch, model, data_in, ...)	Compute image metrics including brightness, contrast, sharpness, and SNR.
make_compute_latents_callback(model, images, ...)	Generate a callback function that calls the encoder on some images to create latent representations, then saves them to a .pkl file.
make_image_metrics_callback(model, data_in, ...)	Generate a callback function that computes image metrics including brightness, contrast, sharpness, and SNR.
make_recon_figure_callback(images, model, ...)	Generate a callback function that produces a figure with example reconstructions.