mirror of
https://gitee.com/microic/tr.git
synced 2025-12-06 10:19:05 +08:00
71 lines
2.2 KiB
Python
Executable File
71 lines
2.2 KiB
Python
Executable File
# coding: utf-8
|
|
import tr
|
|
import sys, cv2, time, os
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
import numpy as np
|
|
|
|
_BASEDIR = os.path.dirname(os.path.abspath(__file__))
|
|
os.chdir(_BASEDIR)
|
|
|
|
|
|
def test():
|
|
print("recognize", tr.recognize("imgs/line.png"))
|
|
|
|
img_path = "imgs/id_card.jpeg"
|
|
# img_path = "imgs/name_card.jpg"
|
|
|
|
img_pil = Image.open(img_path)
|
|
try:
|
|
if hasattr(img_pil, '_getexif'):
|
|
# from PIL import ExifTags
|
|
# for orientation in ExifTags.TAGS.keys():
|
|
# if ExifTags.TAGS[orientation] == 'Orientation':
|
|
# break
|
|
orientation = 274
|
|
exif = dict(img_pil._getexif().items())
|
|
if exif[orientation] == 3:
|
|
img_pil = img_pil.rotate(180, expand=True)
|
|
elif exif[orientation] == 6:
|
|
img_pil = img_pil.rotate(270, expand=True)
|
|
elif exif[orientation] == 8:
|
|
img_pil = img_pil.rotate(90, expand=True)
|
|
except:
|
|
pass
|
|
|
|
MAX_SIZE = 1600
|
|
if img_pil.height > MAX_SIZE or img_pil.width > MAX_SIZE:
|
|
scale = max(img_pil.height / MAX_SIZE, img_pil.width / MAX_SIZE)
|
|
|
|
new_width = int(img_pil.width / scale + 0.5)
|
|
new_height = int(img_pil.height / scale + 0.5)
|
|
img_pil = img_pil.resize((new_width, new_height), Image.ANTIALIAS)
|
|
|
|
color_pil = img_pil.convert("RGB")
|
|
gray_pil = img_pil.convert("L")
|
|
|
|
img_draw = ImageDraw.Draw(color_pil)
|
|
colors = ['red', 'green', 'blue', "purple"]
|
|
|
|
t = time.time()
|
|
n = 1
|
|
for _ in range(n):
|
|
tr.detect(gray_pil, flag=tr.FLAG_RECT)
|
|
print("time", (time.time() - t) / n)
|
|
|
|
results = tr.run(gray_pil, flag=tr.FLAG_ROTATED_RECT)
|
|
|
|
for i, rect in enumerate(results):
|
|
cx, cy, w, h, a = tuple(rect[0])
|
|
print(i, "\t", rect[1], rect[2])
|
|
box = cv2.boxPoints(((cx, cy), (w, h), a))
|
|
box = np.int0(np.round(box))
|
|
|
|
for p1, p2 in [(0, 1), (1, 2), (2, 3), (3, 0)]:
|
|
img_draw.line(xy=(box[p1][0], box[p1][1], box[p2][0], box[p2][1]), fill=colors[i % len(colors)], width=2)
|
|
|
|
color_pil.show()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test()
|