19 lines
434 B
Python
19 lines
434 B
Python
model = dict(
|
|
type='ImageClassifier',
|
|
backbone=dict(
|
|
type='SwinTransformer',
|
|
arch='tiny',
|
|
img_size=224,
|
|
out_indices=(1, 2, 3), # original weight is for detection
|
|
),
|
|
neck=None,
|
|
head=None)
|
|
|
|
data_preprocessor = dict(
|
|
# RGB format normalization parameters
|
|
mean=[103.53, 116.28, 123.675],
|
|
std=[57.375, 57.12, 58.395],
|
|
# convert image from BGR to RGB
|
|
to_rgb=False,
|
|
)
|