ImageCompress.h

Go to the documentation of this file.

#ifndef IMAGE_COMPRESS_H
#define IMAGE_COMPRESS_H

#include "annie/defines.h"
#include "annie/MultiLayerNetwork.h"
#include <vector>

namespace annie {
class TrainingSet;
class Image;
/*abstract */class ImageCompressor  {
  public:
    virtual ~ImageCompressor() {}   //be happy, my dear compiler :)
    class PartialListener   {
      public:
        virtual void showProgress(const Image &partialResult) = 0;
    };
    virtual Image *compress(const Image &i) = 0;

    void setPartialListener(PartialListener *pl)    {   _partialListener = pl; }
  protected:
    void showPartialProgress(const Image &partialResult)    { if( _partialListener)  _partialListener->showProgress(partialResult); }
    ImageCompressor() :  _partialListener(NULL) {}
  private:
    PartialListener *_partialListener;
};

typedef std::vector<annie::Image *> Images;

/*abstract */class TileCompressor : public ImageCompressor  {
  public:
    TileCompressor(uint partSize) : _partSize(partSize) {}
    uint getPartSize()  const { return _partSize; }
    virtual Image *compress(const Image &i);
    virtual Image *compressTiles(const Images &images) = 0;
 
    Images cut(const Image &source) const;
  protected:
    uint _partSize; //<some implementations may allow to change it during lifetime
    uint _px, _py;  //< valid only during compress - count of cols and rows
};

class MLPCompressor : public TileCompressor, public ValueUpdateListener     
{
  public:
    enum { RESHOW = 20 };   //redraw period
    MLPCompressor(uint partSize, uint hidden1, uint hidden2, PublicValues &pv = defaultControl);
    virtual Image *compressTiles(const Images &images);
    virtual void valueChanged(const Value &val);
    
    MultiLayerNetwork &getNet() { return _net; }

    void fillExamples(const Images &images, TrainingSet &dest); //TS is not a result because pass by value of TS is slowwwww
  protected:
    MultiLayerNetwork _net;
    std::vector<Vector> vectored;   //< vectored tiles

    //construct result using vectored and the net
    Image *getResult();
    Creal _reshow;  //re-show period
};



    
        
    // utility functions for creating images
    Images makeSubimages(const annie::Image &i, unsigned xSize, unsigned ySize);

    TrainingSet *makeSubImageExamples(const Images &outImages, std::vector<Vector> &outExamples);

    void runCompressor(ImageCompressor &ic, const char *image, bool graay=false);

}   //annie
#endif

---