/home/rex/ebltrunk/tools/libeblearntools/include/bootstrapping.h

/***************************************************************************
 *   Copyright (C) 2012 by Pierre Sermanet *
 *   pierre.sermanet@gmail.com *
 *   All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *     * Redistribution under a license not approved by the Open Source
 *       Initiative (http://www.opensource.org) must display the
 *       following acknowledgement in all advertising material:
 *        This product includes software developed at the Courant
 *        Institute of Mathematical Sciences (http://cims.nyu.edu).
 *     * The names of the authors may not be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL ThE AUTHORS BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 ***************************************************************************/

#ifndef BOOTSTRAPPING_H_
#define BOOTSTRAPPING_H_

#include <stdio.h>
#include <stdlib.h>

#include "defines_tools.h"
#include "detector.h"
#include "configuration.h"
#include "pascal_xml.h"

using namespace std;

namespace ebl {

  template <typename T, class Tstate = fstate_idx<T> > class bootstrapping {
  public:
    bootstrapping(configuration &conf);
    virtual ~bootstrapping();

    void fprop(detector<T,Tstate> &detect, string &fname, bool reset = true,
               double scale = 1.0);
    void clear();

    // accessors ///////////////////////////////////////////////////////////////

    bboxes& get_bbpos();
    bboxes& get_bbneg();
    bboxes& get_bball();
    bboxes& get_gtall();
    bboxes& get_gtclean();
    bboxes& get_gtrest();
    svector<midx<T> >& get_pos();
    svector<midx<T> >& get_neg();
    svector<midx<T> >& get_all();

    bool activated();
    bool extract_positives();
    bool groundtruth_found(string &frame_name);
    bool skip_frame(string &frame_name);
    intg max_size();
  
    // dataset methods /////////////////////////////////////////////////////////

    void save_dataset(svector<midx<T> > &samples, bboxes &bb, string &outdir,
                      idx<ubyte> &classes);

    // internal methods ////////////////////////////////////////////////////////
  protected:
  
    string groundtruth_file(string &frame_name);
    bboxes load_groundtruth(string &frame_name);
    bboxes load_clean_groundtruth(string &frame_name, configuration &conf,
                                  bboxes &rest);
  
    bboxes get_positives(mstate<Tstate> &outputs, bboxes &groundtruth,
                         mfidxdim &topleft, mfidxdim &topright,
                         mfidxdim &bottomleft, mfidxdim &bottomright,
                         float matching, float min_context);
    bboxes get_negatives(mstate<Tstate> &answers, bboxes &filtered,
                         bboxes &nonfiltered,
                         mfidxdim &topleft, mfidxdim &topright,
                         mfidxdim &bottomleft, mfidxdim &bottomright,
                         float matching, uint nmax, int neg_id, T threshold);
    void add_mirrors(svector<midx<T> > &samples, bboxes &boxes);

    // private members /////////////////////////////////////////////////////////
  protected:
    configuration &conf;
    bool silent;
    string gt_path; 
    bboxes gt_all; 
    bboxes gt_clean; 
    bboxes gt_rest; 
    bboxes bbpos; 
    bboxes bbneg; 
    bboxes bball; 
    svector<midx<T> > spos; 
    svector<midx<T> > sneg; 
    svector<midx<T> > sall; 
    string xml_fullname; 
    bool extract_pos; 
    bool extract_neg; 
    bool _activated;
    intg _max_size;
    bool neg_gt_only; 
    T neg_threshold; 
    mfidxdim bbox_scalings; 
    bool mirror_pos; 
  };

} // end namespace ebl

#include "bootstrapping.hpp"

#endif /* BOOTSTRAPPING_H_ */