SourceXtractor::MLSegmentation Class Reference

Implements a Segmentation based on CNN. More...

#include <MLSegmentation.h>

Inheritance diagram for SourceXtractor::MLSegmentation:

Collaboration diagram for SourceXtractor::MLSegmentation:

Classes
struct	Tile

Public Member Functions
virtual	~MLSegmentation ()=default
	MLSegmentation (std::shared_ptr< SourceFactory > source_factory, std::string model_path, double ml_threshold)
void	labelImage (Segmentation::LabellingListener &listener, std::shared_ptr< const DetectionImageFrame > frame) override
Public Member Functions inherited from SourceXtractor::Segmentation::Labelling
virtual	~Labelling ()=default
	Labelling ()

Private Member Functions
std::vector< MLSegmentation::Tile >	getTiles (const DetectionImage &image) const

Private Attributes
std::shared_ptr< SourceFactory >	m_source_factory
std::string	m_model_path
double	m_ml_threshold

Detailed Description

Implements a Segmentation based on CNN.

Definition at line 32 of file MLSegmentation.h.

Constructor & Destructor Documentation

~MLSegmentation()

virtual SourceXtractor::MLSegmentation::~MLSegmentation ( )

virtualdefault

MLSegmentation()

SourceXtractor::MLSegmentation::MLSegmentation ( std::shared_ptr< SourceFactory > source_factory, std::string model_path, double ml_threshold )

inline

Definition at line 37 of file MLSegmentation.h.

References m_ml_threshold, m_model_path, and m_source_factory.

Member Function Documentation

getTiles()

std::vector< MLSegmentation::Tile > SourceXtractor::MLSegmentation::getTiles ( const DetectionImage & image ) const

private

labelImage()

void SourceXtractor::MLSegmentation::labelImage ( Segmentation::LabellingListener & listener, std::shared_ptr< const DetectionImageFrame > frame )

overridevirtual

Implements SourceXtractor::Segmentation::Labelling.

Definition at line 93 of file MLSegmentation.cpp.

References std::vector< T >::emplace_back(), SourceXtractor::OnnxModel::getInputNb(), SourceXtractor::OnnxModel::getInputShape(), SourceXtractor::OnnxModel::getInputType(), SourceXtractor::CheckImages::getInstance(), Elements::Logging::getLogger(), SourceXtractor::OnnxModel::getOutputShape(), SourceXtractor::OnnxModel::getOutputType(), SourceXtractor::ImageAccessor< T >::getValue(), SourceXtractor::Lutz::labelImage(), m_ml_threshold, m_model_path, m_source_factory, SourceXtractor::FitsWriter::newTemporaryImage(), SourceXtractor::onnx_logger, and SourceXtractor::OnnxModel::run().

Here is the call graph for this function:

Member Data Documentation

m_ml_threshold

double SourceXtractor::MLSegmentation::m_ml_threshold

private

Definition at line 56 of file MLSegmentation.h.

Referenced by labelImage(), and MLSegmentation().

m_model_path

std::string SourceXtractor::MLSegmentation::m_model_path

private

Definition at line 55 of file MLSegmentation.h.

Referenced by labelImage(), and MLSegmentation().

m_source_factory

std::shared_ptr<SourceFactory> SourceXtractor::MLSegmentation::m_source_factory

private

Definition at line 53 of file MLSegmentation.h.

Referenced by labelImage(), and MLSegmentation().

---

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

• SEImplementation/SEImplementation/Segmentation/MLSegmentation.h
• SEImplementation/src/lib/Segmentation/MLSegmentation.cpp