Skip to content

TensorRT implementation of "RepVGG: Making VGG-style ConvNets Great Again"

License

Notifications You must be signed in to change notification settings

upczww/TensorRT-RepVGG

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RepVGG

RepVGG models from "RepVGG: Making VGG-style ConvNets Great Again" https://arxiv.org/pdf/2101.03697.pdf

For the Pytorch implementation, you can refer to DingXiaoH/RepVGG

How to run

  1. generate wts file.
git clone https://github.com/DingXiaoH/RepVGG.git
cd RepVGG

You may convert a trained model into the inference-time structure with

python convert.py [weights file of the training-time model to load] [path to save] -a [model name]

For example,

python convert.py RepVGG-B2-train.pth RepVGG-B2-deploy.pth -a RepVGG-B2

Then copy TensorRT-RepVGG/gen_wts.py to RepVGG and generate .wts file, for example

python gen_wts.py -w RepVGG-B2-deploy.pth -s RepVGG-B2.wts
  1. build and run
cd TensorRT-RepVGG

mkdir build

cd build

cmake ..

make

sudo ./repvgg -s RepVGG-B2  // serialize model to plan file i.e. 'RepVGG-B2.engine'
sudo ./repvgg -d RepVGG-B2  // deserialize plan file and run inference

About

TensorRT implementation of "RepVGG: Making VGG-style ConvNets Great Again"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published