def __init__(self,
filename,
stream=None,
action=lambda *args: args[0],
label=None,
labeller=image.Labeller(),
normalizer=image.Normalizer(),
augmenter=image.Augmenter()):
self.action = action
self.label = label or labeller(filename)
self.norm = normalizer(stream or filename)
self.ext = self._ext()
self.augmenter = augmenter
def test_normalization_bkg_stream(self):
with open('resources/office.png', 'rb') as f:
norm = image.Normalizer(size=64, canvas=f)
img = norm('resources/bag.png')
self.assertEqual(img.shape, (64, 64, 4))
def test_normalization_bkg_path(self):
norm = image.Normalizer(size=64, canvas='resources/office.png')
img = norm('resources/bag.png')
self.assertEqual(img.shape, (64, 64, 4))
def test_normalization_colored_canvas(self):
norm = image.Normalizer(size=64, canvas='FF0000')
img = norm('resources/bag.png')
self.assertEqual(img.shape, (64, 64, 4))
def test_normalization_stream(self):
norm = image.Normalizer(size=64)
with open('resources/bag.png', 'rb') as f:
img = norm(f)
self.assertEqual(img.shape, (42, 64, 4))
def test_normalization_path(self):
norm = image.Normalizer(size=64)
img = norm('resources/bag.png')
self.assertEqual(img.shape, (42, 64, 4))