提交 055b92c0 编写于 作者: W wubinghong

Refine the bifpn for efficientnet-d0

上级 f4a3856f
...@@ -206,7 +206,6 @@ class BiFPN(object): ...@@ -206,7 +206,6 @@ class BiFPN(object):
feats.append(feat) feats.append(feat)
# Build additional input features that are not from backbone. # Build additional input features that are not from backbone.
# P_7 layer we just use pool2d without conv layer & bn, for the same channel with P_6. # P_7 layer we just use pool2d without conv layer & bn, for the same channel with P_6.
# https://github.com/google/automl/blob/master/efficientdet/keras/efficientdet_keras.py#L820
for idx in range(len(inputs), self.levels): for idx in range(len(inputs), self.levels):
if feats[-1].shape[1] != self.num_chan: if feats[-1].shape[1] != self.num_chan:
feat = fluid.layers.conv2d( feat = fluid.layers.conv2d(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册