1d6d142b42
[Docs] Use PyTorch style docs theme ( #457 )
...
* Change docs theme to pytorch-theme
* Update docs rst
* Fix docs title level.
* Add static resources
* Fix lint
2021-09-23 22:21:03 +08:00
0b20aa6873
[Fix]: Fix missing runtime requirement `packaging`. ( #459 )
2021-09-23 18:04:42 +08:00
75b087f27e
[Refactor] Fix TnT compatibility and verbose warning. ( #436 )
...
* Support return tuple in TnT
* Fix verbose warnings.
2021-09-22 19:37:24 +08:00
f865c725f2
[Docs] Fix typo in transforms.py docstring ( #455 )
2021-09-22 11:09:23 +08:00
8024f81de2
[Docs] Fix a typo in transforms ( #454 )
2021-09-22 10:46:59 +08:00
8f68779cc6
[Docs] Update metafile and Readme ( #435 )
...
* Update metafile format.
* Update accuracy of checkpoints.
* Add metafile and readme for tnt.
* Add converted ckpts in swin-transformer.
* Fix tnt ckpt link
* Update swin_transformer metafile
2021-09-18 16:32:46 +08:00
cb09ed54e5
[Enhance] Support setting `--out-items` in `tools/test.py`. ( #437 )
...
* Support setting out_details in `tools/test.py`.
* Add assertion in `eval_metric` and `analyze_results`.
2021-09-15 11:06:16 +08:00
617932dd59
[Fix] Fix simplified model saving in ONNX export tool. ( #438 )
...
* Update pytorch2onnx.py
* Update pytorch2onnx.py
Please check the check_ok result.
* Update pytorch2onnx.py
* Update pytorch2onnx.py
2021-09-10 11:47:05 +08:00
fe25c8d3fb
[Enchence] Add datetime info and saving model using torch<1.6 format ( #439 )
...
* Add date and save ckpt usingg torch<1.6 format
* fix lint
* use digit_version and rasie error when there is no target out_dir
* add '.'
2021-09-10 11:42:38 +08:00
a8f4f82b8e
[Enhance] Improve downstream repositories compatibility ( #421 )
...
* Defaults to return tuple in all backbones.
* Compat downstream of swin transformer.
* Support tuple input for multi label head and stacked head.
* Fix backbone unit tests for tuple output.
* Add downstream inference unit tests for mmdet.
* Update gitignore
* Add unit tests for `return_tuple` option
* Add unit tests for head input tuple.
* Add warning in `simple_test`
* Add TIMMBackbone return tuple.
* Modify timm backbone unit test.
2021-09-08 10:38:57 +08:00
5cfaed6807
[Feature] timm backbones wrapper ( #427 )
...
* Add wrapper to use backbones from timm
* Add tests
* Remove timm from optional deps and modify GitHub workflow.
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-09-06 11:05:31 +08:00
729c6c1192
[Docs] Add `CITATION.cff` ( #428 )
...
* Add CITATION.cff
* Fix typo in setup.py
* Change author in setup.py
2021-09-01 17:07:27 +08:00
0d0b1f5f92
[Docs] Fix sphinx version ( #429 )
2021-09-01 17:07:08 +08:00
ae40c45c55
[Enhance] Rename the option `--options` in some tools to `--cfg-options`. ( #425 )
2021-09-01 12:49:03 +08:00
5aa132a91c
[CI] Add PyTorch 1.9 and Python 3.9 build workflow, and remove some CI. ( #422 )
...
* Add PyTorch 1.9 build workflow, and remove some CI.
* Add Python 3.9 CI
* Show Python 3.9 support.
2021-09-01 11:06:00 +08:00
a41cb2fa93
Bump version to v0.15.0 ( #426 )
2021-08-31 14:33:32 +08:00
5383787512
[Enhance] Better result visualization ( #419 )
...
* Imporve result visualization to support wait time and change the backend
to matplotlib.
* Add unit test for visualization
* Add adaptive dpi function
* Rename `imshow_cls_result` to `imshow_infos`.
* Support str in `imshow_infos`
* Improve docstring.
2021-08-31 10:50:28 +08:00
192b79eea0
[Feature] Support classwise weight in losses ( #388 )
...
* Add classwise weight in losses:CE,BCE,softBCE
* Update unit test
* rm some extra code
* rm some extra code
* fix broadcast
* fix broadcast
* update unit tests
* use new_tensor
* fix lint
2021-08-31 10:44:12 +08:00
6a0a76af0c
[Enhance] Add `hparams` argument in `AutoAugment` and `RandAugment` and some other improvement. ( #398 )
...
* Add hparams argument in `AutoAugment` and `RandAugment`.
And `pad_val` supports sequence instead of tuple only.
* Add unit tests for `AutoAugment` and `hparams` in `RandAugment`.
* Use smaller test image to speed up uni tests.
* Use hparams to simplify RandAugment config in swin-transformer.
* Rename augment config name from `pipeline` to `pipelines`.
* Add some commnet ad docstring.
2021-08-24 18:15:54 +08:00
a9d65271ab
[Docs] Add algorithm readme and update meta yml ( #418 )
...
* Add README.md for models without checkpoints.
* Update model-index.yml
* Update metafile.yml of seresnet
2021-08-24 17:46:46 +08:00
0184527bd4
[Enhance] Enhence SE layer to support custom squeeze channels. ( #417 )
...
* add enhenced SE
* Update
* rm basechannel
* fix docstring
* Update se_layer.py
fix docstring
2021-08-20 13:31:44 +08:00
f9eb9b409b
[Docs] Add Copyright information. ( #413 )
2021-08-17 19:52:42 +08:00
5c4da16084
[Docs] Fix typo 'metirc' ( #411 )
2021-08-17 10:35:26 +08:00
c0a4916a06
[Enhance] Update `digit_version` function. ( #402 )
...
* Update digit_version
* Add digit_version unit test
2021-08-12 12:13:25 +08:00
f8f1700860
[Refactor] Use post_process function to handle pred result processing. ( #390 )
...
Use post_process function to handle pred result processing in `simple_test`.
2021-08-12 11:54:24 +08:00
e4188c828c
Avoid albumentations to install both opencv and opencv-headless. ( #397 )
2021-08-09 10:32:13 +08:00
b54acfd5c4
[Enhance] Avoid unnecessary listdir when building ImageNet. ( #396 )
2021-08-06 10:38:36 +08:00
a747458ba8
[Docs] Update QQ group QR code. ( #393 )
2021-08-05 17:51:53 +08:00
64bbed41f4
[Enhance] Use dynamic mmcv download link in TorchServe dockerfile. ( #387 )
...
* Add mmcv arg
* Declare ARG after FROM
Co-authored-by: Ma Zerun <mzr1996@163.com>
2021-08-04 17:46:40 +08:00
4aec976c87
Delete custom.md
2021-08-04 14:31:53 +08:00
2aa4100595
[Docs] Add PR template and modify issue template ( #380 )
...
* Add language suggestion in issue-template
* Add issue template config
* Add PR template
* Update pr template according to mmcv #1228
2021-08-04 14:30:20 +08:00
ade7b80e44
Bump version to v0.14.0 ( #389 )
2021-08-04 13:25:42 +08:00
df5cec3e92
[Docs] Fix readthedocs config lint ( #384 )
2021-07-31 16:20:42 +08:00
e8822ebaec
[Docs] Fix some bugs in readthedocs pdf build. ( #383 )
...
* Add escape symbols for underscores in formulas.
* Support Chinese characters in pdf output.
2021-07-31 14:49:36 +08:00
469d88c21b
Support output pdf version readthedocs document. ( #382 )
2021-07-31 14:11:50 +08:00
359f56ad58
[Feature] Add transformer in transformer ( #339 )
...
* add tnt_small configs
* add tnt backbone
* test tnt
* add tnt to model_zoo
* rename the config file name
* add optimizor
* move tnt backbone unitest
* add metric
* fix keyname in arch
* encapsulate "inner transformer block" and "outer transformer block"
* fix TnT
* Use `inner_block_cfg` and `outer_block_cfg` instead of `args` and
`kwargs`.
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-07-30 09:16:27 +08:00
db856df43e
[Fix] Fix ImageNet dataset annotation file parse bug. ( #370 )
...
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-07-29 10:21:22 +08:00
c92eefbb4a
[Docs] Fix `Open in colab` link in Colab tutorials. ( #377 )
2021-07-27 19:57:21 +08:00
e3b77e7508
[Fix] Fix `patch_cfg` argument bug in SwinTransformer ( #368 )
2021-07-27 19:19:44 +08:00
1585ca83c8
[Fix] Fix docstring typo and init error in ShuffleNetV1 ( #374 )
2021-07-27 19:17:42 +08:00
33a4c0fe9f
[Fix] Use local ATTENTION registery to avoid conflict with other repo. ( #375 )
2021-07-27 17:59:05 +08:00
0f4cae4d3f
[Feature] Support mim ( #376 )
...
* Support mim extention
* Fix lint.
* Remove extra `demo` setting in mim related files.
2021-07-27 17:58:04 +08:00
99e3d4e49e
[Feature] Add Dockerfile ( #365 )
...
* add Dockerfile
* Update Dockerfile
* Update Dockerfile
* Add docker install guide
* Add docker guide
* Update readme
* Update readme
2021-07-27 14:35:49 +08:00
19ff0d198c
[Docs] Add chinese ipynb tutorial ( #306 )
...
* add chinese ipynb
* Update Chinese Colab tutorials.
* Update colab docs to ensure deterministic.
* Add `open in colab` button.
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-07-27 13:58:27 +08:00
899047a3b3
Fix duplicate `init_weights` call in ViT init function. ( #373 )
2021-07-26 05:33:11 -04:00
57e9692765
Fix broken _base_ link in a resnet config. ( #361 )
2021-07-25 23:50:32 -04:00
474c55d035
Fix mmcv dep version in mminstall.txt ( #346 )
2021-07-23 16:09:55 +08:00
9352fc434a
Fix vgg-19 model link. ( #363 )
2021-07-22 11:29:10 +08:00
15cd34bbef
[Fix] Use zero as default value of `thrs` in metrics. ( #341 )
...
* Use zero as default value of `thrs` in metrics. And it accepcts a number
instead of float now.
* Fix unit test comment
* Don't pass thrs if no thrs.
2021-07-18 16:57:21 +08:00
24e58ba26d
[Feature] Add ResNeSt configs ( #332 )
...
* add configs for resnest training
* fix format
2021-07-14 15:58:19 +08:00
Ma Zerun
RangiLyu
Ezra-Yu
Charlyo
yushu
Miras Amir
Azad Baykara
Vladislav Sovrasov
Miguel Méndez
whcao
pvys
Junjun2016
WRH
LXXXXR