Commit
·
47436d9
1
Parent(s):
061dadf
Update app.py
Browse files
app.py
CHANGED
|
@@ -79,7 +79,7 @@ def get_grayscale(image):
|
|
| 79 |
# Thresholding or Binarization
|
| 80 |
def thresholding(src):
|
| 81 |
return cv2.threshold(src,127,255, cv2.THRESH_TOZERO)[1]
|
| 82 |
-
|
| 83 |
def ocr_with_easy(img):
|
| 84 |
gray_scale_image=get_grayscale(img)
|
| 85 |
thresholding(gray_scale_image)
|
|
@@ -97,7 +97,7 @@ def generate_ocr(Method,img):
|
|
| 97 |
|
| 98 |
print("Method___________________",Method)
|
| 99 |
if Method == 'EasyOCR':
|
| 100 |
-
text_output =
|
| 101 |
if Method == 'KerasOCR':
|
| 102 |
text_output = ocr_with_keras(img)
|
| 103 |
if Method == 'PaddleOCR':
|
|
@@ -127,11 +127,12 @@ def save_details(Method,text_output,img):
|
|
| 127 |
if text_output:
|
| 128 |
splitted_path = os.path.splitext(picture_path)
|
| 129 |
modified_picture_path = splitted_path[0] + curr_datetime + splitted_path[1]
|
| 130 |
-
|
| 131 |
-
with open('
|
| 132 |
f.write(picture_path)
|
| 133 |
-
|
| 134 |
-
img.
|
|
|
|
| 135 |
input_img = modified_picture_path
|
| 136 |
try:
|
| 137 |
df = pd.read_csv("AllDetails.csv")
|
|
|
|
| 79 |
# Thresholding or Binarization
|
| 80 |
def thresholding(src):
|
| 81 |
return cv2.threshold(src,127,255, cv2.THRESH_TOZERO)[1]
|
| 82 |
+
readme.txt
|
| 83 |
def ocr_with_easy(img):
|
| 84 |
gray_scale_image=get_grayscale(img)
|
| 85 |
thresholding(gray_scale_image)
|
|
|
|
| 97 |
|
| 98 |
print("Method___________________",Method)
|
| 99 |
if Method == 'EasyOCR':
|
| 100 |
+
text_output = ocr_withreadme.txt_easy(img)
|
| 101 |
if Method == 'KerasOCR':
|
| 102 |
text_output = ocr_with_keras(img)
|
| 103 |
if Method == 'PaddleOCR':
|
|
|
|
| 127 |
if text_output:
|
| 128 |
splitted_path = os.path.splitext(picture_path)
|
| 129 |
modified_picture_path = splitted_path[0] + curr_datetime + splitted_path[1]
|
| 130 |
+
cv2.imwrite("myimage.jpg", img)
|
| 131 |
+
with open('readme.txt', 'w') as f:
|
| 132 |
f.write(picture_path)
|
| 133 |
+
print("write Successfully")
|
| 134 |
+
# img = Image.open(r"/home/user/app/")
|
| 135 |
+
# img.save(modified_picture_path)
|
| 136 |
input_img = modified_picture_path
|
| 137 |
try:
|
| 138 |
df = pd.read_csv("AllDetails.csv")
|