ModelFitting::OnnxCompactModel< ImageType > Class Template Reference

#include <OnnxCompactModel.h>

Inheritance diagram for ModelFitting::OnnxCompactModel< ImageType >:

Collaboration diagram for ModelFitting::OnnxCompactModel< ImageType >:

Public Member Functions
	OnnxCompactModel (std::vector< std::shared_ptr< SourceXtractor::OnnxModel > > models, std::shared_ptr< BasicParameter > x_scale, std::shared_ptr< BasicParameter > y_scale, std::shared_ptr< BasicParameter > rotation, double width, double height, std::shared_ptr< BasicParameter > x, std::shared_ptr< BasicParameter > y, std::shared_ptr< BasicParameter > flux, std::map< std::string, std::shared_ptr< BasicParameter > > params, std::tuple< double, double, double, double > transform)
virtual	~OnnxCompactModel ()=default
double	getValue (double, double) const override
ImageType	getRasterizedImage (double pixel_scale, std::size_t size_x, std::size_t size_y) const override
Public Member Functions inherited from ModelFitting::CompactModelBase< ImageType >
	CompactModelBase (std::shared_ptr< BasicParameter > x_scale, std::shared_ptr< BasicParameter > y_scale, std::shared_ptr< BasicParameter > rotation, double width, double height, std::shared_ptr< BasicParameter > x, std::shared_ptr< BasicParameter > y, std::tuple< double, double, double, double > transform)
virtual	~CompactModelBase ()=default
Public Member Functions inherited from ModelFitting::ExtendedModel< ImageType >
	ExtendedModel (std::vector< std::unique_ptr< ModelComponent > > &&component_list, std::shared_ptr< BasicParameter > x_scale, std::shared_ptr< BasicParameter > y_scale, std::shared_ptr< BasicParameter > rotation_angle, double width, double height, std::shared_ptr< BasicParameter > x, std::shared_ptr< BasicParameter > y)
	ExtendedModel (ExtendedModel &&)=default
virtual	~ExtendedModel ()=default
double	getWidth () const
double	getHeight () const
Public Member Functions inherited from ModelFitting::PositionedModel
	PositionedModel (std::shared_ptr< BasicParameter > x, std::shared_ptr< BasicParameter > y)
	PositionedModel (PositionedModel &&other)
virtual	~PositionedModel ()=default
double	getX () const
double	getY () const

Private Attributes
std::vector< std::shared_ptr< SourceXtractor::OnnxModel > >	m_models
std::shared_ptr< BasicParameter >	m_flux
std::map< std::string, std::shared_ptr< BasicParameter > >	m_params

Additional Inherited Members
Protected Member Functions inherited from ModelFitting::CompactModelBase< ImageType >
Mat22	getCombinedTransform (double pixel_scale) const
template<typename ModelEvaluator >
float	samplePixel (const ModelEvaluator &model_eval, int x, int y, unsigned int subsampling) const
template<typename ModelEvaluator >
float	sampleStochastic (const ModelEvaluator &model_eval, int x, int y, unsigned int samples=100) const
template<typename ModelEvaluator >
float	adaptiveSamplePixel (const ModelEvaluator &model_eval, int x, int y, unsigned int max_subsampling, float threshold=1.1) const
double	getMaxRadiusSqr (std::size_t size_x, std::size_t size_y, const Mat22 &transform) const
void	renormalize (ImageType &image, double flux) const
Protected Attributes inherited from ModelFitting::CompactModelBase< ImageType >
Mat22	m_jacobian
Mat22	m_inv_jacobian
Protected Attributes inherited from ModelFitting::ExtendedModel< ImageType >
std::vector< std::unique_ptr< ModelComponent > >	m_component_list {}

Detailed Description

template<typename ImageType>
class ModelFitting::OnnxCompactModel< ImageType >

Definition at line 25 of file OnnxCompactModel.h.

Constructor & Destructor Documentation

OnnxCompactModel()

template<typename ImageType >

ModelFitting::OnnxCompactModel< ImageType >::OnnxCompactModel ( std::vector< std::shared_ptr< SourceXtractor::OnnxModel > > models, std::shared_ptr< BasicParameter > x_scale, std::shared_ptr< BasicParameter > y_scale, std::shared_ptr< BasicParameter > rotation, double width, double height, std::shared_ptr< BasicParameter > x, std::shared_ptr< BasicParameter > y, std::shared_ptr< BasicParameter > flux, std::map< std::string, std::shared_ptr< BasicParameter > > params, std::tuple< double, double, double, double > transform )

inline

Definition at line 27 of file OnnxCompactModel.h.

~OnnxCompactModel()

template<typename ImageType >

virtual ModelFitting::OnnxCompactModel< ImageType >::~OnnxCompactModel ( )

virtualdefault

Member Function Documentation

getRasterizedImage()

template<typename ImageType >

ImageType ModelFitting::OnnxCompactModel< ImageType >::getRasterizedImage ( double pixel_scale, std::size_t size_x, std::size_t size_y ) const

inlineoverridevirtual

Reimplemented from ModelFitting::ExtendedModel< ImageType >.

Definition at line 45 of file OnnxCompactModel.h.

References dx, dy, ModelFitting::CompactModelBase< ImageType >::getCombinedTransform(), ModelFitting::logger, ModelFitting::OnnxCompactModel< ImageType >::m_flux, ModelFitting::OnnxCompactModel< ImageType >::m_models, ModelFitting::OnnxCompactModel< ImageType >::m_params, std::max(), pixel_scale, ModelFitting::CompactModelBase< ImageType >::renormalize(), std::transform(), x, and y.

Here is the call graph for this function:

getValue()

template<typename ImageType >

double ModelFitting::OnnxCompactModel< ImageType >::getValue ( double , double  ) const

inlineoverridevirtual

Reimplemented from ModelFitting::ExtendedModel< ImageType >.

Definition at line 41 of file OnnxCompactModel.h.

Member Data Documentation

m_flux

template<typename ImageType >

std::shared_ptr<BasicParameter> ModelFitting::OnnxCompactModel< ImageType >::m_flux

private

Definition at line 122 of file OnnxCompactModel.h.

Referenced by ModelFitting::OnnxCompactModel< ImageType >::getRasterizedImage().

m_models

template<typename ImageType >

std::vector<std::shared_ptr<SourceXtractor::OnnxModel> > ModelFitting::OnnxCompactModel< ImageType >::m_models

private

Definition at line 119 of file OnnxCompactModel.h.

Referenced by ModelFitting::OnnxCompactModel< ImageType >::getRasterizedImage().

m_params

template<typename ImageType >

std::map<std::string, std::shared_ptr<BasicParameter> > ModelFitting::OnnxCompactModel< ImageType >::m_params

private

Definition at line 123 of file OnnxCompactModel.h.

Referenced by ModelFitting::OnnxCompactModel< ImageType >::getRasterizedImage().

---

The documentation for this class was generated from the following file:

• SEImplementation/SEImplementation/Plugin/FlexibleModelFitting/OnnxCompactModel.h