Spaces:
Runtime error
Runtime error
# dataset settings | |
dataset_type = 'CUB' | |
data_preprocessor = dict( | |
num_classes=200, | |
mean=[123.675, 116.28, 103.53], | |
std=[58.395, 57.12, 57.375], | |
# convert image from BGR to RGB | |
to_rgb=True, | |
) | |
train_pipeline = [ | |
dict(type='LoadImageFromFile'), | |
dict(type='Resize', scale=600), | |
dict(type='RandomCrop', crop_size=448), | |
dict(type='RandomFlip', prob=0.5, direction='horizontal'), | |
dict(type='PackClsInputs'), | |
] | |
test_pipeline = [ | |
dict(type='LoadImageFromFile'), | |
dict(type='Resize', scale=600), | |
dict(type='CenterCrop', crop_size=448), | |
dict(type='PackClsInputs'), | |
] | |
train_dataloader = dict( | |
batch_size=8, | |
num_workers=2, | |
dataset=dict( | |
type=dataset_type, | |
data_root='data/CUB_200_2011', | |
test_mode=False, | |
pipeline=train_pipeline), | |
sampler=dict(type='DefaultSampler', shuffle=True), | |
) | |
val_dataloader = dict( | |
batch_size=8, | |
num_workers=2, | |
dataset=dict( | |
type=dataset_type, | |
data_root='data/CUB_200_2011', | |
test_mode=True, | |
pipeline=test_pipeline), | |
sampler=dict(type='DefaultSampler', shuffle=False), | |
) | |
val_evaluator = dict(type='Accuracy', topk=(1, )) | |
test_dataloader = val_dataloader | |
test_evaluator = val_evaluator | |