mirror of
https://github.com/open-mmlab/mmselfsup.git
synced 2025-06-03 14:59:38 +08:00
86726ec615
* update ci workflow * add test for config * update configs * update imagenet 1pct and 10pct * update configs according to mmcls interface * update link * add copyright * update ci * update ci install command * update data_preprocessor field * fix value of mean, std and add comments * remove redundant configs
38 lines
972 B
Python
38 lines
972 B
Python
dataset_type = 'mmcls.ImageNet'
|
|
data_root = 'data/imagenet'
|
|
file_client_args = dict(backend='disk')
|
|
|
|
extract_pipeline = [
|
|
dict(type='LoadImageFromFile', file_client_args=file_client_args),
|
|
dict(type='mmcls.ResizeEdge', scale=256, edge='short'),
|
|
dict(type='CenterCrop', crop_size=224),
|
|
dict(type='PackSelfSupInputs'),
|
|
]
|
|
|
|
train_dataloader = dict(
|
|
batch_size=256,
|
|
num_workers=8,
|
|
dataset=dict(
|
|
type=dataset_type,
|
|
data_root=data_root,
|
|
ann_file='meta/train.txt',
|
|
data_prefix='train',
|
|
pipeline=extract_pipeline),
|
|
sampler=dict(type='DefaultSampler', shuffle=False),
|
|
)
|
|
|
|
val_dataloader = dict(
|
|
batch_size=256,
|
|
num_workers=8,
|
|
dataset=dict(
|
|
type=dataset_type,
|
|
data_root=data_root,
|
|
ann_file='meta/val.txt',
|
|
data_prefix='val',
|
|
pipeline=extract_pipeline),
|
|
sampler=dict(type='DefaultSampler', shuffle=False),
|
|
)
|
|
|
|
# pooling cfg
|
|
pool_cfg = dict(type='AvgPool2d')
|