ITSAIDI
commited on
Commit
·
fe068bb
1
Parent(s):
0a3b303
hgh
Browse files- utilitis.py +1 -0
utilitis.py
CHANGED
|
@@ -77,6 +77,7 @@ def unnormalize_box(bbox, width, height):
|
|
| 77 |
def get_word(bboxes,image):
|
| 78 |
x_min, y_min, x_max, y_max = bboxes
|
| 79 |
roi = image.crop((x_min, y_min, x_max, y_max)) # Region of intrest
|
|
|
|
| 80 |
roi_np = np.array(roi) # To array
|
| 81 |
result = ocr.ocr(roi_np, cls=True) # Apply OCR to ROI
|
| 82 |
if result != [None]:
|
|
|
|
| 77 |
def get_word(bboxes,image):
|
| 78 |
x_min, y_min, x_max, y_max = bboxes
|
| 79 |
roi = image.crop((x_min, y_min, x_max, y_max)) # Region of intrest
|
| 80 |
+
roi = enhance_image(roi,1,1.5)
|
| 81 |
roi_np = np.array(roi) # To array
|
| 82 |
result = ocr.ocr(roi_np, cls=True) # Apply OCR to ROI
|
| 83 |
if result != [None]:
|