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

fast_scnn test: fix BN bug.

pull/58/head
johnzja 2020-08-08 09:51:05 +08:00
parent 8df067e15d
commit 54ae2eeb5f
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_models/test_backbone.py
View File

@ -675,13 +675,14 @@ def test_fastscnn_backbone():
model = FastSCNN()
model.init_weights()
model.train()
imgs = torch.randn(1, 3, 1024, 2048)
num_batch_picts = 4
imgs = torch.randn(num_batch_picts, 3, 1024, 2048)
feat = model(imgs)
assert len(feat) == 3
assert feat[0].shape == torch.Size([1, 64, 128, 256]) # higher-res
assert feat[1].shape == torch.Size([1, 128, 32, 64]) # lower-res
assert feat[2].shape == torch.Size([1, 128, 128, 256]) # FFM output
assert feat[0].shape == torch.Size([num_batch_picts, 64, 128, 256]) # higher-res
assert feat[1].shape == torch.Size([num_batch_picts, 128, 32, 64]) # lower-res
assert feat[2].shape == torch.Size([num_batch_picts, 128, 128, 256]) # FFM output