• Dmitry Sidnev's avatar
    [Feature] enable exporting to onnx for PointRend (#953) · a88d1d28
    Dmitry Sidnev authored
    
    
    * Fix export to onnx for PointRend
    
    * Fix codestyle
    
    * Fix codestyle
    
    * Fix type in docstring
    
    * Minor fix
    
    * Fix export with custom ops
    
    * Fix codestyle
    
    * Add tests for bilinear_grid_sample function
    
    * Remove redundant operation and rename variables
    
    * Fix bug in bilinear_grid_sample and update test
    
    * Fix getting batch size
    
    * skip torch==1.3.1
    
    * remove unused import
    
    * fix lint
    
    * support export with batch
    
    * fix dynamic clip
    
    * skip test for torch<1.5.0
    
    * Add docstrings and comments
    
    * Minor fix
    
    * Recover clipping code
    
    * Fix clamping in pytorch 1.7.0
    
    * Fix bilinear_grid_sampler
    
    * Minor fix
    Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
    a88d1d28
test_bilinear_grid_sample.py 1.76 KB