NeuralNetwork Class Reference

#include <NeuralNetwork.h>

Public Types
enum	TrainingState { NOT_TRAINED , TRAINING , TRAINED }
enum	MlType {   KNN , Bayes , SVM , DT ,   ANN , Boost , RT , NONE ,   NUMTYPES }
enum	MlMode { ML_POSE , ML_MOTION }
typedef std::map< unsigned int, std::vector< std::vector< MlFeature > > >	SamplesMap

Public Member Functions
	NeuralNetwork (int featurePerSample, float sensibility, MlMode mode=ML_POSE, MlType type=KNN)
	~NeuralNetwork ()
void	SetType (MlType type)
MlType	GetType ()
void	SetCategories (const std::vector< std::string > &names)
std::vector< std::string >	GetCategories ()
std::string	GetCategory (unsigned int pos)
int	GetCategoryPos (std::string name)
void	Train ()
void	Load (std::string filename)
void	Save (std::string saveFile)
void	AddTrainingData (std::vector< cv::Point3d > input, std::string label)
void	Reconize (cv::Mat reconizeIn, cv::Mat reconizeOut)
void	AddDetectionData (std::vector< cv::Point3d >)
TrainingState	GetTrainingState ()

Detailed Description

Definition at line 120 of file NeuralNetwork.h.

Member Typedef Documentation

SamplesMap

typedef std::map<unsigned int, std::vector<std::vector<MlFeature> > > NeuralNetwork::SamplesMap

Definition at line 149 of file NeuralNetwork.h.

Member Enumeration Documentation

MlMode

enum NeuralNetwork::MlMode

Enumerator
ML_POSE
ML_MOTION

Definition at line 143 of file NeuralNetwork.h.

MlType

enum NeuralNetwork::MlType

Enumerator
KNN
Bayes
SVM
DT
ANN
Boost
RT
NONE
NUMTYPES

Definition at line 130 of file NeuralNetwork.h.

TrainingState

enum NeuralNetwork::TrainingState

Enumerator
NOT_TRAINED
TRAINING
TRAINED

Definition at line 123 of file NeuralNetwork.h.

Constructor & Destructor Documentation

NeuralNetwork()

NeuralNetwork::NeuralNetwork	(	int	featurePerSample,
		float	sensibility,
		MlMode	mode = ML_POSE,
		MlType	type = KNN
	)

Definition at line 38 of file NeuralNetwork.cpp.

~NeuralNetwork()

NeuralNetwork::~NeuralNetwork

(

)

Definition at line 87 of file NeuralNetwork.cpp.

Member Function Documentation

AddDetectionData()

void NeuralNetwork::AddDetectionData

(

std::vector< cv::Point3d >

input

)

Definition at line 871 of file NeuralNetwork.cpp.

AddTrainingData()

void NeuralNetwork::AddTrainingData	(	std::vector< cv::Point3d >	input,
		std::string	label
	)

Definition at line 157 of file NeuralNetwork.cpp.

GetCategories()

std::vector< std::string > NeuralNetwork::GetCategories

(

)

Definition at line 145 of file NeuralNetwork.cpp.

GetCategory()

std::string NeuralNetwork::GetCategory

(

unsigned int

pos

)

Definition at line 135 of file NeuralNetwork.cpp.

GetCategoryPos()

int NeuralNetwork::GetCategoryPos

(

std::string

name

)

Definition at line 124 of file NeuralNetwork.cpp.

GetTrainingState()

NeuralNetwork::TrainingState NeuralNetwork::GetTrainingState

(

)

Definition at line 112 of file NeuralNetwork.cpp.

GetType()

MlType NeuralNetwork::GetType ( )

inline

Definition at line 208 of file NeuralNetwork.h.

Load()

void NeuralNetwork::Load

(

std::string

filename

)

Definition at line 702 of file NeuralNetwork.cpp.

Reconize()

void NeuralNetwork::Reconize	(	cv::Mat	reconizeIn,
		cv::Mat	reconizeOut
	)

Definition at line 768 of file NeuralNetwork.cpp.

Save()

void NeuralNetwork::Save

(

std::string

saveFile

)

Definition at line 660 of file NeuralNetwork.cpp.

SetCategories()

void NeuralNetwork::SetCategories ( const std::vector< std::string > &  names )

inline

Definition at line 216 of file NeuralNetwork.h.

SetType()

void NeuralNetwork::SetType ( MlType  type )

inline

Definition at line 200 of file NeuralNetwork.h.

Train()

void NeuralNetwork::Train

(

)

Definition at line 391 of file NeuralNetwork.cpp.

---

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

• include/NeuralNetwork.h
• src/NeuralNetwork.cpp