Skip to content

handoku/crnn_tensorrt

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 

Repository files navigation

crnn_tensorrt

build crnn engine from scratch with pure tensorrt api

tensorrt version

v7.2

Feature

  • Implement crnn network from scratch with pure tensorrt layer api
  • Implementing lstm with ILoop API instead of RNNv2
  • Construct network in full dimension mode, support dynamic batch/shape

build and test

cd crnn
mkdir build && cd build && cmake ..
make

generate engine

First, download weights file : crnn.wts

./crnn -s

run test

./crnn -d

About

implement crnn with tensorrt layer api

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages