bachpc commited on
Commit
6353415
1 Parent(s): 2a27a15

Modify threshold

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -36,7 +36,7 @@ structure_class_thresholds = {
36
  "table row": 0.5,
37
  "table column header": 0.38,
38
  "table projected row header": 0.27,
39
- "table spanning cell": 0.45,
40
  "no object": 10
41
  }
42
 
@@ -252,7 +252,7 @@ def visualize_structure(pil_img, structure_result):
252
 
253
  plt.axis('off')
254
  img_buf = io.BytesIO()
255
- plt.savefig(img_buf, bbox_inches='tight', dpi=100)
256
 
257
  return PIL.Image.open(img_buf)
258
 
@@ -280,7 +280,7 @@ def visualize_cells(pil_img, cells):
280
 
281
  plt.axis('off')
282
  img_buf = io.BytesIO()
283
- plt.savefig(img_buf, bbox_inches='tight', dpi=100)
284
 
285
  return PIL.Image.open(img_buf)
286
 
 
36
  "table row": 0.5,
37
  "table column header": 0.38,
38
  "table projected row header": 0.27,
39
+ "table spanning cell": 0.4,
40
  "no object": 10
41
  }
42
 
 
252
 
253
  plt.axis('off')
254
  img_buf = io.BytesIO()
255
+ plt.savefig(img_buf, bbox_inches='tight', dpi=1000)
256
 
257
  return PIL.Image.open(img_buf)
258
 
 
280
 
281
  plt.axis('off')
282
  img_buf = io.BytesIO()
283
+ plt.savefig(img_buf, bbox_inches='tight', dpi=1000)
284
 
285
  return PIL.Image.open(img_buf)
286