Skip to content

upczww/TensorRT-Python-PSENet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Releases

No releases published

Packages

No packages published

Languages