Skip to content
Snippets Groups Projects
  • q.yao's avatar
    6ccb1c0f
    [Feature] add onnxruntime test tool (#498) · 6ccb1c0f
    q.yao authored
    * add onnxruntime test tool, update pytorch2onnx to support slice export
    
    * onnx convert with custom output shape, update test code
    
    * update pytorch2onnx, add rescale_shape support, add document
    
    * update doc for lint error fixing
    
    * remove cpu flag in ort_test.py
    
    * change class name, fix cuda error
    
    * remote comment
    
    * fix bug of torch2onnx
    
    * mIOU to mIoU
    6ccb1c0f
    History
    [Feature] add onnxruntime test tool (#498)
    q.yao authored
    * add onnxruntime test tool, update pytorch2onnx to support slice export
    
    * onnx convert with custom output shape, update test code
    
    * update pytorch2onnx, add rescale_shape support, add document
    
    * update doc for lint error fixing
    
    * remove cpu flag in ort_test.py
    
    * change class name, fix cuda error
    
    * remote comment
    
    * fix bug of torch2onnx
    
    * mIOU to mIoU