ContextQMD
Back to Tensorrtx

EfficientNet

efficientnet/README.md

latest915 B
Original Source

EfficientNet

A TensorRT implementation of EfficientNet. For the Pytorch implementation, you can refer to EfficientNet-PyTorch

How to run

  1. install efficientnet_pytorch
pip install efficientnet_pytorch
  1. gennerate .wts file
python gen_wts.py
  1. build
mkdir build
cd build
cmake ..
make
  1. 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
  1. 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
  1. see if the output is same as pytorch side

For more models, please refer to tensorrtx