Async inference interface (#1647)
* async inference support * implemented concurrent decorator * fixes for Python versions < 3.7 * async methods depend on python version * revert changes in forward method * async_test -> async_simple_test, debug logging is done via logger.debug * add async test * add asynctest to test requirements * async tests are run in Python 3.7 * check CUDA, add docs * fix device * run test only if CUDA is available * fix linting * custom operators can run on nondefault stream * set current stream in kernel launch configuration * example fixes * add async/sync interface comparison benchmark * fix linting
Showing
- docs/GETTING_STARTED.md 43 additions, 0 deletionsdocs/GETTING_STARTED.md
- mmdet/apis/__init__.py 5 additions, 4 deletionsmmdet/apis/__init__.py
- mmdet/apis/inference.py 27 additions, 0 deletionsmmdet/apis/inference.py
- mmdet/models/anchor_heads/rpn_head.py 1 addition, 1 deletionmmdet/models/anchor_heads/rpn_head.py
- mmdet/models/detectors/base.py 24 additions, 0 deletionsmmdet/models/detectors/base.py
- mmdet/models/detectors/test_mixins.py 97 additions, 0 deletionsmmdet/models/detectors/test_mixins.py
- mmdet/models/detectors/two_stage.py 37 additions, 2 deletionsmmdet/models/detectors/two_stage.py
- mmdet/ops/dcn/src/deform_conv_cuda_kernel.cu 7 additions, 6 deletionsmmdet/ops/dcn/src/deform_conv_cuda_kernel.cu
- mmdet/ops/dcn/src/deform_pool_cuda_kernel.cu 3 additions, 3 deletionsmmdet/ops/dcn/src/deform_pool_cuda_kernel.cu
- mmdet/ops/masked_conv/src/masked_conv2d_kernel.cu 4 additions, 2 deletionsmmdet/ops/masked_conv/src/masked_conv2d_kernel.cu
- mmdet/ops/nms/src/nms_kernel.cu 8 additions, 5 deletionsmmdet/ops/nms/src/nms_kernel.cu
- mmdet/ops/roi_align/src/roi_align_cuda.cpp 2 additions, 0 deletionsmmdet/ops/roi_align/src/roi_align_cuda.cpp
- mmdet/ops/roi_align/src/roi_align_kernel.cu 3 additions, 2 deletionsmmdet/ops/roi_align/src/roi_align_kernel.cu
- mmdet/ops/roi_pool/src/roi_pool_kernel.cu 3 additions, 2 deletionsmmdet/ops/roi_pool/src/roi_pool_kernel.cu
- mmdet/ops/sigmoid_focal_loss/src/sigmoid_focal_loss_cuda.cu 2 additions, 2 deletionsmmdet/ops/sigmoid_focal_loss/src/sigmoid_focal_loss_cuda.cu
- mmdet/utils/contextmanagers.py 122 additions, 0 deletionsmmdet/utils/contextmanagers.py
- mmdet/utils/profiling.py 41 additions, 0 deletionsmmdet/utils/profiling.py
- setup.py 1 addition, 1 deletionsetup.py
- tests/async_benchmark.py 104 additions, 0 deletionstests/async_benchmark.py
- tests/requirements.txt 1 addition, 0 deletionstests/requirements.txt
Loading
Please register or sign in to comment