ONNX support - Githubissues

voldemortX / pytorch-auto-drive

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help

BSD 3-Clause "New" or "Revised" License

839 stars 137 forks source link

ONNX support #42

Open voldemortX opened 2 years ago

voldemortX commented 2 years ago

[x] Add doc
[ ] Testing on datasets with onnxruntime
[ ] Visualization with onnxtuntime

Currently unsupported models:

[ ] ENet
[x] SCNN