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()))
