Commit Graph

95 Commits (c85c240fd8fed8c212157589904b078f01074be1)

Author SHA1 Message Date
Zaida Zhou c85c240fd8
[Docs] Fix error when building pdf (#1414)
* [Docs] Fix error when building pdf

* fix link
2021-10-20 14:33:40 +08:00
dingchang 5c25ae1a19
[Feature] Add roipoint_pool3d op from mmdet3d (#1358)
* add ops (roipoint_pool3d) in mmdet3d

* refactor code

* fix typo

* add unit test

* refactor code

* refactor code

* refactor code

* fix typo
2021-10-15 17:15:53 +08:00
dingchang e99bfce608
[Feature] Add assign_score_withk op for paconv from mmdet3d (#1357)
* add ops (assign_score_withk) in mmdet3d

* add ops (assign_score_withk) in mmdet3d

* refactor code

* fix typo

* refactor code

* fix typo

* polish docstring

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-10-15 13:18:31 +08:00
dingchang be5841e45d
[Feature] Add interpolate ops from mmdet3d (#1355)
* add ops (interpolate) in mmdet3d

* refactor code

* fix typo

* fix typo

* fix typo

* refactor code
2021-10-14 20:51:02 +08:00
dingchang 1cd01db935
[Feature] Add gather points op from mmdet3d (#1338)
* add ops (gather points) in mmdet3d

* add ops (gather points) in mmdet3d

* refactor code

* refactor code

* fix typo
2021-10-14 16:58:12 +08:00
dingchang 42c9e71120
[Feature] Add knn op from mmdet3d (#1354)
* add ops (knn) in mmdet3d

* refactor code

* refactor code

* fix typo

* fix typo

* fix typo

* refactor code

* refactor code

* spell typo

* fix spell typo
2021-10-14 13:33:13 +08:00
dingchang 63a6cbe913
[Feature] Add fps op from mmdet3d (#1337)
* add ops (furthest point sample) in mmdet3d

* refactor code

* refactor code

* fix typo

* fix typo

* fix typo

* refactor code

* fix typo

* define DIVUP in common_cuda_helper.hpp
2021-10-13 20:39:31 +08:00
Eugene Liu 599163e645
Add Deformable Conv CustomOp for onnxruntime (#1343)
* add onnx dcn

* replace gemm with torch C++ api

* small update

* fix cpp clang format

* pefer generic GEMM than torch ATen library

* addressing comments

* add ops path check
2021-10-13 20:28:24 +08:00
Zaida Zhou 48d5b095a2
Polish readme (#1320)
* [Docs] Polish README

* [Docs] Polish README

* [Docs] Polish README

* fix typo

* update qq link

* fix typo

* update LICENSES.md
2021-10-13 20:25:49 +08:00
Zaida Zhou f22c9eb4a4
Add code spell hook to pre-commit and fix typos (#1384)
* Add code-spell hook to pre-commit

* Add code-spell hook to pre-commit and fix typos

* fix setup.cfg
2021-10-04 20:13:54 +08:00
Zaida Zhou f69a3a44e3
[Docs] Translate building documentation (#1353)
* [Docs] Translate building documentation

* polish translation
2021-09-24 17:02:59 +08:00
q.yao afe0794ce8
[Docs] Add docs_zh_CN of TensorRT (#1336)
* Add docs_zh_CN of TensorRT

* fix according to doc-cn-onnx

* update doc of cummax cummin

* Apply suggestions from code review

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* update heading

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-09-24 17:00:35 +08:00
q.yao 2b39d7a8ec
[Docs] Add zh_cn document of ONNX (#1331)
* add doc-cn of ONNX

* Update docs_zh_CN/deployment/onnxruntime_custom_ops.md

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Apply suggestions from code review

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* update doc of cummax

* fix en doc of softnms

* update heading

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-09-24 16:57:20 +08:00
Ty Feng 6f533ff113
[Docs] cnn.md translated (#1246)
* cnn.md translated

* polish translation

* polish translation

Co-authored-by: zhouzaida <zhouzaida@163.com>
2021-09-24 16:54:32 +08:00
Miao Zheng b92ea0b5df
[Feature] Add Correlation CUDA op (#1361) 2021-09-23 21:08:41 +08:00
dingchang f3dfc4135b
[Feature] Add ballquery op from mmdet3d (#1332) 2021-09-23 21:08:00 +08:00
Zaida Zhou 1216e5fe7f
[Docs] Use PyTorch sphinx theme (#1321)
* [Docs] Use PyTorch sphinx theme

* add sphinx-copybutton

* add twitter and zhihu link

* remove shpinx_rtd_theme from doc.xtx

* update docs.txt

* update conf.py
2021-09-14 16:35:41 +08:00
Wei Ji 9016fef560
Fix typo on pip install mmcv-full torch1.9 cpu (#1313)
* Fix typo on pip install mmcv-full torch1.9 cpu

* 修改 pip install mmcv-full torch1.9 cpu 的错别字

* Fix typos in installation.md

Point to https://download.openmmlab.com/mmcv/dist/cpu/torch1.9.0/index.html
2021-09-07 17:32:11 +08:00
q.yao 8aab4f2595
[Fix] Update requirement of ONNX Runtime (#1290)
* update requirements of onnxruntime

* update document

* use python setup.py instead of pip install
2021-08-23 23:31:23 +08:00
Haian Huang(深度眸) f022d57702
Add Chinese Documentation of runner (#1202)
* Add Chinese Documentation of runner

* Fix error

* Fix lint

* update

* update comment
2021-08-23 14:25:24 +08:00
Paige Wang 1790e9f265
add modulated_deform_conv in onnxruntime support (#1281)
* add modulated_deform_conv in onnxruntime support

* Add docs descriptions

* Add gpu test in test_onnx.py

* code format

* remove new usage and move if outside for loop

* use memset when bias is nullptr
2021-08-23 11:22:24 +08:00
jiang1997 9f0cc5a6e4
[WIP] Apply to translate "docs_zh_CN/understand_mmcv/utils.md" (#1232)
* Copy utils.md from docs

* Add Chinese translation of docs_zh_CN/understand_mmcv/utils.md

* recommit with pre-commit hook

* polish

* Update docs_zh_CN/understand_mmcv/utils.md

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update docs_zh_CN/understand_mmcv/utils.md

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* fix path of image

* set priority when build html

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zhouzaida <zhouzaida@163.com>
2021-08-23 11:12:17 +08:00
Yining Li 4d9b43debd
Support deprecation info in Config (#1275)
* Support deprecation checking in Config

* add unittest for config deprecation

* support reference link in deprecation info

* add doc

* Update config.md
2021-08-23 10:36:37 +08:00
Zaida Zhou aa64ee6a7c
[Docs] Use myst parser (#1287)
* add myst-parser extension for sphinx

* make note style more beautiful
2021-08-22 09:44:51 +08:00
Zaida Zhou 9c58d9e750
Fix typo (#1280)
* fix typo

* fix typo
2021-08-19 10:04:54 +08:00
Zaida Zhou 9679ce8c99
[Docs] Remove the _static in docs_zh_CN (#1277)
* [Docs] Remove _static in docs_zh_CN

* fix html_static_path

* fix shpinx warning
2021-08-17 21:16:18 +08:00
Zaida Zhou 8d2d781225
[Docs] Fix pdf build (#1266)
* [Fix] Fix pdf error when building docs

* fix docs

* fix pdf

* fix pdf

* fix pdf
2021-08-17 16:03:55 +08:00
Zaida Zhou dfb48c87ae
[Refactor] Refactor the directory of csrc (#1206)
* [Refactor] Refactor the csrc directory

* update MANIFEST.in

* fix hip

* add csrc readme

* trailing whitespace

* fix syntax error in setup.py

* add compatibility docs

* move parrots_cudawarpfunction.cuh to common/cuda

* fix grammar, update directory tree

* fix MANIFEST.in

* Add new structre of csrc in compatibility.md

* Add original structre of csrc in compatibility.md

* fix typo

* remove TODO

* modify according to comment

* format

Co-authored-by: grimoire <yaoqian@sensetime.com>
2021-08-10 15:09:19 +08:00
Miao Zheng 94a677de3f
[Docs] Add PR document (#1209)
* [Docs] Add PR document

* format

* idx modify

* revise according to comments

* add blank line

* grammar

* insert blank line after title

* revise according to comments

* revise according to comments

* typo

* resize image3

* lint

* del print

* fix lint

* del unused

Co-authored-by: zhouzaida <zhouzaida@163.com>
2021-08-09 22:29:21 +08:00
Zaida Zhou 3040299bb5
[Docs] Update readme and docs (#1190)
* [Docs] Update readme and docs

* add Chinese introduction

* fix table error
2021-07-20 17:18:28 +08:00
Zaida Zhou db580dda8d
[Docs] Build Chinese docs (#1073)
* Remove _build directroy in docs_zh_CN

* Change utils title to Chinese

* Translate documents

* Translate documents

* Use symbolic link to avoid repeated images

* Use symbolic link to avoid repeated images

* fix readme.md

* update copyright

* refactor docs

* rename title to Chinese
2021-07-07 13:10:04 +08:00
Zaida Zhou 66cefaffa3
[Docs] Refactor docs (#1102)
* [Docs] Refactor documentation

* [Docs] Refactor documentation

* refactor docs

* refactor docs

* set sphinx==3.1.2

* fix typo

* modify according to comment

* modify according to comment

* modify according to comment

* [Docs] delete unnecessary file

* fix title

* rename

* rename
2021-07-02 20:53:00 +08:00
Zaida Zhou 76d9bf1efb
[Docs] Fix error when cv2 is mocked (#1152) 2021-06-28 17:31:36 +08:00
lizz d9effbd1d0
Support variables in base files for configs (#1083)
* Support variables in base files for configs

Signed-off-by: lizz <lizz@sensetime.com>

* Test json and yaml as well

Signed-off-by: lizz <lizz@sensetime.com>

* Add test for recusive base

Signed-off-by: lizz <lizz@sensetime.com>

* Test misleading values

Signed-off-by: lizz <lizz@sensetime.com>

* Improve comments

Signed-off-by: lizz <lizz@sensetime.com>

* Add doc

Signed-off-by: lizz <lizz@sensetime.com>

* Improve doc

Signed-off-by: lizz <lizz@sensetime.com>

* More tests

Signed-off-by: lizz <lizz@sensetime.com>

* Harder test case

Signed-off-by: lizz <lizz@sensetime.com>

* use BASE_KEY instead of base

Signed-off-by: lizz <lizz@sensetime.com>
2021-06-25 14:30:58 +08:00
q.yao 004c00675f
[Feature]: add modulated deformable conv TensorRT support (#1078)
* add modulated dcn, better dcn plugin

* clangformat

* update documentation
2021-06-16 21:38:07 +08:00
LXXXXR c9b009f1ba
fix typo (#1094) 2021-06-11 20:08:08 +08:00
LXXXXR 1076958cd4
[Docs] Add runner tutorial (#1082)
* add runner doc

* revised according to comments and remove non existing links

* revise according to comments
2021-06-09 22:41:02 +08:00
SemyonBevzuk bf2c9fa8d2
[Feature] NMS update (#957)
* Add score_threshold and max_num to NMS

* Fix codestyle

* Fix codestyle

* Fix inds in nms

* Update nms docstring

* Move score_threshold and max_num arguments

* Fix args order in docstring

* fix lint of c++ file

* Remove torch.onnx.is_in_onnx_export() and add max_num to batched_nms for separate classes.

* Rewrote max_num handling in NMSop.symbolic

* Added processing max_output_boxes_per_class when exporting to TensorRT

* Added score_threshold and max_num for NMS in test_onnx.py and test_tensorrt.py

* Remove _is_value(max_num)

* fix ci errors with torch==1.3.1

* Update test_batched_nms in test_nms.py

* Added tests for preprocess_onnx

* Moved 'test_tensorrt_preprocess.py' and 'preprocess', updated 'remove_tmp_file'.

* Update mmcv/tensorrt/__init__.py

* Fix segfault torch==1.3.1 (remove onnx.checker.check_model)

* Returned 'onnx.checker.check_model' with torch version check

* Changed torch version from 1.3.1 to 1.4.0

* update version check

* remove check for onnx

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2021-05-31 23:42:26 +08:00
RunningLeon 4d42365ac2
[Feature]: add TensorRT InstanceNormalization plugin (#1034)
* add instancenorm plugin

* resolve comments

* fix lint

* fix typo
2021-05-25 20:03:57 +08:00
lizz 13888df2aa
Fix typos (#1041)
* Fix typos

Signed-off-by: lizz <lizz@sensetime.com>

* Add deprecation warning

Signed-off-by: lizz <lizz@sensetime.com>
2021-05-24 14:51:45 +08:00
q.yao 9d1436fb6c
[Feature] add cummax/cummin tensorrt plugin (#1031)
* add cummax/cummin tensorrt plugin

* fix isort

* fix with clang-format

* fix with clang-format again

* add document
2021-05-24 13:55:21 +08:00
Zaida Zhou a1d3bf1c80
[Docs] Update weight initialization in cnn.md (#912)
* [Docs] Update weight initialization in cnn.md

* Update cnn.md

* [Docs] Update weight initialization in cnn.md

* [Docs] Update weight initialization in cnn.md

* [Docs] Update weight initialization in cnn.md

* [Docs] Update weight initialization in cnn.md

* [Docs] Update weight initialization in cnn.md

* modify docs according to comments

* modify docs for weight init according to the comment

* change init_weight() to init_weights()

* modify according to comment
2021-05-13 20:29:36 +08:00
v-qjqs 934b549e23
Support custom operators cummax and cummin for onnxruntime (#1010)
* support custom op `mmcv::cummax` for onnxruntime in mmcv

* fix clang-format lint error

* support mmcv::cummin, reformat codes

* fix merge from master

* add docs for mmcv::cummax and mmcv::cummin

* format doc

* add assertion for torch version, when exporting `cummax` to onnx

* add more comments for torch version

* handle exporting to onnx in `soft_nms`

* commit for test_onnx

* remove `is_in_onnx_export` in softnms

* add more comments

* fix c++ lint error

* add known issues doc for `cummax`

* fix known issues doc
2021-05-10 21:33:27 +08:00
v-qjqs db6b0542c7
[Feature]: Support corner_pool related custom operators for onnxruntime in mmcv (#997)
* supports for onnxruntime custom op `mmcv::MMCVTopPool`

* supports for onnxruntime custom op `mmcv::MMCVCornerPool`, involving TopPool, BottomPool, LeftPool and RightPool

* add unittest for corner_pool

* supports mmcv::CornerPool without memcpy

* add docs for mmcv::CornerPool

* re-add docs for mmcv::CornerPool

* fix output dtype doc

* reformat

* format with pre-commit

* format

* fix lint error, by using google clang-format style for c/c++
2021-05-01 16:28:25 +08:00
zhouzaida ae411509dd
[Docs] Add QR code in CN README (#954) 2021-04-16 21:22:51 +08:00
q.yao e478e9ff7f
[Doc]: Add document of TensorRT and onnxruntime custom ops (#920)
* update tensorrt plugin document

* add onnxruntime custom ops document

* format document

* add release note to onnxruntime_op and tensorrt_plugin

* update document

* add deployment.rst

* add grid_sampler onnxruntime document

* fix lint

* add allow_different_nesting tag

* add allow_different_nesting tag

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2021-04-12 14:06:28 +08:00
Jerry Jiarui XU 375605fba8
add model registry (#760)
* add model registry

* fixed infer scoep

* fixed build func

* add docstring

* add md

* support multi level

* clean comments

* add docs

* fixed parent

* add more doc

* add value error, add docstring

* fixed docs

* change to local/global search

* resolve comments

* fixed test

* update some docstring

* update docs (minior)

* update docs

* update docs
2021-04-10 21:33:34 +08:00
QingChuanWS b5e1facc85
[Refactor]: Remove deprecated onnx-simplify source code (#927)
* remove custom simplify model

* update docs about onnx-simplify
2021-04-09 16:17:11 +08:00
Jintao Lin be6541d480
[Feature] Add EvalHook which will be used in downstream projects (#739)
* add EvalHook

* add EvalHook

* refactor docstring

* polish

* use cp instead symlink

* add error info

* Update eval.py

* Update eval.py

* update

* add engine depandancy

* add comments

* fix unittest

* fix

* update unittest

* update unittest

* add docstring

* fix docstring
2021-04-08 16:42:30 +08:00
ftbabi 8735815a0d
Add GELU activation function (#843)
* Support GELU activation function

* Support different torch version

* Fix bug in importing TORCH_VERSION

* Wrap GELU module for PyTorch 1.3.x

* Wrap GELU module for PyTorch 1.3.x
2021-02-22 10:55:42 +08:00