handle DCN import error

rodnet/models/rodnet_cdc_v2.py

@@ -2,7 +2,11 @@ import torch.nn as nn
 
 from .backbones.cdc import RadarVanilla
 from .modules.mnet import MNet
-from ..ops.dcn import DeformConvPack3D
+
+try:
+    from ..ops.dcn import DeformConvPack3D
+except:
+    print("Warning: DCN modules are not correctly imported!")
 
 
 class RODNetCDCDCN(nn.Module):

rodnet/models/rodnet_hg_v2.py

@@ -3,7 +3,11 @@ import torch.nn as nn
 
 from .backbones.hg import RadarStackedHourglass
 from .modules.mnet import MNet
-from ..ops.dcn import DeformConvPack3D
+
+try:
+    from ..ops.dcn import DeformConvPack3D
+except:
+    print("Warning: DCN modules are not correctly imported!")
 
 
 class RODNetHGDCN(nn.Module):

rodnet/models/rodnet_hgwi_v2.py

@@ -3,7 +3,11 @@ import torch.nn as nn
 
 from .backbones.hgwi import RadarStackedHourglass
 from .modules.mnet import MNet
-from ..ops.dcn import DeformConvPack3D
+
+try:
+    from ..ops.dcn import DeformConvPack3D
+except:
+    print("Warning: DCN modules are not correctly imported!")
 
 
 class RODNetHGwIDCN(nn.Module):

rodnet/ops/dcn/__init__.py

-from .deform_conv_2d import DeformConv2D, DeformConvPack2D
-from .deform_conv_2d import ModulatedDeformConv2D, ModulatedDeformConvPack2D
-from .deform_pool_2d import DeformRoIPooling2D, DeformRoIPoolingPack2D
-from .deform_pool_2d import ModulatedDeformRoIPoolingPack2D
-from .deform_conv_3d import DeformConv3D, DeformConvPack3D
-from .deform_conv_3d import ModulatedDeformConv3D, ModulatedDeformConvPack3D
-# from .deform_pool_3d import DeformRoIPooling3D, DeformRoIPoolingPack3D
-# from .deform_pool_3d import ModulatedDeformRoIPoolingPack3D
+try:
+    from .deform_conv_2d import DeformConv2D, DeformConvPack2D
+    from .deform_conv_2d import ModulatedDeformConv2D, ModulatedDeformConvPack2D
+    from .deform_pool_2d import DeformRoIPooling2D, DeformRoIPoolingPack2D
+    from .deform_pool_2d import ModulatedDeformRoIPoolingPack2D
+    from .deform_conv_3d import DeformConv3D, DeformConvPack3D
+    from .deform_conv_3d import ModulatedDeformConv3D, ModulatedDeformConvPack3D
+    # from .deform_pool_3d import DeformRoIPooling3D, DeformRoIPoolingPack3D
+    # from .deform_pool_3d import ModulatedDeformRoIPoolingPack3D
+except:
+    print("Warning: DCN modules are not correctly imported!")