mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
81f10ccbb7
* [Feature] add EdgeNeXt for dev-1.x * [Feature] add update * Imporve EdgeNeXt * Fix div error in low pytorch version Co-authored-by: mzr1996 <mzr1996@163.com>
20 lines
428 B
Python
20 lines
428 B
Python
_base_ = ['./edgenext-base_8xb256_in1k.py']
|
|
|
|
# dataset setting
|
|
|
|
test_pipeline = [
|
|
dict(type='LoadImageFromFile'),
|
|
dict(
|
|
type='ResizeEdge',
|
|
scale=269,
|
|
edge='short',
|
|
backend='pillow',
|
|
interpolation='bicubic'),
|
|
dict(type='CenterCrop', crop_size=256),
|
|
dict(type='PackClsInputs')
|
|
]
|
|
|
|
val_dataloader = dict(dataset=dict(pipeline=test_pipeline))
|
|
|
|
test_dataloader = val_dataloader
|