cb0ee1e339
Remove redundant self.init_cfg ( #959 )
2021-04-17 10:06:44 +08:00
ae411509dd
[Docs] Add QR code in CN README ( #954 )
2021-04-16 21:22:51 +08:00
1bc912eb57
Bump version to v1.3.1 ( #950 )
2021-04-16 10:53:27 +08:00
54ece10ffb
[Docs] Add mmocr link ( #941 )
...
* [Docs] Add mmocr link
* Update README.md
* Update README_zh-CN.md
2021-04-14 20:07:07 +08:00
47d29795ff
[Fix] Fix train error EpochBasedRunner ( #942 )
2021-04-14 20:00:20 +08:00
8d7e0a6d6a
[Refactor]: add init_cfg in transformer base classes ( #946 )
2021-04-14 18:04:08 +08:00
79f8cbd661
[Bug] Fix DeformConv2d bias error and add tests ( #940 )
...
* [Bug] Fix DeformConv2d bias error and add tests
* fix repr
* revise tests
* lint
2021-04-12 20:31:34 +08:00
3bcc796d38
[Fix]: Fix cuda compile error on visual studio 16.9 ( #891 )
...
* replace floor/ceil to floorf/ceilf
to eliminate cuda compilation errors in the latest version of Visual Studio 16.9
* fix lint error
* fix lint
* fix lint
* Update .pre-commit-config.yaml
* Update .pre-commit-config.yaml
Co-authored-by: wangruohui <12756472+wangruohui@users.noreply.github.com>
2021-04-12 14:08:21 +08:00
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
2fadb1a5ec
[Fix] Revise override in init_cfg ( #930 )
...
* [Fix] Config deep copy in initialize_override
* add asserts&comments
* add test
* test org init_cfg
* test override without name
* typo
2021-04-12 14:05:28 +08:00
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
47825b194e
[Refactoring] Revise init_weight in BaseModule ( #905 )
...
* [Refactoring] Add deprecated API warning
* revise test
* fix lint
* fix lint
2021-04-10 21:20:53 +08:00
89efc607ef
[Fix] OneCycleLrUpdaterHook adds total_steps ( #929 )
...
* [Fix] OneCycleLrUpdaterHook interface
* revise according to comments
* revise according to comments
* add test
* fix lint
* revise according to comments
* minors
* add pytest param
* fix lint
* ci
2021-04-10 21:19:45 +08:00
f959a34db9
[Docs] Docstring for DeformConv ( #921 )
...
* [Docs] Docstring for DeformConv
* fix docstring
* fix according to comments
* revise according to comments
* lint
2021-04-10 21:19:06 +08:00
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
d636257e0d
[Fix] Fix wandb logger drop result bug ( #913 )
...
* fix wandb logger drop result bug by delete step param
* add global_step in wandb log to help align train and val step log
* fix wandb hook test unit fail bug
* fix lint issue
* add with_step param of WandbLoggerHook in wandb.py
2021-04-09 12:57:49 +08:00
d525cfde10
[Fix] Fix bug of lr updater hook ( #907 )
...
* [Fix] fix bug of lr update hook
* [Fix] fix bug of lr update hook
* [Fix] Fix bug of lr updater hook
2021-04-09 10:13:32 +08:00
03a2e3a12a
Fix onnx interpolate conversion ( #917 )
...
* fix onnx interpolate recommit
* fix bugs on torch==1.6.0
* remove print
* fix error in torch==1.5.1
2021-04-08 18:57:46 +08:00
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
b591eb5005
[Fix] Fix sphinx build error ( #931 )
2021-04-08 16:26:29 +08:00
9de39b1e94
Fix typo in build_conv_layer docstring ( #928 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-07 22:56:37 +08:00
83cf25b284
[feature]:add onnxruntime custom op grid_sample ( #916 )
...
* add onnxruntime custom op grid_sample
* update code
* update code
* update code
* update code
* update code
* update code
* update code
* update code
* update code
* update code
* update code
* update code
* update code
* update code
2021-04-06 18:31:24 +08:00
f0c43fdc32
[Feature] Add OneCycleLrUpdaterHook ( #906 )
...
* [Feature] Add OneCycleLrUpdaterHook
* fix docstring
* fix docstring
* Remove redundant code
2021-04-02 09:44:18 +08:00
3ae1b257fa
fix initialization bug of override cfg ( #918 )
2021-04-02 09:43:16 +08:00
9d80f56ae8
[Feature] : Add grid_sampler TensorRT Plugin ( #865 )
...
* add grid sample trt support
* fix align_corners=True
* change 'intep' to 'interp', fix bugs of grid sampler, better test script
* remove unused import
* move source comment to the top of trt_grid_sampler_kernel.cu
2021-03-31 00:28:08 +08:00
03491ed28a
[Enhancement]: add more detail of warning in basemodule ( #899 )
2021-03-29 11:37:03 +08:00
75d907c9d1
Bump version to v1.3.0 ( #910 )
...
* Bump version to v1.3.0
* Update README.md
* Update README_zh-CN.md
2021-03-28 14:04:24 +08:00
8647b9a1b1
[Feature] Add cutout and lighting ( #909 )
...
* add cutout
* add adjust_lighting
* minor fix
2021-03-26 18:58:19 +08:00
65fec73509
[Refactoring] Revise function of layers and override keys in init_cfg ( #893 )
...
* [Refactoring]Approve init_cfg
* revise unit test
* revise according to comments
* add comment for wholemodule
2021-03-26 16:57:59 +08:00
a9803da10b
[Feature]: Add transformer base class ( #892 )
...
* add transformer base class
* fix according to commments
* resiger module
* change to use basemodule
* add _ between build function
* split the name
2021-03-25 21:31:50 +08:00
5f5e8e83c2
[Refactoring] Add Caffe2Xavier Initializer ( #902 )
...
* [Refactoring] Add Caffe2Xavier Initializer
* fix lint
2021-03-24 13:25:36 +08:00
933b052d95
[Feature] Add cuda ops: UpFirDn2d and fused_bias_leakyrelu ( #900 )
...
* add upfirdn2d op
* fix bug in pybind
* add fused bias leakyrelu
* fix bug in fused-bias-leakyrelu
* fix lint error
* fix bug in build cpu version
* fix bug in build cpu version
* fix lint
* fix comment from zww
Co-authored-by: zhangshilong <zhangshilong@sensetime.com>
2021-03-21 21:23:09 +08:00
371a21728f
fix bug of convmodule ( #889 )
...
* fix bug of convmodule
* fix bug of convmodule
* fix unitest
* remove assert
2021-03-20 23:08:20 +08:00
97730c2316
[Feature] support to calculate FLOPs of GN, IN, LN ( #897 )
...
* [Feature] support to calculate FLOPs of GN, IN, LN
* Update flops_counter.py
* Update flops_counter.py
2021-03-19 14:04:49 +08:00
00870b9c4e
[Refactoring] Revise function of is_init flag in BaseModule ( #898 )
2021-03-18 20:52:45 +08:00
967d9d5812
[Refactoring]Add Sequential with init_weight ( #884 )
...
* [Refactoring]Add BaseSequtial with init_weight
* revise according to comments
* revise comments
* minors
* baseseq2seq
* add modulelist
* revise minors
* fix isort
* format
2021-03-17 15:45:28 +08:00
73bff4ead1
add tifffile backend support for image io ( #868 )
...
* add tifffile backend support for imageio
* add tifffile package for test requirements
* add tifffile package for test requirements
* small tiffile
2021-03-12 10:28:30 +08:00
9ba1f76005
[Feature] : Add Deformable Conv2d TensorRT Plugin ( #858 )
...
* add dcn tensorrt plugin
* prepare for fp16 support
* fix for lint
* limit column buffer
* add docstring to memcpyPermute
2021-03-11 15:41:47 +08:00
57f3a6142f
fix scatter in pytorch18 ( #882 )
...
* fix scatter in pytorch18
* remove blanks
2021-03-11 13:13:31 +08:00
6fc6f75a76
use ubuntu-18.04 in ci ( #883 )
2021-03-10 12:51:44 +08:00
a4bb2c5b04
[Feature] Support auto contrast ( #881 )
...
* support auto contrast
* add comments
* Fix unittest for pillow 6.2.2
* revised according to comments
2021-03-09 18:46:35 +08:00
3dd98daaa2
fix test unit of nms and batched_nms for tensorrt ( #872 )
2021-03-08 16:57:19 +08:00
ce10425d2f
[Feature] Add adjust sharpness ( #864 )
...
* add adjust sharpness
* revised according to commnts
* fix bug
* fix docstring
* revise according to comments
* fix blank lines
* revised according to comments
2021-03-08 10:23:49 +08:00
27e9ede5b9
[Fix]: fix bugs in some photometric transforms ( #863 )
2021-03-03 18:58:15 +08:00
6719cde43a
add HTTP backend ( #860 )
...
* add HTTP backend
* add HTTPbackend
* fix linting
* fix linting
* add docs attributes http
* fix format
* add real http link test for http backend
2021-03-03 16:23:04 +08:00
e076c8b042
[Feature] Support revise_keys in load_checkpoint(). ( #829 )
...
* Simplified the code.
* Improved chkpt compatibility.
* One may modify the checkpoint via adding keywords.
* Tiny.
* Following reviewer's suggestion.
* Added unit_test.
* Fixed.
* Modify the state_dict with construction.
* Added test.
* Modified。
* Mimimalised the modification.
* Added the docstring.
* Format.
* Improved.
* Tiny.
* Temp file.
* Added assertion.
* Doc string.
* Fixed.
2021-03-03 10:59:11 +08:00
34b552b881
[Feature] Support save last checkpoint ( #853 )
...
* [Feature] Support save last checkpoint
* move to before_run, update doc
* move to after train
* add comments
2021-02-28 12:50:17 +08:00
e5eaf2a729
update video/io.py ( #856 )
2021-02-27 10:28:39 +08:00
48d9902585
Add new parrots extension implementation for all ops ( #794 )
...
* delete all parrots file
add bbox_overlaps new parrots op impl
* support first new impl parrts op (bbox_overlaps)(success test)
* add box_iou_rotated op, test succeed
* add carafe and carafe_naive op, test succeed (one parrots bug need fix)
* add cc_attention op, test success
* add corner_pool op, test success
* add parrots op deform_conv, test success
* add deform_roi_pool op, test success (but has question)
* add focal loss op, test success (gradcheck)
* add masked_conv2d op, test success
* add modulated_deform_conv op, test success
* add nms and nms_rotated op, test success
* add psamask op, test success
* add roi_align op, test_success
* add roi_pool op, test success
* add sync_bn op, test success
* add tin_shift op, test success
* fix test_deform_roi_pool, add parrots test
* skip test_onnx because parrots does not support onnx
* fix c++ lint
* fix python lint
* fix python lint
2021-02-26 19:05:25 +08:00
72e4cc12bd
Fix pytorch2onnx failed for interpolate op with PyTorch==1.6.0(mmdet#4646) ( #848 )
...
* Fix pytorch2onnx for yolov3 with torch==1.6.0
* update and add test for F.interpolate
* add test_onnx.py with build_cuda
2021-02-26 10:49:12 +08:00
ZhangShilong
zhouzaida
Miao Zheng
SFMDI
q.yao
Jerry Jiarui XU
QingChuanWS
shenmishajing
q.yao
Jintao Lin
lizz
tangyanf
Wenwei Zhang
LXXXXR
Rui Xu
sshuair
RunningLeon
Ycr
z55250825