106 Commits

Author	SHA1	Message	Date
Miras Amir	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
Ma Zerun	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
Ezra-Yu	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
Ma Zerun	f9eb9b409b	[Docs] Add Copyright information. (#413)	2021-08-17 19:52:42 +08:00
whcao	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
Ma Zerun	1a7cebe4b9	[Refactor] Refactor unittest (#321) ... * Refactor unit tests folder structure. * Remove label smooth and Vit test in `test_classifiers.py` * Rename test_utils in dataset to test_dataset_utils * Split test_models/test_utils/test_utils.py to multiple sub files. * Add unit tests of classifiers and heads * Use patch context manager. * Add unit test of `is_tracing`, and add warning in `is_tracing` if torch verison is smaller than 1.6.0	2021-07-08 22:49:05 +08:00