---

locator.h

Go to the documentation of this file.

#ifndef __LOCATOR_TAG
#define __LOCATOR_TAG

#include "locator_defines.h"
#include "frame.h"
#include "annie/annie.h"

#define VERBOSE false

#define INPUT_NODES 345

#define ITERATIONS 500

#define LEARNING_RATE 0.25

#define MOMENTUM 0.15

#define OK_ERROR 0.015

#define TRUE_OUTPUT 0.95

#define FALSE_OUTPUT 0.05

#define CUT_OFF ((TRUE_OUTPUT + FALSE_OUTPUT)/2.0)

#define MAX_OK_ERROR ((TRUE_OUTPUT - CUT_OFF) - .05)

#define MAX_COUNTER_PER_CYCLE 500

#define MAX_FACES_PER_CYCLE 100

#define FACE_DISTANCE 6

#define FACE_MIN 15

class face_locator_network
{
public:
  face_locator_network(int hidden);

  face_locator_network(string FileName);

  ~face_locator_network();
  bool save(string FileName);

  float run_net(annie::VECTOR & data);

  float run_net(double * data);
  
private:
  annie::TwoLayerNetwork * net;
  bool is_set;
};

annie::TrainingSet * make_face_training_set(string FaceDir, string fileName);


bool training_set_guess(string FaceDir, string NonFaceDir, string fileName, int num_images);

bool add_non_training_set(string NonFaceDir, annie::TrainingSet & T );
void newTrain(annie::TrainingSet & ts,unsigned int hidden);

void trainNN(string NonFaceDir,  annie::TrainingSet & all, annie::TrainingSet & ts, string fileName, int hidden);

class locator_filter: public Filter
{
public:
  locator_filter(string FileName);
  
  locator_filter();

  virtual ~locator_filter();
  
  
  virtual bool filter_frame(FrameClass & frame);
  virtual void property_changed();
  
private:
  string filename;
  face_locator_network * net;
  bool set;
};


extern unsigned int start_stop[INPUT_HEIGHT*2];

#ifdef DYNAMIC_LINK
extern "C"{
  Filter * create_instance();
#ifndef NO_GUI
  GdkPixmap *create_icon(GtkWidget *drawing_area);
#endif
}
#endif

#endif

---