def test_singleImage(self): image = self.makeImage(numpy.arange(12), 0, 'file.jpg') classes = [0] expected = {} expected['data'] = arrayEqualsTo(numpy.arange(12).reshape(-1,1)) expected['labels'] = [0] expected['filenames'] = ['file.jpg'] target = SingleBatchBuilder() self.assertEqual(expected,target.build([image], classes))
def test_multipleLabels(self): image1 = self.makeImage(numpy.ones(12)*1, 0, 'file1.jpg') image2 = self.makeImage(numpy.ones(12)*2, 3, 'file2.jpg') image3 = self.makeImage(numpy.ones(12)*3, 5, 'file3.jpg') classes = [0,3,5] expected = {} data = numpy.asarray([numpy.ones(12), numpy.ones(12)*2, numpy.ones(12)*3]).T expected['data'] = arrayEqualsTo(data) expected['labels'] = [0,1,2] expected['filenames'] = ['file1.jpg','file2.jpg','file3.jpg'] target = SingleBatchBuilder() result = target.build([image1, image2, image3], classes) self.assertEqual(expected,result)
def Create(nTrainingBatches = 1): return ConvnetBatchCreator(BatchBuilder(SingleBatchBuilder(), MetaBatchBuilder(), nTrainingBatches), BatchRepository(FileSystem(), cPickleSerializer()))