images.h

   1 /*
   2  *  mlp-mnist is an implementation of a multi-layer neural network.
   3  *
   4  *  Copyright (c) 2006 École Polytechnique Fédérale de Lausanne,
   5  *  http://www.epfl.ch
   6  *
   7  *  Written by Francois Fleuret <francois@fleuret.org>
   8  *
   9  *  This file is part of mlp-mnist.
  10  *
  11  *  mlp-mnist is free software: you can redistribute it and/or modify
  12  *  it under the terms of the GNU General Public License version 3 as
  13  *  published by the Free Software Foundation.
  14  *
  15  *  mlp-mnist is distributed in the hope that it will be useful, but
  16  *  WITHOUT ANY WARRANTY; without even the implied warranty of
  17  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  *  General Public License for more details.
  19  *
  20  *  You should have received a copy of the GNU General Public License
  21  *  along with mlp-mnist.  If not, see <http://www.gnu.org/licenses/>.
  22  *
  23  */
  24
  25 // $Id: images.h,v 1.1 2005-12-13 17:19:11 fleuret Exp $
  26
  27 #ifndef IMAGES_H
  28 #define IMAGES_H
  29
  30 #include <iostream>
  31 #include <fstream>
  32 #include <stdint.h>
  33
  34 using namespace std;
  35
  36 class PixelMaps {
  37 public:
  38   unsigned int _nb_ref;
  39   unsigned char *_core;
  40   PixelMaps(int size);
  41   ~PixelMaps();
  42   PixelMaps *add_ref();
  43   void del_ref();
  44 };
  45
  46 class ImageSet {
  47   int _nb_pics, _nb_obj;
  48   int _width, _height;
  49   PixelMaps *_pixel_maps;
  50   unsigned char **_pixels, *_labels;
  51   bool *_used_picture;
  52
  53 public:
  54   ImageSet();
  55   ~ImageSet();
  56
  57   inline int nb_pics() { return _nb_pics; }
  58   inline int nb_obj() { return _nb_obj; }
  59   inline int width() { return _width; }
  60   inline int height() { return _height; }
  61   inline unsigned char *pixels(int p) { return _pixels[p]; }
  62   inline unsigned char pixel(int p, int x, int y) { return _pixels[p][x + y * _width]; }
  63   inline unsigned char label(int p) { return _labels[p]; }
  64
  65   void reset_used_pictures();
  66   int nb_unused_pictures();
  67   int pick_unused_picture();
  68
  69   void load_mnist_format(char *picture_file_name, char *label_file_name);
  70
  71   void sample_among_unused_pictures(ImageSet &is, int nb);
  72
  73 };
  74
  75 #endif