def test_general(self):
batch_size = 2
generator = generators.ImageGenerator(
testUtils.Test_carsDb.CARS_DB_PATH,
rootdir=testUtils.Test_carsDb.CARS_DB_ROOTDIR,
batch_size=batch_size)
self.assertEqual(len(generator), 2) # 2 batches (with 2 and 1 images).
batch = generator[0]
self.assertTrue(isinstance(batch, dict))
self.assertTrue('image' in batch)
self.assertTrue(isinstance(batch['image'], list))
self.assertEqual(len(batch['image']), batch_size)
self.assertTrue('mask' in batch)
self.assertTrue('objects' in batch)
self.assertTrue('imagefile' in batch)
self.assertTrue('name' in batch)
self.assertTrue('score' in batch)
self.assertTrue(isinstance(batch['objects'], list))
# sample['objects'] has a list of dicts. Check each dict.
self.assertEqual(len(batch['objects']), batch_size)
objects_in_sample = batch['objects'][0]
self.assertTrue(isinstance(objects_in_sample, list))
object_in_sample = objects_in_sample[0]
self.assertTrue(isinstance(object_in_sample, dict))
self.assertTrue('x1' in object_in_sample)
self.assertTrue('y1' in object_in_sample)
self.assertTrue('width' in object_in_sample)
self.assertTrue('height' in object_in_sample)
self.assertTrue('name' in object_in_sample)
self.assertTrue('score' in object_in_sample)
def test_where_object(self):
# All objects should be "cars".
generator = generators.ImageGenerator(
testUtils.Test_carsDb.CARS_DB_PATH,
rootdir=testUtils.Test_carsDb.CARS_DB_ROOTDIR,
where_object='name == "car"')
batch = generator[0]
self.assertEqual(len(batch['objects']), 1) # 1 car in the 1st image.
# All objects should be "bus".
generator = generators.ImageGenerator(
testUtils.Test_carsDb.CARS_DB_PATH,
rootdir=testUtils.Test_carsDb.CARS_DB_ROOTDIR,
where_object='name == "bus"')
batch = generator[0]
self.assertEqual(len(batch['objects'][0]),
0) # No buses in the 1st image.
def test_used_keys_list(self):
batch_size = 2
generator = generators.ImageGenerator(
testUtils.Test_carsDb.CARS_DB_PATH,
rootdir=testUtils.Test_carsDb.CARS_DB_ROOTDIR,
used_keys=['mask', 'score'],
batch_size=batch_size)
self.assertEqual(len(generator), 2) # 2 batches (with 2 and 1 objects)
batch = generator[0]
self.assertTrue(isinstance(batch, list))
self.assertEqual(len(batch), 2)
def test_used_keys_dict(self):
batch_size = 2
generator = generators.ImageGenerator(
testUtils.Test_carsDb.CARS_DB_PATH,
rootdir=testUtils.Test_carsDb.CARS_DB_ROOTDIR,
used_keys={
'mask': 'mask_',
'score': 'score_'
},
batch_size=batch_size)
self.assertEqual(len(generator), 2) # 2 batches (with 2 and 1 images).
batch = generator[0]
self.assertTrue(isinstance(batch, dict))
self.assertEqual(len(batch), 2)
self.assertTrue('mask_' in batch)
self.assertTrue('score_' in batch)
def test_where_image(self):
generator = generators.ImageGenerator(
testUtils.Test_carsDb.CARS_DB_PATH,
rootdir=testUtils.Test_carsDb.CARS_DB_ROOTDIR,
where_image='imagefile == "images/000000.jpg"')
self.assertEqual(len(generator), 1) # 1 image.