Commit Graph

576 Commits (655f3c3fe9728b74e0b052f96c8ea870999b7579)
 

Author SHA1 Message Date
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
Rui Xu cd96a84add
[Feature] Add clamp activation layer. (#685)
* add clamp without unittest

* add clamp-act with unit test

* fix name bug

* use logical and

* fix logical_and

* fix linting

* rename ClampLayer to Clamp

* rename ClampLayer to Clamp

Co-authored-by: nbei <631557085@qq.com>
2020-12-09 10:28:32 +08:00
Wenwei Zhang 1e925a05a2
Skip pat build in PR (#699)
* skip pat build in PR

* rename
2020-12-08 19:19:35 +08:00
Wang Xinjiang c9f96855b0
Allow list index keys in Config.merge_from_dict (#696)
* Allow list keys in Config merge from dict

* Reformat

* Set allow_list_keys default as True in merge_from_dict

* Fix docstring

* fix a small typo
2020-12-08 17:17:27 +08:00
Jintao Lin eaf25af6c4
use CUDA_HOME from parrots_wrapper (#678) 2020-12-01 10:37:12 +08:00
WRH d2ca6cebb3
[DOC] build on windows (#663)
* previous doc

* init version of windows installation

* refine doc

* merge build instructions

* fix styles

* add some detail

* change an item to note

Co-authored-by: Kai Chen <chenkaidev@gmail.com>
2020-12-01 10:36:20 +08:00
Y. Xiong 91a7fee03a
bump version to 1.2.1 (#672) 2020-11-26 15:58:37 +08:00
Wang Xinjiang 0d562118b5
Fix rotated nms (#668) 2020-11-25 14:13:26 +08:00
BigBigDream a978764883
Add box_iou_rotated, ml_nms_rotated and nms_rotated (#625)
* add box_iou_rotated, ml_nms_rotated and nms_rotated

* fix lint

* fix lint

* fix .py lint

* fix cpp lint

* add newline at the end

* add new line

* fix unittest

* config google style

* fix lint

* lint

* lint

* yapf

* update

* fix lint

* fix lint

* fix lint

* fix

* fix format

* fix format

* add modified from

* add docstring and update others

* update docstring

* update docstring

* update

* fix bug

* fix bug

* fix bug

Co-authored-by: Cao Yuhang <yhcao6@gmail.com>
2020-11-25 11:00:30 +08:00
RunningLeon f61bb642e5
[Enhance]: Better support for pytorch2onnx for detection models (#648)
* [feature]: Support convertion from Pytorch to ONNX for detection models

* [fix] RuntimeError with integer divide when converting to onnx

* Resolve pylint error

* add comment
2020-11-24 20:04:00 +08:00
wang shiguang 23ecd83d59
Add parrots ci (#662)
* add parrots ci

* update with ops build

* Update build.yml

* cridents env

* Update build.yml
2020-11-24 15:10:24 +08:00
Wenwei Zhang d2fa1abc53
Add documentations for registry (#656)
* add documentations for registry

* update example

* update sphinx

* add intro

* resolve comments
2020-11-23 15:29:22 +08:00
Jerry Jiarui XU 987cb58341
[Enhance] Refactor logger (#659)
* [Enhance] Refactor logger

* fixed test

* make commit optional

* remove debug info

* fixed test
2020-11-23 10:50:18 +08:00
Wenwei Zhang dfa36dfe3b
Merge pull request #652 from dreamerlin/3d
[Feature] Add 3D support in wrapper
2020-11-20 16:33:54 +08:00
yamengxi ec43b671ab
[Enhance]enhance hsigmoid (#657)
* enhance hsigmoid

* delete int
2020-11-19 21:26:26 +08:00
dreamerlin 1a12ac753a fix 2020-11-18 23:03:49 +08:00
dreamerlin e70dca8f86 add conv3d 2020-11-18 21:24:20 +08:00
dreamerlin 144e7567a7 use pytest.mark.parametrize 2020-11-18 15:35:21 +08:00
dreamerlin 86d9f4684a add MaxPool3d 2020-11-15 19:07:46 +08:00
dreamerlin 8ccea20234 add ConvTranspose3d 2020-11-15 18:56:02 +08:00
Jintao Lin 1290bdd181
update docstring (#651) 2020-11-15 18:48:03 +08:00
q.yao c390e327fa
add onnx support to roi_align and roi_pool (#593)
* add onnx support to roi_align and roi_pool

* add test on roi_align and roi_pool

* fix conflict with master
2020-11-14 19:56:35 +08:00
Jiang Yinzuo b3e5a74d22
update README.md for torch 1.7 installation (#642)
* update README.md for torch 1.7 installation

* add cuda 11.0

Co-authored-by: Kai Chen <chenkaidev@gmail.com>
2020-11-14 12:05:57 +08:00
yamengxi c6c230df1b
[Feature] Add CLAHE method (#647)
* add CLAHE

* add CLAHE

* restore

* Add docstring

* modify docstring

* modify CLAHE to clahe

* fix syntax error

* simplify assert

* simplify assert

* add assert test

* fix unittest bug

* fix syntax bug

* fix assert bug
2020-11-11 22:34:14 +08:00
yamengxi d9ef9dabe2
[Feature] Add LUT transform (#646)
* add lut_transform

* restore

* fix test bug

* add test

* Fix syntax bug

* Delete test id

* modfiy docstrinng

* simplify assert
2020-11-11 17:32:41 +08:00