93e728769d
* create psenet create psenet with weight from tensorflow * delete some useless code * repalce tab with 4 blanks * fix network bug, rewrite post-processing pse algorithm * update readme * update readme * add RepVGG * fix typo * add hrnetseg w18 w32 w48 * add hrnetseg with ocr w18 w32 w48 * merge hrnet and small, add hrnet_ocr * fix warning * change project name * add efficientnet |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| efficientnet.cpp | ||
| gen_wts.py | ||
| logging.h | ||
| README.md | ||
| utils.hpp | ||
EfficientNet
A TensorRT implementation of EfficientNet. For the Pytorch implementation, you can refer to EfficientNet-PyTorch
How to run
- install
efficientnet_pytorch
pip install efficientnet_pytorch
- gennerate
.wtsfile
python gen_wts.py
- build
mkdir build
cd build
cmake ..
make
- serialize model to engine
./efficientnet -s [.wts] [.engine] [b0 b1 b2 b3 ... b7] // serialize model to engine file
such as
./efficientnet -s ../efficientnet-b3.wts efficientnet-b3.engine b3
- deserialize and do infer
./efficientnet -d [.engine] [b0 b1 b2 b3 ... b7] // deserialize engine file and run inference
such as
./efficientnet -d efficientnet-b3.engine b3
- see if the output is same as pytorch side
For more models, please refer to tensorrtx