def test_endtoend_twocats(self): inpfile = os.path.join(TEST_DIRECTORY, 'test_images', 'twocats.png') im = lab.load_color_image(inpfile) oim = object_hash(im) result = lab.seam_carving(im, 100) lab.save_color_image(result, '/Users/yaxinliu/Downloads/lab1/Seam.png')
def seams_one(images): for i in images: inpfile = os.path.join(TEST_DIRECTORY, 'test_images', f'{i}.png') im = lab.load_color_image(inpfile) oim = object_hash(im) result = lab.seam_carving(im, 1) assert object_hash(im) == oim, 'Be careful not to modify the original image!' expfile = os.path.join(TEST_DIRECTORY, 'test_results', f'{i}_1seam.png') compare_color_images(result, lab.load_color_image(expfile))
def seams_endtoend(inp_name, out_name, number): inpfile = os.path.join(TEST_DIRECTORY, 'test_images', inp_name) im = lab.load_color_image(inpfile) oim = object_hash(im) for i in range(1, number): result = lab.seam_carving(im, i) assert object_hash(im) == oim, 'Be careful not to modify the original image!' expfile = os.path.join(TEST_DIRECTORY, 'test_results', out_name, f'{i:02d}.png') compare_color_images(result, lab.load_color_image(expfile))
def test_endtoend_smallfrog(self): inpfile = os.path.join(TEST_DIRECTORY, 'test_images', 'smallfrog.png') im = lab.load_color_image(inpfile) oim = object_hash(im) for i in range(1, 31): result = lab.seam_carving(im, i) self.assertEqual(object_hash(im), oim, 'Be careful not to modify the original image!') expfile = os.path.join(TEST_DIRECTORY, 'test_results', 'seams_smallfrog', f'{i:02d}.png') self.compare_color_images(result, lab.load_color_image(expfile))