├── .gitignore ├── DocShadow ├── models │ ├── __init__.py │ ├── backbone.py │ ├── blocks.py │ ├── blocks_.py │ └── model.py └── utils │ ├── __init__.py │ └── utils.py ├── LICENSE ├── README.md ├── assets ├── latency.png └── sample.jpg ├── eval.py ├── export.py ├── export_coreml.py ├── infer.py ├── onnx_runner ├── __init__.py └── docshadow.py ├── requirements-onnx.txt ├── requirements.txt └── weights ├── .gitkeep └── download.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/.gitignore -------------------------------------------------------------------------------- /DocShadow/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import DocShadow 2 | -------------------------------------------------------------------------------- /DocShadow/models/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/DocShadow/models/backbone.py -------------------------------------------------------------------------------- /DocShadow/models/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/DocShadow/models/blocks.py -------------------------------------------------------------------------------- /DocShadow/models/blocks_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/DocShadow/models/blocks_.py -------------------------------------------------------------------------------- /DocShadow/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/DocShadow/models/model.py -------------------------------------------------------------------------------- /DocShadow/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import load_checkpoint 2 | -------------------------------------------------------------------------------- /DocShadow/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/DocShadow/utils/utils.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/README.md -------------------------------------------------------------------------------- /assets/latency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/assets/latency.png -------------------------------------------------------------------------------- /assets/sample.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/assets/sample.jpg -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/eval.py -------------------------------------------------------------------------------- /export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/export.py -------------------------------------------------------------------------------- /export_coreml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/export_coreml.py -------------------------------------------------------------------------------- /infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/infer.py -------------------------------------------------------------------------------- /onnx_runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/onnx_runner/__init__.py -------------------------------------------------------------------------------- /onnx_runner/docshadow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/onnx_runner/docshadow.py -------------------------------------------------------------------------------- /requirements-onnx.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/requirements-onnx.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/requirements.txt -------------------------------------------------------------------------------- /weights/.gitkeep: -------------------------------------------------------------------------------- 1 | ONNX models will be exported here. 2 | -------------------------------------------------------------------------------- /weights/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fabio-sim/DocShadow-ONNX-TensorRT/HEAD/weights/download.sh --------------------------------------------------------------------------------