• Negin Raoof's avatar
    ONNX export for variable input sizes (#1840) · 986d2423
    Negin Raoof authored
    
    
    * fixes and tests for variable input size
    
    * transform test fix
    
    * Fix comment
    
    * Dynamic shape for keypoint_rcnn
    
    * Update test_onnx.py
    
    * Update rpn.py
    
    * Fix for split on RPN
    
    * Fixes for feedbacks
    
    * flake8
    
    * topk fix
    
    * Fix build
    
    * branch on tracing
    
    * fix for scalar tensor
    
    * Fixes for script type annotations
    
    * Update rpn.py
    
    * clean up
    
    * clean up
    
    * Update rpn.py
    
    * Updated for feedback
    
    * Fix for comments
    
    * revert to use tensor
    
    * Added test for box clip
    
    * Fixes for feedback
    
    * Fix for feedback
    
    * ORT version revert
    
    * Update ort
    
    * Update .travis.yml
    
    * Update test_onnx.py
    
    * Update test_onnx.py
    
    * Tensor sizes
    
    * Fix for dynamic split
    
    * Try disable tests
    
    * pytest verbose
    
    * revert one test
    
    * enable tests
    
    * Update .travis.yml
    
    * Update .travis.yml
    
    * Update .travis.yml
    
    * Update test_onnx.py
    
    * Update .travis.yml
    
    * Passing device
    
    * Fixes for test
    
    * Fix for boxes datatype
    
    * clean up
    Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
    986d2423
transform.py 8.99 KB