mmpretrain

Commit Graph

Author	SHA1	Message	Date
Ma Zerun	643fb192cd	[Enhance] Enhance feature extraction function. (#593 ) * Fix MobileNet V3 configs * Refactor to support more powerful feature extraction. * Add unit tests * Fix unit test * Imporve according to comments * Update checkpoints path * Fix unit tests * Add docstring of `simple_test` * Add docstring of `extract_feat` * Update model zoo	2021-12-17 15:55:02 +08:00
Ezra-Yu	34d5a25281	[Refactor] Support passing arguments to loss from head. (#523 )	2021-11-10 17:12:34 +08:00
Ma Zerun	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
Ma Zerun	f9eb9b409b	[Docs] Add Copyright information. (#413 )	2021-08-17 19:52:42 +08:00
Ma Zerun	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
Ma Zerun	71621a5f62	Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347 )	2021-07-07 11:55:53 +08:00
AllentDan	a24a9f6faa	[Fix] Build compatible with low pytorch versions (#301 ) * add version compatible for torchscript * doc * doc again * fix lint * fix lint isort	2021-06-14 23:25:35 +08:00
AllentDan	c2f01e0dcd	[Feature] Add torchscript deployment (#279 ) * add torchscript deploy * fix lint * add check and delete \	2021-06-12 21:50:48 +08:00
Miao Zheng	4ca21c7d03	[WIP] Refactoring weights initialization (#270 ) * [WIP] Refactoring weights initialization * fix lint and constant init cfg * fix pretrained bug * fix typo * fix isort * revise model utils	2021-06-10 10:54:34 +08:00
mzr1996	b7b520881f	Update CONTRIBUTING.md according to mmcv (#210 ) * Update CONTRIBUTING.md according to mmcv * Docstring formatting by docformatter * Update openmmlab website.	2021-04-14 21:22:37 +08:00
LXXXXR	07bb15e5fd	[Feature] Add heads and config for multilabel task (#145 ) * resolve conflicts add heads and config for multilabel tasks * minor change * remove evaluating mAP in head * add baseline config * add configs * reserve only one config * minor change * fix minor bug * minor change * minor change * add unittests and fix docstrings	2021-01-25 18:10:14 +08:00

11 Commits (a5a9fe30a92a0a9f406365be670c09a47308d297)