mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix (#30) 

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
pull/34/head
Xiaojie Li 2020-08-13 09:50:14 +08:00 committed by GitHub
parent 7af9419ffa
commit 608cea2469
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/_base_/datasets/imagenet_bs32.py
View File

@ -13,7 +13,7 @@ train_pipeline = [
]
test_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='Resize', size=256),
dict(type='Resize', size=(256, -1)),
dict(type='CenterCrop', crop_size=224),
dict(type='Normalize', **img_norm_cfg),
dict(type='ImageToTensor', keys=['img']),

2
configs/_base_/datasets/imagenet_bs32_pil_resize.py
View File

@ -13,7 +13,7 @@ train_pipeline = [
]
test_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='Resize', size=256, backend='pillow'),
dict(type='Resize', size=(256, -1), backend='pillow'),
dict(type='CenterCrop', crop_size=224),
dict(type='Normalize', **img_norm_cfg),
dict(type='ImageToTensor', keys=['img']),

2
configs/_base_/datasets/imagenet_bs64.py
View File

@ -13,7 +13,7 @@ train_pipeline = [
]
test_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='Resize', size=256),
dict(type='Resize', size=(256, -1)),
dict(type='CenterCrop', crop_size=224),
dict(type='Normalize', **img_norm_cfg),
dict(type='ImageToTensor', keys=['img']),

2
configs/_base_/datasets/imagenet_bs64_pil_resize.py
View File

@ -13,7 +13,7 @@ train_pipeline = [
]
test_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='Resize', size=256, backend='pillow'),
dict(type='Resize', size=(256, -1), backend='pillow'),
dict(type='CenterCrop', crop_size=224),
dict(type='Normalize', **img_norm_cfg),
dict(type='ImageToTensor', keys=['img']),