mirror of https://github.com/NVlabs/SegFormer.git
Update ade20k_repeat.py
parent
c867a383ff
commit
b0304c4faa
|
@ -24,7 +24,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']),
|
||||
|
|
Loading…
Reference in New Issue