Sayeem26s commited on
Commit
b67482d
·
verified ·
1 Parent(s): 1b1761c

Update readme.md

Browse files
Files changed (1) hide show
  1. README.md +66 -3
README.md CHANGED
@@ -1,3 +1,66 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - object-detection
5
+ - chart-understanding
6
+ - document-ai
7
+ - cascade-rcnn
8
+ - swin-transformer
9
+ ---
10
+
11
+ # Chart Element Detector
12
+
13
+ A chart element detection model based on
14
+ [CACHED](https://github.com/pengyu965/ChartDete)
15
+ (Context-Aware Chart Element Detection).
16
+
17
+ ## Model Details
18
+
19
+ - **Architecture**: Cascade R-CNN + Swin Transformer + FPN
20
+ - **Task**: Chart element detection and localization
21
+ - **Classes**: 18 chart element classes
22
+ - **Dataset**: PMC Chart Dataset
23
+ - **COCO AP**: 0.729
24
+
25
+ ## Classes
26
+
27
+ x_tick_label, y_tick_label, x_tick, y_tick,
28
+ x_axis_title, y_axis_title, chart_title,
29
+ legend_marker, legend_label, legend_title,
30
+ value_label, mark_label, tick_grouping,
31
+ plot_area, x_axis_area, y_axis_area,
32
+ legend_area, others
33
+
34
+ ## Output Format
35
+ ```json
36
+ {
37
+ "chart": [
38
+ {
39
+ "x1": 10.0,
40
+ "y1": 20.0,
41
+ "x2": 100.0,
42
+ "y2": 200.0,
43
+ "score": 0.95,
44
+ "class": "chart_title"
45
+ }
46
+ ]
47
+ }
48
+ ```
49
+
50
+ ## Requirements
51
+ ```
52
+ torch==1.13.1
53
+ mmdet==2.28.2
54
+ mmcv-full==1.7.0
55
+ ```
56
+
57
+ ## Citation
58
+ ```bibtex
59
+ @inproceedings{yan2023cached,
60
+ title={CACHED: Context-Aware Chart Element Detection},
61
+ author={Yan, Pengyu and Ahmed, Saleem and Doermann, David},
62
+ booktitle={ICDAR},
63
+ year={2023}
64
+ }
65
+ ```
66
+ ```