diff --git a/ppdet/modeling/backbones/senet.py b/ppdet/modeling/backbones/senet.py index eb0bad33f2def79189b647dd7723f1fad5ae90e1..db1e29b9779aef5fc8ccd4ca2f4cffeedee715f1 100644 --- a/ppdet/modeling/backbones/senet.py +++ b/ppdet/modeling/backbones/senet.py @@ -16,6 +16,8 @@ import paddle.nn as nn from ppdet.core.workspace import register, serializable from .resnet import ResNet, Blocks, BasicBlock, BottleNeck +from ..shape_spec import ShapeSpec +from .name_adapter import NameAdapter __all__ = ['SENet', 'SERes5Head'] diff --git a/ppdet/modeling/necks/fpn.py b/ppdet/modeling/necks/fpn.py index 0633fb5b2713c5ddc1c1c2e03ce3dd7eec90dbcd..d08ca415c7acce7d4495dae5834d53961ea9df57 100644 --- a/ppdet/modeling/necks/fpn.py +++ b/ppdet/modeling/necks/fpn.py @@ -33,7 +33,7 @@ class FPN(nn.Layer): Args: in_channels (list[int]): input channels of each level which can be derived from the output shape of backbone by from_config - out_channel (list[int]): output channel of each level + out_channel (int): output channel of each level spatial_scales (list[float]): the spatial scales between input feature maps and original input image which can be derived from the output shape of backbone by from_config