Expand cls model zoo (#55)

1. expand cls model zoo
2. uniform load_pretrained

Author: tuofeilunhifi

configs/classification/cifar10/r50_b128_300e_jpg.py

@@ -2,7 +2,6 @@
 # model settings
 model = dict(
     type='Classification',
-    pretrained=None,
     backbone=dict(
         type='ResNet',
         depth=50,

configs/classification/cifar10/swintiny_b64_5e_jpg.py

@@ -2,7 +2,6 @@
 # model settings
 model = dict(
     type='Classification',
-    pretrained=None,
     backbone=dict(
         type='PytorchImageModelWrapper',
         model_name='swin_tiny_patch4_window7_224',

configs/classification/imagenet/hrnet/hrnetw18_b32x8_100e_jpg.py

@@ -7,7 +7,6 @@
 # model settings
 model = dict(
     type='Classification',
-    pretrained=None,
     backbone=dict(type='HRNet', arch='w18', multi_scale_output=True),
     neck=dict(type='HRFuseScales', in_channels=(18, 36, 72, 144)),
     head=dict(

configs/classification/imagenet/hrnet/imagenet_hrnetw30_jpg.py

@@ -1,3 +1,5 @@
 _base_ = './hrnetw18_b32x8_100e_jpg.py'
 # model settings
-model = dict(backbone=dict(type='HRNet', arch='w30', multi_scale_output=True))
+model = dict(
+    backbone=dict(type='HRNet', arch='w30', multi_scale_output=True),
+    neck=dict(type='HRFuseScales', in_channels=(30, 60, 120, 240)))

configs/classification/imagenet/hrnet/imagenet_hrnetw32_jpg.py

@@ -1,3 +1,5 @@
 _base_ = './hrnetw18_b32x8_100e_jpg.py'
 # model settings
-model = dict(backbone=dict(type='HRNet', arch='w32', multi_scale_output=True))
+model = dict(
+    backbone=dict(type='HRNet', arch='w32', multi_scale_output=True),
+    neck=dict(type='HRFuseScales', in_channels=(32, 64, 128, 256)))

configs/classification/imagenet/hrnet/imagenet_hrnetw40_jpg.py

@@ -1,3 +1,5 @@
 _base_ = './hrnetw18_b32x8_100e_jpg.py'
 # model settings
-model = dict(backbone=dict(type='HRNet', arch='w40', multi_scale_output=True))
+model = dict(
+    backbone=dict(type='HRNet', arch='w40', multi_scale_output=True),
+    neck=dict(type='HRFuseScales', in_channels=(40, 80, 160, 320)))

configs/classification/imagenet/hrnet/imagenet_hrnetw44_jpg.py

@@ -1,3 +1,5 @@
 _base_ = './hrnetw18_b32x8_100e_jpg.py'
 # model settings
-model = dict(backbone=dict(type='HRNet', arch='w44', multi_scale_output=True))
+model = dict(
+    backbone=dict(type='HRNet', arch='w44', multi_scale_output=True),
+    neck=dict(type='HRFuseScales', in_channels=(44, 88, 176, 352)))

configs/classification/imagenet/hrnet/imagenet_hrnetw48_jpg.py

@@ -1,4 +1,6 @@
 _base_ = './hrnetw18_b32x8_100e_jpg.py'
 
 # model settings
-model = dict(backbone=dict(type='HRNet', arch='w48', multi_scale_output=True))
+model = dict(
+    backbone=dict(type='HRNet', arch='w48', multi_scale_output=True),
+    neck=dict(type='HRFuseScales', in_channels=(48, 96, 192, 384)))

configs/classification/imagenet/hrnet/imagenet_hrnetw64_jpg.py

@@ -1,3 +1,5 @@
 _base_ = './hrnetw18_b32x8_100e_jpg.py'
 # model settings
-model = dict(backbone=dict(type='HRNet', arch='w64', multi_scale_output=True))
+model = dict(
+    backbone=dict(type='HRNet', arch='w64', multi_scale_output=True),
+    neck=dict(type='HRFuseScales', in_channels=(64, 128, 256, 512)))

configs/classification/imagenet/resnet/imagenet_resnet101_jpg.py

@@ -2,18 +2,8 @@
 # model settings
 model = dict(
     type='Classification',
-    pretrained=None,
     backbone=dict(
         type='ResNet',
         depth=101,
         out_indices=[4],  # 0: conv-1, x: stage-x
-        norm_cfg=dict(type='BN')),
-    head=dict(
-        type='ClsHead',
-        with_avg_pool=True,
-        in_channels=2048,
-        loss_config=dict(
-            type='CrossEntropyLossWithLabelSmooth',
-            label_smooth=0,
-        ),
-        num_classes=1000))
+        norm_cfg=dict(type='BN')))

configs/classification/imagenet/resnet/imagenet_resnet50_jpg.py

@@ -1,19 +1,8 @@
 _base_ = './resnet50_b32x8_100e_jpg.py'
 # model settings
 model = dict(
-    type='Classification',
-    pretrained=None,
     backbone=dict(
         type='ResNet',
         depth=50,
         out_indices=[4],  # 0: conv-1, x: stage-x
-        norm_cfg=dict(type='BN')),
-    head=dict(
-        type='ClsHead',
-        with_avg_pool=True,
-        in_channels=2048,
-        loss_config=dict(
-            type='CrossEntropyLossWithLabelSmooth',
-            label_smooth=0,
-        ),
-        num_classes=1000))
+        norm_cfg=dict(type='BN')))

configs/classification/imagenet/resnet/resnet50_b32x8_100e_jpg.py

@@ -7,7 +7,6 @@
 # model settings
 model = dict(
     type='Classification',
-    pretrained=None,
     backbone=dict(
         type='ResNet',
         depth=50,

configs/classification/imagenet/resnext/resnext50-32x4d_b32x8_100e_jpg.py

@@ -7,7 +7,6 @@
 # model settings
 model = dict(
     type='Classification',
-    pretrained=None,
     backbone=dict(
         type='ResNeXt',
         depth=50,

configs/classification/imagenet/swint/swin_tiny_patch4_window7_224_b64x16_300e_jpg.py

@@ -16,7 +16,6 @@
         mode='batch',
         label_smoothing=0.1,
         num_classes=1000),
-    pretrained=None,
     backbone=dict(
         type='PytorchImageModelWrapper',
         model_name='swin_tiny_patch4_window7_224',

configs/classification/imagenet/timm/cait/cait_s24_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='cait_s24_224'))

configs/classification/imagenet/timm/cait/cait_xxs24_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='cait_xxs24_224'))

configs/classification/imagenet/timm/cait/cait_xxs36_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='cait_xxs36_224'))

configs/classification/imagenet/timm/coat/coat_mini.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='coat_mini'))

