Revert SPPF to SPP in r6.0

zhiqwang · Dec 1, 2021 · eea4e4d · eea4e4d
1 parent 12389a3
commit eea4e4d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yolort/models/path_aggregation_network.py b/yolort/models/path_aggregation_network.py
@@ -3,7 +3,7 @@
 
 import torch
 from torch import nn, Tensor
-from yolort.v5 import Conv, BottleneckCSP, C3, SPPF
+from yolort.v5 import Conv, BottleneckCSP, C3, SPP
 
 
 class IntermediateLevelP6(nn.Module):
@@ -106,7 +106,7 @@ def __init__(
         depth_gain = max(round(3 * depth_multiple), 1)
 
         if version == "r6.0":
-            init_block = SPPF(in_channels[-1], in_channels[-1], k=5)
+            init_block = SPP(in_channels[-1], in_channels[-1], k=(5, 9, 13))
         elif version in ["r3.1", "r4.0"]:
             init_block = block(in_channels[-1], in_channels[-1], n=depth_gain, shortcut=False)
         else: