mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
fec3da781f
* rebase * add glip * update glip * add links * rename * fix doc --------- Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
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,
|
|
)
|