Skip to content

chenyangMl/crnn_libtorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Introduce

Depolying pytorch-trained model with libtorch.

Requirment

  • cmake 3.0
  • libtorch-1.4(cpu)
  • opencv 4.1.1

Usage

cd crnn_libtorch
mkdir build && cd build
cmake -DCMAKE_PREFIX_PATH="/home/username/libtorch" ..
make -j4
./CrnnDeploy ../src/crnn.pt ../src/keys.txt ../src/test.jpg

Tips: modify the libtoch path for your lib path.

Crnn convert usage

You can convert your trained model with convert.py.

requirement

  • python 3.6
  • torch 1.2.0
cd crnn_libtorch
python crnn_convert/convert.py

Next Working

  • Model compression.

About

Deploy pytorch-trained model via libtorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published