Commit Graph

700 Commits (db6b0542c7ca13edcbad0bb90687bf680a685844)
 

Author SHA1 Message Date
Y. Xiong 9e601ca6ca
Bump version to v1.2.6 (#799) 2021-01-21 14:34:49 +08:00
lizz 00658e8cf3
Fix copy source file path (#796) 2021-01-20 15:54:53 +08:00
q.yao 2ab544fc29
[Feature] : Add ScatterND TensorRT Plugin (#786)
* add scatter plugin

* fix bugs of scatternd

* add trt scatternd plugin

* format code with clang-format

* add test for scatternd

* skip test_tensorrt in CI

* remove unused variable

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2021-01-20 11:15:07 +08:00
BigBigDream 8e3a801596
fix mmcv ci for parrots (#782)
* fix mmcv ci for parrots

* fix mmcv ci

* fix lint
2021-01-14 14:31:57 +08:00
Rui Xu f169fb5279
[Feature] Support not to register lr scheduler hook (#774)
* support no lr scheduler hook

* fix lint
2021-01-09 15:11:46 +08:00
Qiaofei Li 3d6032c6db
Add dataset classes name info to meta for saving ckpt (#776)
* add CLASSES to meta info

* Update checkpoint.py

* add unit test for CLASSES name

* clean up the tmp folder

* use tempfile to clean up temp folder
2021-01-08 16:40:43 +08:00
congee 905c9b43b8
Add common testing function of MM repos (#743)
* add testing function

add unittest for check_dict

add unittest for the function in testing

* polish docstring of testing.py

rename some function

* remove  in is_all_zeros

* modify the comment of check_dict

* modify the testing.py according to feedback

* add test about numpy for function dict_contains_subset

* applying unified style
2021-01-08 13:18:08 +08:00
Zhiyuan Chen daab369e99
Add torch.no_grad() decorator to the whole val workflow(#777) 2021-01-08 13:16:49 +08:00
RunningLeon 477f0c0a39
[Fix]: Set installation of onnx simplify's related packages to optional (#779)
* Fix onnx, onnxruntime, onnxoptimizer import

* Revert "Fix onnx, onnxruntime, onnxoptimizer import"

This reverts commit 820e3785a1.

* remove simplify from init file

* update test for onnx simplify

* update onnx doc
2021-01-08 13:15:37 +08:00
LXXXXR 51a0c23f60
[Feature] Support load checkpoint from ceph (#778)
* support load checkpoint using ceph

* minor change
2021-01-08 13:08:16 +08:00
Rui Xu 276883f120
[Feature] Allow register multi-name for a module simultaneously (#775)
* allow register multi-name for a module simultaneously

* add assertion for name type

* use isintance intead of is_str

* fix bug in unit test

* fix unit test
2021-01-07 22:27:20 +08:00
q.yao 0de9e149c0
[Feature]: Add custom operators support for TensorRT in mmcv (#686)
* 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

* resolve format and add test dependency: tensorrt

* tensorrt should be installed from source not from pypi

* update naming style and input check

* resolve lint error

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2021-01-06 11:05:19 +08:00
Y. Xiong 643009e445
Bump version to v1.2.5 (#770) 2021-01-04 19:10:38 +08:00
Rui Xu 32bf3499bc
[Bug] Fix ddp bug when pytorch>=1.7 (#769)
* fix ddp bug when pytorch>=1.7

* fix ddp bug when pytorch>=1.7

* support pat

* fix docs

* use print_log instead of get_logger

Co-authored-by: nbei <631557085@qq.com>
2021-01-03 18:51:45 +08:00
Wenwei Zhang 080474b9a6
Fix typo in docs (#768)
* fix typo in docs

* add contents

* resolve typo
2021-01-02 14:52:13 +08:00
RunningLeon 83732ec2e3
[Feature] add onnx simplify (#751)
* [Enhance]: add onxx simplify

* add simple doc

* add unit test

* update docstring

* resolve some comment

* add test dependency:onnxoptimizer

* Fix onnxruntime register empty libpath

* test onnxruntime version

* set checker to false

* skip test_simplify for torch<1.5.0
2020-12-31 19:16:48 +08:00
tangyanf 741e52dd5e
[Feature]: add custom op roialign for onnxruntime (#724)
* add ort custom op roialign

* Update roi_align.py

* Update test_onnx.py

* Update test_onnx.py

* Update test_onnx.py

* Update test_onnx.py

* Update onnxruntime_register.cpp

* Update roiAlign.h

* Update roiAlign.cpp

* lint modification

* update roiAlign.cpp

* lint check

* lint check

* lint fix

* lint fix

* fix lint

* add link to commit

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2020-12-29 14:09:30 +08:00
Y. Xiong 4f0f1f90f9
bump version to v1.2.4 (#764) 2020-12-28 14:20:55 +08:00
Kai Chen b19492154e
remove the lagacy pip installation (#765) 2020-12-28 14:20:37 +08:00
Kai Chen c1acedcc3b
Remove pytest from the runtime dependencies (#763)
* update the CI to avoid extra dependencies

* remove the dependency of pytest

* fix typo

* remove skip_no_parrots from __init__
2020-12-28 13:23:48 +08:00
Y. Xiong 5d488e4a2a
bump version to 1.2.3 (#761) 2020-12-26 17:55:01 +08:00
BigBigDream 86c9ac2cb9
add 'iof' mode for box_iou_rotated (#753)
* add iof mode for box_iou_rotated

* update doc

* fix lint

* fix lint

* fix lint

* fix lint

* fix lint
2020-12-25 17:06:45 +08:00
Wenwei Zhang de767fc126
reformat test config to pass yapf (#759) 2020-12-25 16:55:02 +08:00
Y. Xiong edf9014c5b
Fix pip installation indexing bugs under pip 20.3 and follow PEP440 (#748)
* add block lines

* fix typo
2020-12-24 19:51:54 +08:00
BigBigDream 655f3c3fe9
fix mmcv_ci test_wrappers.py for parrots (#758) 2020-12-24 19:50:46 +08:00
yamengxi 86e0d62a6d
Add MobileNetV3 link (#757)
* add mobilenetv3 pretrained

* add name

* add name
2020-12-24 16:30:35 +08:00
Jerry Jiarui XU 555015ecc6
Add ResNet18_V1c link (#755) 2020-12-24 14:29:20 +08:00
q.yao 94810f2297
[Feature]: Add custom operators support for onnxruntime in mmcv (#612)
* add onnx support to roi_align and roi_pool

* add softnms ort support

* fix for lint

* format cpp code with clang-format:google

* add new empty line to the end of head files in onnxruntime

* update to pytorch1.7

* add test of softnms to onnxruntime

* fix for lint

* remote print in ops/info.py

* change import order, fix for flake8

* fix include

* add assert torch>=1.7.0

* [doc]: add document for onnxruntime custom operator

* update onnxruntime version to v1.5.1 for softnms

* remove doc menu

* Resolve lint for markdown

* resolve naming style in onnxruntime_op.md

* Use old cpp apis, optimize test_onnx.py

* Fixing strings in tests/test_ops/test_onnx.py

* code format with yapf

* fix soft_nms parrot

* add import in onnxruntime setup, avoid conflict

* fix doc and add assert

* change cpp guard

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2020-12-23 11:03:55 +08:00
Y. Xiong 8b4e5de43d
Add nms compile issue to trouble_shooting page (#733)
* Add nms compile issue to trouble_shooting page

* after lint

* update link to online doc
2020-12-22 17:02:42 +08:00
LXXXXR 09661d9754
[Feature] support upload and download checkpoint from pavimodelcloud (#725)
* support upload and download checkpoint from pavimodelcloud

* try import pavi

* add unittest
2020-12-22 12:59:07 +08:00
Rui Xu 826d3a7b68
[Bug] Avoid recursion error when using customized norm layer without _abbr_ (#732)
* avoid recursion error when using customized norm layer without _abbr_

* fix unittest bug
2020-12-22 10:19:52 +08:00
LXXXXR adc59b9f28
fix links in mmcls.json (#745) 2020-12-22 10:19:08 +08:00
shilong ae982b2806
fix grad_output uncontiguous (#737)
* fix grad_output uncontiguous

* add comments for uncontiguous
2020-12-21 17:21:20 +08:00
hiyyg dedd79940e
add more mmcls pretrained models (#741) 2020-12-21 17:20:11 +08:00
Wang Xinjiang 96ebfa652b
Allow to replace nested tuple and list via options (#731)
* Allow to replace nested tuple and list via options

* Add comments

* Fix single nested items

* Simplify the code

* Simplify the code

* Simplify the code

* Simplify the code

* Update docstring

* Update docstring

* Support quotation mark

* modify docstring
2020-12-20 21:28:48 +08:00
WRH f0e68404d2
Fix some bug as well as unit test for nms_rotate (#728)
* add const to nms_rorate in pybind

* fix test nms rotated

* skip test instead of passing it

* fix lint

* update pytest skip syntax
2020-12-18 15:17:11 +08:00
nicheng0019 8eae77793f
Update checkpoint.py (#722) 2020-12-18 14:01:47 +08:00
Jintao Lin 3392a4ea00
fix the comment of init_weights in ``ConvModule`` (#730)
* polish the comment of init_weights in ConvModule

* polish the comment

* polish the comment
2020-12-18 14:00:47 +08:00
Wenwei Zhang 95acffb910
support mpi launcher (#726) 2020-12-18 10:11:10 +08:00
BigBigDream 508a322fba
Support aligned mode for box_iou_rotated (#677)
* support aligned and parrots cpu for  box_iou_roatetd

* add aligned doc

* fix lint

* fix lint

* fix lint

* fix lint

* fix bug

* fix bug

* fix bug

* fix lint

* fix lint

* fix bug

* fix bug
2020-12-16 11:49:44 +08:00
lizz 8008f475ab
Support mmcv bricks in flops compuation (#715)
Signed-off-by: lizz <lizz@sensetime.com>
2020-12-15 18:23:52 +08:00
lml131 9befc39811
Lml/jit decorator (#673)
* add jit decorator

* add parrots_jit.py

* modify test_parrots_jit.py

* modify for lint

* fix isort

* skip test_parrots_jit.py when build without pytorch

* try ci

* rm log

* fix double quote

* modify for comments and use partial_shape instead of full_shape

* fix for lint

* small modify for parrots 0.9.0rc0

* def skip no elena directly
2020-12-15 18:20:37 +08:00
Y. Xiong e30dc4f88a
bump version to v1.2.2 (#714)
* bump version to 1.2.1

* update installation table in readme

* break new line

* del mmcv upper bound for old installation

* reminder on latest

* bump version to v1.2.2
2020-12-14 22:35:11 +08:00
BigBigDream 02920db2cc
fix roi_align ci for parrots (#708)
* fix roi_align ci for parrots

* fix lint
2020-12-13 20:01:25 +08:00
Rui Xu b7136e3953
[Refactor] remove the c implementation of flow_warp (#707)
* remove flow-warp-c

* remove flow warp in setup.py and imports

* fix floor and ceil bug

* fix broadcast bug

* add warnings and assertions

* fix bilinear bug

* pass unit test

* pass unit test

* pass unit test

* pass unit test

* fix value bug

* fix lint

* fix lint

* add mk lint

* update docs

* update docs

* fix bug in imports

* fix bug in setup.py
2020-12-13 19:43:25 +08:00
mohammad minooei 1a4c0240dc
check MASTER_ADDR in the environment variable before assignment (#705)
* check MASTER_ADDR in the environment variable before assignment

* Update dist_utils.py

use MASTER_ADDR in the environment variable if it already exists

Co-authored-by: mohammad <minooee@vas24.ir>
2020-12-12 19:17:45 +08:00
Kai Chen afc35f13ff
relax the error bound in colorspace conversion (#706) 2020-12-12 19:16:11 +08:00
Y. Xiong cff01c5f55
Fix pip 20.3 (#703)
* bump version to 1.2.1

* update installation table in readme

* break new line

* del mmcv upper bound for old installation

* reminder on latest
2020-12-12 11:32:40 +08:00
Wenwei Zhang d5241b7087
Allow the initial value to be set in dynamic mode (#692)
* allow the initial value to be set in dynamic mode

* add logging for grad norm back

* resolve comments
2020-12-09 22:34:37 +08:00
Wenwei Zhang a0902e6de8
allow non-master build of parrots (#702) 2020-12-09 21:28:06 +08:00