mmsegmentation/tests/test_models
sennnnn 5c195db1bd Add option for output shape of ViT (#530)
* Add arg: final_reshape to control if converting output feature information from NLC to NCHW;

* Fix the default value of final_reshape;

* Modify arg: final_reshape to arg: out_shape;

* Fix some unit test bug;
2021-05-05 22:49:28 -07:00
..
test_backbones Add option for output shape of ViT (#530) 2021-05-05 22:49:28 -07:00
test_heads refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_losses [Enhance] Support reading class_weight from file in loss functions to help MMDet3D (#513) 2021-04-29 01:04:15 -07:00
test_necks add upsample neck (#512) 2021-04-24 21:22:09 -07:00
test_segmentors refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_utils Adjust vision transformer backbone architectures (#524) 2021-04-30 10:37:47 -07:00
__init__.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_forward.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00