This is a TensorRT implement of Tensorflow_PSENet
How to run
- build
pse
algorithm
cd pse
make
- prepare trained checkpoints in
./model
, likes
./model
--checkpoint
--model.ckpt-606001.data-00000-of-00001
--model.ckpt-606001.index
--model.ckpt-606001.meta
- run main.py
python main.py --model_dir MODEL_DIR --engine_path ENGINE_PATH --image_path IMAGE_PATH