Update segformer.b5.640x640.ade.160k.py

2021-06-24 16:52:09 +08:00 · 2021-06-24 16:52:09 +08:00 · a7c05dbc96
parent b0304c4faa
commit a7c05dbc96
1 changed files with 1 additions and 1 deletions
--- a/local_configs/segformer/B5/segformer.b5.640x640.ade.160k.py
+++ b/local_configs/segformer/B5/segformer.b5.640x640.ade.160k.py
@ -31,7 +31,7 @@ test_pipeline = [
        # img_ratios=[0.5, 0.75, 1.0, 1.25, 1.5, 1.75],
        flip=False,
        transforms=[
-            dict(type='AlignedResize', keep_ratio=True, size_divisor=32),
+            dict(type='AlignedResize', keep_ratio=True, size_divisor=32), # Ensure the long and short sides are divisible by 32
            dict(type='RandomFlip'),
            dict(type='Normalize', **img_norm_cfg),
            dict(type='ImageToTensor', keys=['img']),