Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
mmpretrain
Commits
495d9ed9
Commit
495d9ed9
authored
Jun 24, 2025
by
limm
Browse files
add part code
parent
59b09903
Pipeline
#2799
canceled with stages
Changes
1000
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
391 additions
and
0 deletions
+391
-0
configs/_base_/models/tinyvit/tinyvit-21m.py
configs/_base_/models/tinyvit/tinyvit-21m.py
+25
-0
configs/_base_/models/tinyvit/tinyvit-5m.py
configs/_base_/models/tinyvit/tinyvit-5m.py
+25
-0
configs/_base_/models/tnt_s_patch16_224.py
configs/_base_/models/tnt_s_patch16_224.py
+29
-0
configs/_base_/models/twins_pcpvt_base.py
configs/_base_/models/twins_pcpvt_base.py
+31
-0
configs/_base_/models/twins_svt_base.py
configs/_base_/models/twins_svt_base.py
+31
-0
configs/_base_/models/van/van_base.py
configs/_base_/models/van/van_base.py
+13
-0
configs/_base_/models/van/van_large.py
configs/_base_/models/van/van_large.py
+13
-0
configs/_base_/models/van/van_small.py
configs/_base_/models/van/van_small.py
+22
-0
configs/_base_/models/van/van_tiny.py
configs/_base_/models/van/van_tiny.py
+22
-0
configs/_base_/models/vgg11.py
configs/_base_/models/vgg11.py
+10
-0
configs/_base_/models/vgg11bn.py
configs/_base_/models/vgg11bn.py
+11
-0
configs/_base_/models/vgg13.py
configs/_base_/models/vgg13.py
+10
-0
configs/_base_/models/vgg13bn.py
configs/_base_/models/vgg13bn.py
+11
-0
configs/_base_/models/vgg16.py
configs/_base_/models/vgg16.py
+10
-0
configs/_base_/models/vgg16bn.py
configs/_base_/models/vgg16bn.py
+11
-0
configs/_base_/models/vgg19.py
configs/_base_/models/vgg19.py
+10
-0
configs/_base_/models/vgg19bn.py
configs/_base_/models/vgg19bn.py
+11
-0
configs/_base_/models/vig/pyramid_vig_base.py
configs/_base_/models/vig/pyramid_vig_base.py
+32
-0
configs/_base_/models/vig/pyramid_vig_medium.py
configs/_base_/models/vig/pyramid_vig_medium.py
+32
-0
configs/_base_/models/vig/pyramid_vig_small.py
configs/_base_/models/vig/pyramid_vig_small.py
+32
-0
No files found.
Too many changes to show.
To preserve performance only
1000 of 1000+
files are displayed.
Plain diff
Email patch
configs/_base_/models/tinyvit/tinyvit-21m.py
0 → 100644
View file @
495d9ed9
# Model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'TinyViT'
,
arch
=
'21m'
,
img_size
=
(
224
,
224
),
window_size
=
[
7
,
7
,
14
,
7
],
out_indices
=
(
3
,
),
drop_path_rate
=
0.2
,
gap_before_final_norm
=
True
,
init_cfg
=
[
dict
(
type
=
'TruncNormal'
,
layer
=
[
'Conv2d'
,
'Linear'
],
std
=
.
02
,
bias
=
0.
),
dict
(
type
=
'Constant'
,
layer
=
[
'LayerNorm'
],
val
=
1.
,
bias
=
0.
),
]),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
576
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
))
configs/_base_/models/tinyvit/tinyvit-5m.py
0 → 100644
View file @
495d9ed9
# Model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'TinyViT'
,
arch
=
'5m'
,
img_size
=
(
224
,
224
),
window_size
=
[
7
,
7
,
14
,
7
],
out_indices
=
(
3
,
),
drop_path_rate
=
0.0
,
gap_before_final_norm
=
True
,
init_cfg
=
[
dict
(
type
=
'TruncNormal'
,
layer
=
[
'Conv2d'
,
'Linear'
],
std
=
.
02
,
bias
=
0.
),
dict
(
type
=
'Constant'
,
layer
=
[
'LayerNorm'
],
val
=
1.
,
bias
=
0.
),
]),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
320
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
))
configs/_base_/models/tnt_s_patch16_224.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'TNT'
,
arch
=
's'
,
img_size
=
224
,
patch_size
=
16
,
in_channels
=
3
,
ffn_ratio
=
4
,
qkv_bias
=
False
,
drop_rate
=
0.
,
attn_drop_rate
=
0.
,
drop_path_rate
=
0.1
,
first_stride
=
4
,
num_fcs
=
2
,
init_cfg
=
[
dict
(
type
=
'TruncNormal'
,
layer
=
'Linear'
,
std
=
.
02
),
dict
(
type
=
'Constant'
,
layer
=
'LayerNorm'
,
val
=
1.
,
bias
=
0.
)
]),
neck
=
None
,
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
384
,
loss
=
dict
(
type
=
'LabelSmoothLoss'
,
label_smooth_val
=
0.1
,
mode
=
'original'
),
topk
=
(
1
,
5
),
init_cfg
=
dict
(
type
=
'TruncNormal'
,
layer
=
'Linear'
,
std
=
.
02
)))
configs/_base_/models/twins_pcpvt_base.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'PCPVT'
,
arch
=
'base'
,
in_channels
=
3
,
out_indices
=
(
3
,
),
qkv_bias
=
True
,
norm_cfg
=
dict
(
type
=
'LN'
,
eps
=
1e-06
),
norm_after_stage
=
[
False
,
False
,
False
,
True
],
drop_rate
=
0.0
,
attn_drop_rate
=
0.
,
drop_path_rate
=
0.3
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
512
,
loss
=
dict
(
type
=
'LabelSmoothLoss'
,
label_smooth_val
=
0.1
,
mode
=
'original'
),
cal_acc
=
False
),
init_cfg
=
[
dict
(
type
=
'TruncNormal'
,
layer
=
'Linear'
,
std
=
0.02
,
bias
=
0.
),
dict
(
type
=
'Constant'
,
layer
=
'LayerNorm'
,
val
=
1.
,
bias
=
0.
)
],
train_cfg
=
dict
(
augments
=
[
dict
(
type
=
'Mixup'
,
alpha
=
0.8
),
dict
(
type
=
'CutMix'
,
alpha
=
1.0
)
]),
)
configs/_base_/models/twins_svt_base.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'SVT'
,
arch
=
'base'
,
in_channels
=
3
,
out_indices
=
(
3
,
),
qkv_bias
=
True
,
norm_cfg
=
dict
(
type
=
'LN'
),
norm_after_stage
=
[
False
,
False
,
False
,
True
],
drop_rate
=
0.0
,
attn_drop_rate
=
0.
,
drop_path_rate
=
0.3
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
768
,
loss
=
dict
(
type
=
'LabelSmoothLoss'
,
label_smooth_val
=
0.1
,
mode
=
'original'
),
cal_acc
=
False
),
init_cfg
=
[
dict
(
type
=
'TruncNormal'
,
layer
=
'Linear'
,
std
=
0.02
,
bias
=
0.
),
dict
(
type
=
'Constant'
,
layer
=
'LayerNorm'
,
val
=
1.
,
bias
=
0.
)
],
train_cfg
=
dict
(
augments
=
[
dict
(
type
=
'Mixup'
,
alpha
=
0.8
),
dict
(
type
=
'CutMix'
,
alpha
=
1.0
)
]),
)
configs/_base_/models/van/van_base.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VAN'
,
arch
=
'base'
,
drop_path_rate
=
0.1
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
512
,
init_cfg
=
None
,
# suppress the default init_cfg of LinearClsHead.
loss
=
dict
(
type
=
'LabelSmoothLoss'
,
label_smooth_val
=
0.1
,
mode
=
'original'
),
cal_acc
=
False
))
configs/_base_/models/van/van_large.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VAN'
,
arch
=
'large'
,
drop_path_rate
=
0.2
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
512
,
init_cfg
=
None
,
# suppress the default init_cfg of LinearClsHead.
loss
=
dict
(
type
=
'LabelSmoothLoss'
,
label_smooth_val
=
0.1
,
mode
=
'original'
),
cal_acc
=
False
))
configs/_base_/models/van/van_small.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VAN'
,
arch
=
'small'
,
drop_path_rate
=
0.1
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
512
,
init_cfg
=
None
,
# suppress the default init_cfg of LinearClsHead.
loss
=
dict
(
type
=
'LabelSmoothLoss'
,
label_smooth_val
=
0.1
,
mode
=
'original'
),
cal_acc
=
False
),
init_cfg
=
[
dict
(
type
=
'TruncNormal'
,
layer
=
'Linear'
,
std
=
0.02
,
bias
=
0.
),
dict
(
type
=
'Constant'
,
layer
=
'LayerNorm'
,
val
=
1.
,
bias
=
0.
)
],
train_cfg
=
dict
(
augments
=
[
dict
(
type
=
'Mixup'
,
alpha
=
0.8
),
dict
(
type
=
'CutMix'
,
alpha
=
1.0
)
]),
)
configs/_base_/models/van/van_tiny.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VAN'
,
arch
=
'tiny'
,
drop_path_rate
=
0.1
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
256
,
init_cfg
=
None
,
# suppress the default init_cfg of LinearClsHead.
loss
=
dict
(
type
=
'LabelSmoothLoss'
,
label_smooth_val
=
0.1
,
mode
=
'original'
),
cal_acc
=
False
),
init_cfg
=
[
dict
(
type
=
'TruncNormal'
,
layer
=
'Linear'
,
std
=
0.02
,
bias
=
0.
),
dict
(
type
=
'Constant'
,
layer
=
'LayerNorm'
,
val
=
1.
,
bias
=
0.
)
],
train_cfg
=
dict
(
augments
=
[
dict
(
type
=
'Mixup'
,
alpha
=
0.8
),
dict
(
type
=
'CutMix'
,
alpha
=
1.0
)
]),
)
configs/_base_/models/vgg11.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VGG'
,
depth
=
11
,
num_classes
=
1000
),
neck
=
None
,
head
=
dict
(
type
=
'ClsHead'
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/vgg11bn.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VGG'
,
depth
=
11
,
norm_cfg
=
dict
(
type
=
'BN'
),
num_classes
=
1000
),
neck
=
None
,
head
=
dict
(
type
=
'ClsHead'
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/vgg13.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VGG'
,
depth
=
13
,
num_classes
=
1000
),
neck
=
None
,
head
=
dict
(
type
=
'ClsHead'
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/vgg13bn.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VGG'
,
depth
=
13
,
norm_cfg
=
dict
(
type
=
'BN'
),
num_classes
=
1000
),
neck
=
None
,
head
=
dict
(
type
=
'ClsHead'
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/vgg16.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VGG'
,
depth
=
16
,
num_classes
=
1000
),
neck
=
None
,
head
=
dict
(
type
=
'ClsHead'
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/vgg16bn.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VGG'
,
depth
=
16
,
norm_cfg
=
dict
(
type
=
'BN'
),
num_classes
=
1000
),
neck
=
None
,
head
=
dict
(
type
=
'ClsHead'
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/vgg19.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VGG'
,
depth
=
19
,
num_classes
=
1000
),
neck
=
None
,
head
=
dict
(
type
=
'ClsHead'
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/vgg19bn.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'VGG'
,
depth
=
19
,
norm_cfg
=
dict
(
type
=
'BN'
),
num_classes
=
1000
),
neck
=
None
,
head
=
dict
(
type
=
'ClsHead'
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/vig/pyramid_vig_base.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'PyramidVig'
,
arch
=
'base'
,
k
=
9
,
act_cfg
=
dict
(
type
=
'GELU'
),
norm_cfg
=
dict
(
type
=
'BN'
),
graph_conv_type
=
'mr'
,
graph_conv_bias
=
True
,
epsilon
=
0.2
,
use_stochastic
=
False
,
drop_path
=
0.1
,
norm_eval
=
False
,
frozen_stages
=
0
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'VigClsHead'
,
num_classes
=
1000
,
in_channels
=
1024
,
hidden_dim
=
1024
,
act_cfg
=
dict
(
type
=
'GELU'
),
dropout
=
0.
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
),
train_cfg
=
dict
(
augments
=
[
dict
(
type
=
'Mixup'
,
alpha
=
0.8
),
dict
(
type
=
'CutMix'
,
alpha
=
1.0
)
]),
)
configs/_base_/models/vig/pyramid_vig_medium.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'PyramidVig'
,
arch
=
'medium'
,
k
=
9
,
act_cfg
=
dict
(
type
=
'GELU'
),
norm_cfg
=
dict
(
type
=
'BN'
),
graph_conv_type
=
'mr'
,
graph_conv_bias
=
True
,
epsilon
=
0.2
,
use_stochastic
=
False
,
drop_path
=
0.1
,
norm_eval
=
False
,
frozen_stages
=
0
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'VigClsHead'
,
num_classes
=
1000
,
in_channels
=
768
,
hidden_dim
=
1024
,
act_cfg
=
dict
(
type
=
'GELU'
),
dropout
=
0.
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
),
train_cfg
=
dict
(
augments
=
[
dict
(
type
=
'Mixup'
,
alpha
=
0.8
),
dict
(
type
=
'CutMix'
,
alpha
=
1.0
)
]),
)
configs/_base_/models/vig/pyramid_vig_small.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'PyramidVig'
,
arch
=
'small'
,
k
=
9
,
act_cfg
=
dict
(
type
=
'GELU'
),
norm_cfg
=
dict
(
type
=
'BN'
),
graph_conv_type
=
'mr'
,
graph_conv_bias
=
True
,
epsilon
=
0.2
,
use_stochastic
=
False
,
drop_path
=
0.1
,
norm_eval
=
False
,
frozen_stages
=
0
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'VigClsHead'
,
num_classes
=
1000
,
in_channels
=
640
,
hidden_dim
=
1024
,
act_cfg
=
dict
(
type
=
'GELU'
),
dropout
=
0.
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
),
train_cfg
=
dict
(
augments
=
[
dict
(
type
=
'Mixup'
,
alpha
=
0.8
),
dict
(
type
=
'CutMix'
,
alpha
=
1.0
)
]),
)
Prev
1
…
10
11
12
13
14
15
16
17
18
…
50
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment