def test_all_1(self): with tempfile.NamedTemporaryFile(suffix='.jpg') as tmpfile: in_img = PIL.Image.open("tests/data/black_border_problem.jpg") out_img = pillowfight.ace(in_img, seed=0xDEADBEE) # unpaper order out_img = pillowfight.unpaper_blackfilter(out_img) out_img = pillowfight.unpaper_noisefilter(out_img) out_img = pillowfight.unpaper_blurfilter(out_img) out_img = pillowfight.unpaper_masks(out_img) out_img = pillowfight.unpaper_grayfilter(out_img) out_img = pillowfight.unpaper_border(out_img) in_img.close() # beware of JPG compression self.assertEqual(out_img.mode, "RGB") out_img.save(tmpfile.name) out_img.close() out_img = PIL.Image.open(tmpfile.name) expected_img = PIL.Image.open( "tests/data/black_border_problem_all.jpg") self.assertEqual(out_img.tobytes(), expected_img.tobytes()) expected_img.close()
def test_noisefilter(self): with tempfile.NamedTemporaryFile(suffix='.jpg') as tmpfile: in_img = PIL.Image.open("tests/data/black_border_problem.jpg") out_img = pillowfight.unpaper_noisefilter(in_img) in_img.close() # beware of JPG compression self.assertEqual(out_img.mode, "RGB") out_img.save(tmpfile.name) out_img.close() out_img = PIL.Image.open(tmpfile.name) expected_img = PIL.Image.open( "tests/data/black_border_problem_noisefilter.jpg") self.assertEqual(out_img.tobytes(), expected_img.tobytes()) expected_img.close()
out = img.resize(tuple(i * 2 for i in img.size), PIL.Image.LANCZOS) process = True if process: out = pillowfight.ace(img, slope=10, limit=1000, samples=100, seed=None) # Canny edge detection # out = pillowfight.canny(out) # Gaussian blur # out = pillowfight.gaussian(out, sigma=2.0, nb_stddev=5) # sobel # out = pillowfight.sobel(out) # Stroke Width Transform # SWT_OUTPUT_BW_TEXT = 0 # default # SWT_OUTPUT_GRAYSCALE_TEXT = 1 # SWT_OUTPUT_ORIGINAL_BOXES = 2 out = pillowfight.swt(out, output_type=2) #unpaper out = pillowfight.unpaper_blackfilter(out) out = pillowfight.unpaper_blurfilter(out) out = pillowfight.unpaper_border(out) # out = pillowfight.unpaper_grayfilter(out) #adds white rectangles? out = pillowfight.unpaper_masks(out) out = pillowfight.unpaper_noisefilter(out) print(pytesseract.image_to_string(out)) out.save("out.png")