• q.yao's avatar
    [Feature] : Add NonMaxSuppression TensorRT Plugin (#787) · 1ee5315e
    q.yao authored
    
    
    * start trt plugin prototype
    
    * Add test module, modify roialign convertor
    
    * finish roi_align trt plugin
    
    * fix conflict of RoiAlign and MMCVRoiAlign
    
    * fix for lint
    
    * fix test tensorrt module
    
    * test_tensorrt move import to test func
    
    * add except error type
    
    * add tensorrt to setup.cfg
    
    * code format with yapf
    
    * fix for clang-format
    
    * move tensorrt_utils to mmcv/tensorrt, add comments, better test module
    
    * fix line endings, docformatter
    
    * isort init, remove trailing whitespace
    
    * add except type
    
    * fix setup.py
    
    * put import extension inside trt setup
    
    * change c++ guard, update pytest script, better setup, etc
    
    * sort import with isort
    
    * sort import with isort
    
    * move init of plugin lib to init_plugins.py
    
    * add scatternd, nms plugin (WIP)
    
    * fix bugs of trt_nms
    
    * add trt nms test module
    
    * fix bugs of scatternd
    
    * code optimize, add comment about nms kernel
    
    * fix transform_if bug of trt_nms_kernel
    
    * fix struct name
    
    * default nms offset=0, fix bugs of batched input
    
    * format with clang-format
    
    * onnx preprocess
    
    * much better nms implementation, no need to transfer memory between host and device
    
    * update preprocess_onnx
    
    * parse constant tensor from initializer in preprocess_onnx
    
    * update nms
    
    * remove unnecessary codes
    
    * workspace aligned address
    
    * format trt_plugin_helper.hpp
    
    * fix index memory bugs
    
    * set alignment to 16 by default
    
    * fix lint
    
    * fix nms offset
    
    * fix bugs of preprocess onnx
    
    * update test for nms
    
    * tensorrt only accept int32, not int64
    
    * update nms comments
    
    * fix indexing for scores in nms
    
    * update trt temp
    
    * make trt-nms compatiable to #803
    
    * fix lint
    
    * add docstring to trt_nms_kernel.cuda, add description to preprocess_onnx
    
    * add comment to score indexing
    
    * fix bugs of max output boxes
    Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
    1ee5315e
test_tensorrt.py 9.9 KB