add onnx exporter for mobilenet networks

lufficc / SSD

High quality, fast, modular reference implementation of SSD in PyTorch

MIT License

1.52k stars 384 forks source link

add onnx exporter for mobilenet networks #200

Closed SamSamhuns closed 1 year ago

SamSamhuns commented 3 years ago

fixed cpu inference mode in demo.py linter edits in README add venv ignore in .gitignore add pytorch2onnx.py to convert mobilnets to onnx models changed inference.py to include PostprocessorOnnx class