Python PyTorch FeaturePyramidNetwork用法及代码示例

本文简要介绍python语言中 torchvision.ops.FeaturePyramidNetwork 的用法。

用法:

class torchvision.ops.FeaturePyramidNetwork(in_channels_list: List[int], out_channels: int, extra_blocks: Optional[torchvision.ops.feature_pyramid_network.ExtraFPNBlock] = None)

参数：

• in_channels_list(list[int]) -传递给模块的每个特征图的通道数

• out_channels(int) -FPN 表示的通道数

• extra_blocks(ExtraFPNBlock或者None) -如果提供，将执行额外的操作。期望将fpn特征、原始特征和原始特征的名称作为输入，并返回一个新的特征图列表及其对应的名称

从一组特征图的顶部添加 FPN 的模块。这是基于“Feature Pyramid Network for Object Detection”。

特征图目前应该是按深度递增的顺序排列的。

模型的输入预计是 OrderedDict[Tensor]，包含将在其上添加 FPN 的特征图。

例子：

>>> m = torchvision.ops.FeaturePyramidNetwork([10, 20, 30], 5)
>>> # get some dummy data
>>> x = OrderedDict()
>>> x['feat0'] = torch.rand(1, 10, 64, 64)
>>> x['feat2'] = torch.rand(1, 20, 16, 16)
>>> x['feat3'] = torch.rand(1, 30, 8, 8)
>>> # compute the FPN on top of x
>>> output = m(x)
>>> print([(k, v.shape) for k, v in output.items()])
>>> # returns
>>>   [('feat0', torch.Size([1, 5, 64, 64])),
>>>    ('feat2', torch.Size([1, 5, 16, 16])),
>>>    ('feat3', torch.Size([1, 5, 8, 8]))]