/TensorRT-Python-PSENet

TensorRT Python api implement of PSENet (https://github.com/liuheng92/tensorflow_PSENet)

Primary LanguageC++

This is a TensorRT implement of Tensorflow_PSENet

How to run

  1. build pse algorithm
cd pse
make
  1. prepare trained checkpoints in ./model, likes
./model
 --checkpoint
 --model.ckpt-606001.data-00000-of-00001
 --model.ckpt-606001.index
 --model.ckpt-606001.meta
  1. run main.py
python main.py --model_dir MODEL_DIR --engine_path ENGINE_PATH  --image_path IMAGE_PATH