summarize_images.hpp File Reference

#include "lbann/callbacks/callback.hpp"
#include <google/protobuf/message.h>
#include <lbann/base.hpp>
#include <string>
#include <vector>

Include dependency graph for summarize_images.hpp:

Go to the source code of this file.

Classes
class	lbann::callback::image_output_strategy
	Interface for strategies for determining which images to output to the summarizer. More...
class	lbann::callback::categorical_accuracy_strategy
class	lbann::callback::autoencoder_strategy
	Subclass of image_output_strategy to dump autoencoder images. More...
class	lbann::callback::summarize_images
	Callback to dump images to event files based on strategy. More...

Namespaces
	lbann_data
	lbann
	lbann::callback

Functions
std::unique_ptr< image_output_strategy >	lbann::callback::build_categorical_accuracy_strategy_from_pbuf (google::protobuf::Message const &)
std::unique_ptr< image_output_strategy >	lbann::callback::build_track_sample_ids_strategy_from_pbuf (google::protobuf::Message const &)
Layer const &	lbann::callback::get_layer_by_name (model const &m, std::string const &layer_name)
	Get a layer from model based on name. More...
std::unique_ptr< callback_base >	lbann::callback::build_summarize_images_callback_from_pbuf (const google::protobuf::Message &, const std::shared_ptr< lbann_summary > &summarizer)