FIX: update GCA config for new LoadImageFromFile api
@@ -52,12 +52,15 @@ train_pipeline = [
]
test_pipeline = [
dict(
- type='LoadAlpha', key='alpha', flag='grayscale', save_origin_img=True),
+ type='LoadAlpha',
+ key='alpha',
+ flag='grayscale',
+ save_original_img=True),
dict(
type='LoadImageFromFile',
key='trimap',
flag='grayscale',
- save_origin_img=True),
+ save_original_img=True),
dict(type='LoadImageFromFile', key='merged'),
dict(type='Pad', keys=['alpha', 'trimap', 'merged'], mode='reflect'),
dict(type='RescaleToZeroOne', keys=['merged', 'alpha', 'ori_alpha']),
@@ -96,7 +99,8 @@ data = dict(
optimizers = dict(type='Adam', lr=4e-4, betas=[0.5, 0.999])
# learning policy
lr_config = dict(
- policy='Cosine',
+ policy='CosineAnealing',
+ min_lr=0,
by_epoch=False,
warmup='linear',
warmup_iters=5000,
