def test_hff_dynamic_size():
feature_maps = [
torch.randn((4, 3, 512, 512)),
torch.randn((4, 3, 256, 256)),
torch.randn((4, 3, 128, 128)),
torch.randn((4, 3, 64, 64)),
]
hff = HFF(upsample_scale=2)
output = hff(feature_maps)
assert output.size(2) == 512
assert output.size(3) == 512
def test_hff_static_size():
feature_maps = [
torch.randn((4, 3, 512, 512)),
torch.randn((4, 3, 384, 384)),
torch.randn((4, 3, 256, 256)),
torch.randn((4, 3, 128, 128)),
torch.randn((4, 3, 32, 32)),
]
hff = HFF(sizes=[(512, 512), (384, 384), (256, 256), (128, 128), (32, 32)])
output = hff(feature_maps)
assert output.size(2) == 512
assert output.size(3) == 512