configs/classification/imagenet/timm/coat/coat_tiny.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='coat_tiny'))

configs/classification/imagenet/timm/convit/convit_base.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='convit_base'))

configs/classification/imagenet/timm/convit/convit_small.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='convit_small'))

configs/classification/imagenet/timm/convit/convit_tiny.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='convit_tiny'))

configs/classification/imagenet/timm/convmixer/convmixer_1024_20_ks9_p14.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='convmixer_1024_20_ks9_p14'))

configs/classification/imagenet/timm/convmixer/convmixer_1536_20.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='convmixer_1536_20'))

configs/classification/imagenet/timm/convmixer/convmixer_768_32.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='convmixer_768_32'))

configs/classification/imagenet/timm/convnext/convnext_base.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='convnext_base'))

configs/classification/imagenet/timm/convnext/convnext_large.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='convnext_large'))

configs/classification/imagenet/timm/convnext/convnext_small.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='convnext_small'))

configs/classification/imagenet/timm/convnext/convnext_tiny.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='convnext_tiny'))

configs/classification/imagenet/timm/crossvit/crossvit_base_240.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='crossvit_base_240'))

configs/classification/imagenet/timm/crossvit/crossvit_small_240.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='crossvit_small_240'))

configs/classification/imagenet/timm/crossvit/crossvit_tiny_240.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='crossvit_tiny_240'))

