Inheritance diagram for lbann::transform::grayscale:

Collaboration diagram for lbann::transform::grayscale:

Public Member Functions
transform *	copy () const override

std::string	get_type () const override

void	apply (utils::type_erased_matrix &data, std::vector< size_t > &dims) override

Public Member Functions inherited from lbann::transform::transform
	transform ()=default

	transform (const transform &)=default

transform &	operator= (const transform &)=default

virtual	~transform ()=default

virtual description	get_description () const

virtual bool	supports_non_inplace () const

virtual void	apply (utils::type_erased_matrix &data, CPUMat &out, std::vector< size_t > &dims)

Additional Inherited Members
Static Protected Member Functions inherited from lbann::transform::transform
static float	get_uniform_random (float a, float b)

static bool	get_bool_random (float p)

static El::Int	get_uniform_random_int (El::Int a, El::Int b)

Detailed Description

Convert an image to grayscale.

Definition at line 38 of file grayscale.hpp.

Member Function Documentation

void lbann::transform::grayscale::apply	(	utils::type_erased_matrix &	data,
		std::vector< size_t > &	dims
	)

overridevirtual

Apply the transform to data.

Parameters

data	The input data to transform, which is modified in-place. The matrix shuold be contiguous.
dims	The dimensions of the data tensor. For "plain data", dims should have one entry, giving its size. For images, dims should have three entries: channels, height, width.