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