MengniWang commited on
Commit
99aa815
1 Parent(s): cce3621

upload int8 onnx models

Browse files

Signed-off-by: Mengni Wang <mengni.wang@intel.com>

README.md CHANGED
@@ -1,3 +1,42 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ tags:
4
+ - table-transformer
5
+ - table detection
6
+ - table structure recognition
7
+ - int8
8
+ - Intel® Neural Compressor
9
+ - neural-compressor
10
+ - PostTrainingStatic
11
+ - onnx
12
+ datasets:
13
+ - bsmock/pubtables-1m
14
  ---
15
+ # INT8 Table Transformer
16
+
17
+ ## Post-training static quantization
18
+
19
+ ### ONNX
20
+
21
+ This repo contains the models for:
22
+
23
+ 1) Table detection,
24
+ 2) Table structure recognition,
25
+
26
+ The original FP32 PyTorch model comes from [bsmock/tatr-pubtables1m-v1.0](https://huggingface.co/bsmock/tatr-pubtables1m-v1.0). The INT8 ONNX models are quantized with [Intel® Neural Compressor](https://github.com/intel/neural-compressor).
27
+
28
+ Refer to this [link](https://github.com/intel/neural-compressor/tree/master/examples/onnxrt/object_detection/table_transformer/quantization/ptq_static) for model preparation, quantization and benchmark scripts.
29
+
30
+ #### Test result
31
+
32
+ Table detection:
33
+
34
+ | |INT8|FP32|
35
+ |---|:---:|:---:|
36
+ | **COCO metrics (AP)** |0.9691|0.9706|
37
+ | **Model size (MB)** |56|111|
38
+
39
+ Table structure recognition:
40
+ | |INT8|FP32|
41
+ |---|:---:|:---:|
42
+ | **Model size (MB)** |56|111|
pubtables1m_detection_detr_r18_static_int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5f2ab65ab041671f1f0cda9bb04763a723a4fe9fe06660f35dfb6e5bb34823a
3
+ size 58252723
pubtables1m_structure_detr_r18_static_int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d97579a38190965a99f8a8299428384b29e325eab589d9c9183fc6f139318e4a
3
+ size 58287283