未验证 提交 c064334a 编写于 作者: W wjm 提交者: GitHub

fix_fpn_senet (#6791) (#6794)

上级 d02bcd93
...@@ -16,6 +16,8 @@ import paddle.nn as nn ...@@ -16,6 +16,8 @@ import paddle.nn as nn
from ppdet.core.workspace import register, serializable from ppdet.core.workspace import register, serializable
from .resnet import ResNet, Blocks, BasicBlock, BottleNeck from .resnet import ResNet, Blocks, BasicBlock, BottleNeck
from ..shape_spec import ShapeSpec
from .name_adapter import NameAdapter
__all__ = ['SENet', 'SERes5Head'] __all__ = ['SENet', 'SERes5Head']
......
...@@ -33,7 +33,7 @@ class FPN(nn.Layer): ...@@ -33,7 +33,7 @@ class FPN(nn.Layer):
Args: Args:
in_channels (list[int]): input channels of each level which can be in_channels (list[int]): input channels of each level which can be
derived from the output shape of backbone by from_config 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 spatial_scales (list[float]): the spatial scales between input feature
maps and original input image which can be derived from the output maps and original input image which can be derived from the output
shape of backbone by from_config shape of backbone by from_config
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册