svrt.c

   1
   2 /*
   3  *  svrt is the ``Synthetic Visual Reasoning Test'', an image
   4  *  generator for evaluating classification performance of machine
   5  *  learning systems, humans and primates.
   6  *
   7  *  Copyright (c) 2017 Idiap Research Institute, http://www.idiap.ch/
   8  *  Written by Francois Fleuret <francois.fleuret@idiap.ch>
   9  *
  10  *  This file is part of svrt.
  11  *
  12  *  svrt is free software: you can redistribute it and/or modify it
  13  *  under the terms of the GNU General Public License version 3 as
  14  *  published by the Free Software Foundation.
  15  *
  16  *  svrt is distributed in the hope that it will be useful, but
  17  *  WITHOUT ANY WARRANTY; without even the implied warranty of
  18  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19  *  General Public License for more details.
  20  *
  21  *  You should have received a copy of the GNU General Public License
  22  *  along with selector.  If not, see <http://www.gnu.org/licenses/>.
  23  *
  24  */
  25
  26 #include <TH/TH.h>
  27
  28 #include "svrt_generator.h"
  29
  30 THByteTensor *generate_vignettes(long n_problem, THLongTensor *labels) {
  31   struct VignetteSet vs;
  32   long nb_vignettes;
  33   long st0, st1, st2;
  34   long v, i, j;
  35   long *m, *l;
  36   unsigned char *a, *b;
  37
  38   nb_vignettes = THLongTensor_size(labels, 0);
  39   m = THLongTensor_storage(labels)->data + THLongTensor_storageOffset(labels);
  40   st0 = THLongTensor_stride(labels, 0);
  41   l = (long *) malloc(sizeof(long) * nb_vignettes);
  42   for(v = 0; v < nb_vignettes; v++) {
  43     l[v] = *m;
  44     m += st0;
  45   }
  46
  47   svrt_generate_vignettes(n_problem, nb_vignettes, l, &vs);
  48   free(l);
  49
  50   THLongStorage *size = THLongStorage_newWithSize(3);
  51   size->data[0] = vs.nb_vignettes;
  52   size->data[1] = vs.height;
  53   size->data[2] = vs.width;
  54
  55   THByteTensor *result = THByteTensor_newWithSize(size, NULL);
  56   THLongStorage_free(size);
  57
  58   st0 = THByteTensor_stride(result, 0);
  59   st1 = THByteTensor_stride(result, 1);
  60   st2 = THByteTensor_stride(result, 2);
  61
  62   unsigned char *r = vs.data;
  63   for(v = 0; v < vs.nb_vignettes; v++) {
  64     a = THByteTensor_storage(result)->data + THByteTensor_storageOffset(result) + v * st0;
  65     for(i = 0; i < vs.height; i++) {
  66       b = a + i * st1;
  67       for(j = 0; j < vs.width; j++) {
  68         *b = (unsigned char) (*r);
  69         r++;
  70         b += st2;
  71       }
  72     }
  73   }
  74
  75   free(vs.data);
  76
  77   return result;
  78 }