bachpc commited on
Commit
9bdc8fc
1 Parent(s): c58ad4a

Modify class threshold

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -27,13 +27,13 @@ structure_class_names = [
27
  ]
28
  structure_class_map = {k: v for v, k in enumerate(structure_class_names)}
29
  structure_class_thresholds = {
30
- 'table': 0.5,
31
- 'table column': 0.5,
32
- 'table row': 0.5,
33
- 'table column header': 0.25,
34
- 'table projected row header': 0.25,
35
- 'table spanning cell': 0.25,
36
- 'no object': 10
37
  }
38
 
39
 
 
27
  ]
28
  structure_class_map = {k: v for v, k in enumerate(structure_class_names)}
29
  structure_class_thresholds = {
30
+ "table": 0.42,
31
+ "table column": 0.56,
32
+ "table row": 0.5,
33
+ "table column header": 0.38,
34
+ "table projected row header": 0.27,
35
+ "table spanning cell": 0.45,
36
+ "no object": 10
37
  }
38
 
39