[Refactor]: Unified parameter initialization (#567)
* [Refactor]: Unified parameter initialization * fixed pretrained
Showing
- mmseg/models/backbones/cgnet.py 32 additions, 28 deletionsmmseg/models/backbones/cgnet.py
- mmseg/models/backbones/fast_scnn.py 16 additions, 13 deletionsmmseg/models/backbones/fast_scnn.py
- mmseg/models/backbones/hrnet.py 73 additions, 46 deletionsmmseg/models/backbones/hrnet.py
- mmseg/models/backbones/mobilenet_v2.py 31 additions, 19 deletionsmmseg/models/backbones/mobilenet_v2.py
- mmseg/models/backbones/mobilenet_v3.py 31 additions, 20 deletionsmmseg/models/backbones/mobilenet_v3.py
- mmseg/models/backbones/resnet.py 55 additions, 46 deletionsmmseg/models/backbones/resnet.py
- mmseg/models/backbones/unet.py 32 additions, 25 deletionsmmseg/models/backbones/unet.py
- mmseg/models/backbones/vit.py 13 additions, 5 deletionsmmseg/models/backbones/vit.py
- mmseg/models/decode_heads/decode_head.py 7 additions, 9 deletionsmmseg/models/decode_heads/decode_head.py
- mmseg/models/decode_heads/point_head.py 3 additions, 5 deletionsmmseg/models/decode_heads/point_head.py
- mmseg/models/necks/fpn.py 9 additions, 10 deletionsmmseg/models/necks/fpn.py
- mmseg/models/segmentors/base.py 4 additions, 19 deletionsmmseg/models/segmentors/base.py
- mmseg/models/segmentors/cascade_encoder_decoder.py 4 additions, 19 deletionsmmseg/models/segmentors/cascade_encoder_decoder.py
- mmseg/models/segmentors/encoder_decoder.py 7 additions, 22 deletionsmmseg/models/segmentors/encoder_decoder.py
- mmseg/models/utils/res_layer.py 2 additions, 1 deletionmmseg/models/utils/res_layer.py
- tests/test_models/test_backbones/test_resnet.py 5 additions, 4 deletionstests/test_models/test_backbones/test_resnet.py
- tests/test_models/test_backbones/test_unet.py 3 additions, 7 deletionstests/test_models/test_backbones/test_unet.py
- tests/test_models/test_forward.py 1 addition, 0 deletionstests/test_models/test_forward.py
- tools/train.py 1 addition, 0 deletionstools/train.py
Loading
Please register or sign in to comment