eva02-tiny-p14_headless.py 492 Bytes
Newer Older
renzhc's avatar
renzhc committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
model = dict(
    type='ImageClassifier',
    backbone=dict(
        type='ViTEVA02',
        arch='t',
        img_size=224,
        patch_size=14,
        final_norm=False,
        out_type='avg_featmap'),
    neck=None,
    head=None,
)

data_preprocessor = dict(
    # RGB format normalization parameters
    mean=[0.48145466 * 255, 0.4578275 * 255, 0.40821073 * 255],
    std=[0.26862954 * 255, 0.26130258 * 255, 0.27577711 * 255],
    # convert image from BGR to RGB
    to_rgb=True,
)