Name		Name	Last commit message	Last commit date
parent directory ..
samples		samples
CMakeLists.txt		CMakeLists.txt
README.md		README.md
config.yaml		config.yaml
efficientnet.cpp		efficientnet.cpp
efficientnet.h		efficientnet.h
export_onnx.py		export_onnx.py
inference.py		inference.py
label.txt		label.txt
main.cpp		main.cpp

README.md

EfficientNet Keras=>ONNX=>TensorRT

1.Reference

efficientnet arxiv: EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks
efficientnet github: https://github.com/qubvel/efficientnet
keras2onnx: https://github.com/onnx/keras-onnx
pypi: https://pypi.org/project/efficientnet

run this command to install efficientnet

pip install efficientnet

2.Export ONNX Model

python3 export_onnx.py

3.Build efficientnet_trt Project

mkdir build && cd build
cmake ..
make -j

4.run efficientnet_trt

./efficientnet_trt ../config.yaml ../samples