未验证 提交 118e8b75 编写于 作者: W wangxinxin08 提交者: GitHub

fix data format in ppyolo (#2403)

上级 6716d44c
...@@ -313,6 +313,7 @@ class PPYOLOFPN(nn.Layer): ...@@ -313,6 +313,7 @@ class PPYOLOFPN(nn.Layer):
self.spp = kwargs.get('spp', False) self.spp = kwargs.get('spp', False)
self.conv_block_num = kwargs.get('conv_block_num', 2) self.conv_block_num = kwargs.get('conv_block_num', 2)
self.data_format = data_format
if self.coord_conv: if self.coord_conv:
ConvLayer = CoordConv ConvLayer = CoordConv
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册