Update EasyOpticalCharacterRecognition.py
Browse files
EasyOpticalCharacterRecognition.py
CHANGED
|
@@ -5,7 +5,6 @@ from tensorflow.keras.models import load_model
|
|
| 5 |
from tensorflow.keras.preprocessing.image import img_to_array
|
| 6 |
import pickle
|
| 7 |
import os
|
| 8 |
-
from google.colab import drive
|
| 9 |
|
| 10 |
# === Load model and label encoder ===
|
| 11 |
model_path = 'MobileNetBest_Model.h5'
|
|
@@ -14,15 +13,6 @@ pkl_path = 'MobileNet_Label_Encoder.pkl'
|
|
| 14 |
model = load_model(model_path)
|
| 15 |
print("✅ Model loaded.")
|
| 16 |
|
| 17 |
-
if os.path.exists(pkl_path):
|
| 18 |
-
with open(pkl_path, 'rb') as f:
|
| 19 |
-
label_map = pickle.load(f)
|
| 20 |
-
index_to_label = {v: k for k, v in label_map.items()}
|
| 21 |
-
print("✅ Label encoder loaded.")
|
| 22 |
-
else:
|
| 23 |
-
index_to_label = {0: "Handwritten", 1: "Computerized"}
|
| 24 |
-
print("⚠️ Label encoder not found, using default mapping.")
|
| 25 |
-
|
| 26 |
# === Classification function ===
|
| 27 |
def classify_text_region(region_img):
|
| 28 |
try:
|
|
@@ -73,6 +63,4 @@ def AnnotatedTextDetection_EasyOCR_from_array(img):
|
|
| 73 |
def process_image(input_image):
|
| 74 |
img = cv2.cvtColor(input_image, cv2.COLOR_RGB2BGR)
|
| 75 |
result_img, text_result = AnnotatedTextDetection_EasyOCR_from_array(img)
|
| 76 |
-
return result_img, text_result
|
| 77 |
-
|
| 78 |
-
|
|
|
|
| 5 |
from tensorflow.keras.preprocessing.image import img_to_array
|
| 6 |
import pickle
|
| 7 |
import os
|
|
|
|
| 8 |
|
| 9 |
# === Load model and label encoder ===
|
| 10 |
model_path = 'MobileNetBest_Model.h5'
|
|
|
|
| 13 |
model = load_model(model_path)
|
| 14 |
print("✅ Model loaded.")
|
| 15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
# === Classification function ===
|
| 17 |
def classify_text_region(region_img):
|
| 18 |
try:
|
|
|
|
| 63 |
def process_image(input_image):
|
| 64 |
img = cv2.cvtColor(input_image, cv2.COLOR_RGB2BGR)
|
| 65 |
result_img, text_result = AnnotatedTextDetection_EasyOCR_from_array(img)
|
| 66 |
+
return result_img, text_result
|
|
|
|
|
|