Convert Pytorch pretrain -> TensoRT engine directly for CRAFT (Character-Region Awareness For Text detection)
- Convert CRAFT Text detection pretrain Pytorch model into TensorRT engine directly, without ONNX step between
- CRAFT: (forked from https://github.com/clovaai/CRAFT-pytorch) Official Pytorch implementation of CRAFT text detector | Paper | Pretrained Model | Supplementary
- Using torch2trt_dynamic from https://github.com/grimoire/torch2trt_dynamic (branch of https://github.com/NVIDIA-AI-IOT/torch2trt with dynamic shapes support)
Implementation of convenient converter from Pytorch to Tensor RT directly for CRAFT text detector. This repo is only about converting Pytorch model into Tensor RT. For Tensor RT inference, check out:
- Inference using Tensor RT standalone (https://github.com/k9ele7en/ONNX-TensorRT-Inference-CRAFT-pytorch)
- Advance inference pipeline using NVIDIA Triton Server (https://github.com/k9ele7en/triton-tensorrt-CRAFT-pytorch)
k9ele7en. Give 1 star if you find some value in this repo.
Thank you.
[MIT License] A short, permissive software license. Basically, you can do whatever you want as long as you include the original copyright and license notice in any copy of the software/source.
7 Aug, 2021: Initial repo, converter run success.
$ pip install -r requirements.txt
Check details at ./README_Env.md
| Model name | Used datasets | Languages | Purpose | Model Link |
|---|---|---|---|---|
| General | SynthText, IC13, IC17 | Eng + MLT | For general purpose | Click |
| IC15 | SynthText, IC15 | Eng | For IC15 only | Click |
| LinkRefiner | CTW1500 | - | Used with the General Model | Click |
$ cd converters
$ python torch2trt.py
INFO - Converting CRAFT Pytorch pth to TensorRT engine... (torch2trt.py:31)
...
INFO - Compare Pytorch output vs TensorRT engine output...
...
INFO - Convert CRAFT Pytorch pth to tensorRT engine sucess (torch2trt.py:83)