nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision
27 stars 6 forks source link

[FEATURE] Update ONNX exporter for general cases #30

Open alphinside opened 4 years ago

alphinside commented 4 years ago

Is your feature request related to a problem? Please describe. DETR convert to ONNX error, because currently vortex exporter for object detection is tightly coupled with NMS (and for handling Pytorch limitation on ONNX export )

Describe the solution you'd like A clear and concise description of what you want to happen.

Describe alternatives you've considered A clear and concise description of any alternative solutions or features you've considered.

Additional context Add any other context or screenshots about the feature request here.

Please complete it @alifahrri