datdo2717 commited on
Commit
816ed2b
·
1 Parent(s): c5b5437
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ from Segmentation import *
6
  from perpestive_transform import *
7
  from Rotate.rotate_function import *
8
  from Rotate import rotate_function
9
- from Restomer import *
10
  def process_image(input_image):
11
  original_img=segmentation_doc(input_image)
12
  output_images=[]
@@ -17,7 +17,7 @@ def process_image(input_image):
17
  mask_img=cv2.imread(mask)
18
  extract_document=extract(original_img,mask_img)
19
  result=rotate_function.rotate(extract_document)
20
- result=run_script(result, 720, 32)
21
  output_images.append(result)
22
  return output_images
23
 
 
6
  from perpestive_transform import *
7
  from Rotate.rotate_function import *
8
  from Rotate import rotate_function
9
+ # from Restomer import *
10
  def process_image(input_image):
11
  original_img=segmentation_doc(input_image)
12
  output_images=[]
 
17
  mask_img=cv2.imread(mask)
18
  extract_document=extract(original_img,mask_img)
19
  result=rotate_function.rotate(extract_document)
20
+ # result=run_script(result, 720, 32)
21
  output_images.append(result)
22
  return output_images
23