configs/classification/imagenet/timm/deit/deit_base_distilled_patch16_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='deit_base_distilled_patch16_224'))

configs/classification/imagenet/timm/deit/deit_base_patch16_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='deit_base_patch16_224'))

configs/classification/imagenet/timm/gmixer/gmixer_24_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='gmixer_24_224'))

configs/classification/imagenet/timm/gmlp/gmlp_s16_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='gmlp_s16_224'))

configs/classification/imagenet/timm/levit/levit_128.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='levit_128'))

configs/classification/imagenet/timm/levit/levit_192.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='levit_192'))

configs/classification/imagenet/timm/levit/levit_256.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='levit_256'))

configs/classification/imagenet/timm/mlp-mixer/mixer_b16_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='mixer_b16_224'))

configs/classification/imagenet/timm/mlp-mixer/mixer_l16_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='mixer_l16_224'))

configs/classification/imagenet/timm/mobilevit/mobilevit_s.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='mobilevit_s'))

configs/classification/imagenet/timm/mobilevit/mobilevit_xs.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='mobilevit_xs'))

configs/classification/imagenet/timm/mobilevit/mobilevit_xxs.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='mobilevit_xxs'))

configs/classification/imagenet/timm/nest/jx_nest_base.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='jx_nest_base'))

configs/classification/imagenet/timm/nest/jx_nest_small.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='jx_nest_small'))

configs/classification/imagenet/timm/nest/jx_nest_tiny.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='jx_nest_tiny'))

configs/classification/imagenet/timm/pit/pit_b_distilled_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='pit_b_distilled_224'))

configs/classification/imagenet/timm/pit/pit_s_distilled_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='pit_s_distilled_224'))

configs/classification/imagenet/timm/poolformer/poolformer_m36.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='poolformer_m36'))

configs/classification/imagenet/timm/poolformer/poolformer_m48.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='poolformer_m48'))

configs/classification/imagenet/timm/poolformer/poolformer_s12.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='poolformer_s12'))

configs/classification/imagenet/timm/poolformer/poolformer_s24.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='poolformer_s24'))

configs/classification/imagenet/timm/poolformer/poolformer_s36.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='poolformer_s36'))

configs/classification/imagenet/timm/resmlp/resmlp_12_distilled_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='resmlp_12_distilled_224'))

configs/classification/imagenet/timm/resmlp/resmlp_24_distilled_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='resmlp_24_distilled_224'))

configs/classification/imagenet/timm/resmlp/resmlp_36_distilled_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='resmlp_36_distilled_224'))

configs/classification/imagenet/timm/resmlp/resmlp_big_24_distilled_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='resmlp_big_24_distilled_224'))

configs/classification/imagenet/timm/sequencer/sequencer2d_l.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='sequencer2d_l'))

configs/classification/imagenet/timm/sequencer/sequencer2d_m.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='sequencer2d_m'))

configs/classification/imagenet/timm/sequencer/sequencer2d_s.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='sequencer2d_s'))

configs/classification/imagenet/timm/shuffle_transformer/shuffletrans_base_p4_w7_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='shuffletrans_base_p4_w7_224'))

configs/classification/imagenet/timm/shuffle_transformer/shuffletrans_small_p4_w7_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='shuffletrans_small_p4_w7_224'))

configs/classification/imagenet/timm/shuffle_transformer/shuffletrans_tiny_p4_w7_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='shuffletrans_tiny_p4_w7_224'))

configs/classification/imagenet/timm/swint/dynamic_swin_small_p4_w7_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='dynamic_swin_small_p4_w7_224'))

configs/classification/imagenet/timm/swint/dynamic_swin_tiny_p4_w7_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='dynamic_swin_tiny_p4_w7_224'))

configs/classification/imagenet/timm/swint/swin_base_patch4_window7_224.py

@@ -0,0 +1,4 @@
+_base_ = '../timm_config.py'
+
+# model settings
+model = dict(backbone=dict(model_name='swin_base_patch4_window7_224'))