[Fix] Normalizing trimap in indexnet_dimaug_mobv2_1x16_78k_comp1k.py
@@ -59,7 +59,7 @@ test_pipeline = [
save_original_img=True),
dict(type='LoadImageFromFile', key='merged'),
dict(type='Pad', keys=['trimap', 'merged'], mode='reflect'),
- dict(type='RescaleToZeroOne', keys=['merged']),
+ dict(type='RescaleToZeroOne', keys=['merged', 'trimap']),
dict(type='Normalize', keys=['merged'], **img_norm_cfg),
dict(
type='Collect',
