SourceXtractorPlusPlus
0.16
Please provide a description of the project.
|
Implements a Segmentation based on CNN. More...
#include <MLSegmentation.h>
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 |
![]() | |
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 |
Implements a Segmentation based on CNN.
Definition at line 32 of file MLSegmentation.h.
|
virtualdefault |
|
inline |
Definition at line 37 of file MLSegmentation.h.
|
private |
|
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(), Elements::Logging::info(), SourceXtractor::Lutz::labelImage(), m_ml_threshold, m_model_path, m_source_factory, SourceXtractor::onnx_logger, SourceXtractor::OnnxModel::run(), x, and y.
|
private |
Definition at line 56 of file MLSegmentation.h.
Referenced by labelImage().
|
private |
Definition at line 55 of file MLSegmentation.h.
Referenced by labelImage().
|
private |
Definition at line 53 of file MLSegmentation.h.
Referenced by labelImage().