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

README.md

ghostnet PyTorch=>ONNX=>TensorRT

1.Reference

arxiv: GhostNet: More Features from Cheap Operations
github: https://github.com/huawei-noah/ghostnet
get ghostnet weights from here: ghostnet/pytorch

2.Export ONNX Model

python3 export_onnx.py

3.Build ghostnet_trt Project

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

4.run ghostnet_trt

./ghostnet_trt ../config.yaml ../samples