fix: add more fallbacks for mmpkg (#510)

annotator/mmpkg/mmseg/models/decode_heads/cc_head.py

@@ -4,7 +4,10 @@ from ..builder import HEADS
 from .fcn_head import FCNHead
 
 try:
-    from annotator.mmpkg.mmcv.ops import CrissCrossAttention
+    try: 
+        from mmcv.ops import CrissCrossAttention
+    except ImportError:
+        from annotator.mmpkg.mmcv.ops import CrissCrossAttention
 except ModuleNotFoundError:
     CrissCrossAttention = None
 

annotator/mmpkg/mmseg/models/decode_heads/point_head.py

@@ -2,8 +2,13 @@
 
 import torch
 import torch.nn as nn
-from annotator.mmpkg.mmcv.cnn import ConvModule, normal_init
-from annotator.mmpkg.mmcv.ops import point_sample
+
+try: 
+    from mmcv.cnn import ConvModule, normal_init
+    from mmcv.ops import point_sample
+except ImportError:
+    from annotator.mmpkg.mmcv.cnn import ConvModule, normal_init
+    from annotator.mmpkg.mmcv.ops import point_sample
 
 from annotator.mmpkg.mmseg.models.builder import HEADS
 from annotator.mmpkg.mmseg.ops import resize

annotator/mmpkg/mmseg/models/decode_heads/psa_head.py

@@ -8,7 +8,10 @@ from ..builder import HEADS
 from .decode_head import BaseDecodeHead
 
 try:
-    from annotator.mmpkg.mmcv.ops import PSAMask
+    try: 
+        from mmcv.ops import PSAMask
+    except ImportError:
+        from annotator.mmpkg.mmcv.ops import PSAMask
 except ModuleNotFoundError:
     PSAMask = None