Update ShuffleResnet.py

MVIG-SJTU · Nov 24, 2021 · bcfbc99 · bcfbc99
1 parent ddaf4b9
commit bcfbc99
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/alphapose/models/layers/ShuffleResnet.py b/alphapose/models/layers/ShuffleResnet.py
@@ -6,7 +6,6 @@
 import torch.nn as nn
 import torch.nn.functional as F
 
-from .dcn import DCN
 from .PixelUnshuffle import PixelUnshuffle
 from .SE_module import SELayer
 
@@ -81,6 +80,7 @@ def __init__(self, inplanes, planes, stride=1,
                 conv_layers.append(nn.Conv2d(planes * 4, planes, kernel_size=3, stride=1,
                                              padding=1, bias=False))
             else:
+                from .dcn import DCN
                 conv_layers.append(DCN(planes * 4, planes, dcn, kernel_size=3, stride=1,
                                        padding=1, bias=False))
             self.conv2 = nn.Sequential(*conv_layers)
@@ -89,6 +89,7 @@ def __init__(self, inplanes, planes, stride=1,
                 self.conv2 = nn.Conv2d(planes, planes, kernel_size=3, stride=stride,
                                        padding=1, bias=False)
             else:
+                from .dcn import DCN
                 self.conv2 = DCN(planes, planes, dcn, kernel_size=3, stride=stride,
                                  padding=1, bias=False)