- nanodet: https://github.com/RangiLyu/nanodet
- get nanodet pretrained weights from here: COCO pretrain weight for torch>=1.6(Google Drive) | COCO pretrain weight for torch<=1.5(Google Drive)
git clone https://github.com/RangiLyu/nanodet.git
copy export.py into nanodet/tools and run export.py to generate nanodet-m.onnx.
not support dynamic batch
python3 tools/export.py
INPUT
[1,3,320,320]
OUTPUT
[1,1600,80]
[1,400,80]
[1,100,80]
[1,1600,32]
[1,400,32]
[1,100,32]
nanodet-m
nanodet-g
nanodet-EfficientNet-Lite0_320
nanodet-EfficientNet-Lite1_416
nanodet-EfficientNet-Lite2_512
nanodet-RepVGG-A0_416