add CondInst

adet/config/defaults.py

@@ -199,6 +199,29 @@
 # Whether to compute loss on original mask (binary mask).
 _C.MODEL.MEInst.LOSS_ON_MASK = False
 
+# ---------------------------------------------------------------------------- #
+# CondInst Options
+# ---------------------------------------------------------------------------- #
+_C.MODEL.CONDINST = CN()
+
+# the downsampling ratio of the final instance masks to the input image
+_C.MODEL.CONDINST.MASK_OUT_STRIDE = 4
+_C.MODEL.CONDINST.MAX_PROPOSALS = -1
+
+_C.MODEL.CONDINST.MASK_HEAD = CN()
+_C.MODEL.CONDINST.MASK_HEAD.CHANNELS = 8
+_C.MODEL.CONDINST.MASK_HEAD.NUM_LAYERS = 3
+_C.MODEL.CONDINST.MASK_HEAD.USE_FP16 = False
+_C.MODEL.CONDINST.MASK_HEAD.DISABLE_REL_COORDS = False
+
+_C.MODEL.CONDINST.MASK_BRANCH = CN()
+_C.MODEL.CONDINST.MASK_BRANCH.OUT_CHANNELS = 8
+_C.MODEL.CONDINST.MASK_BRANCH.IN_FEATURES = ["p3", "p4", "p5"]
+_C.MODEL.CONDINST.MASK_BRANCH.CHANNELS = 128
+_C.MODEL.CONDINST.MASK_BRANCH.NORM = "BN"
+_C.MODEL.CONDINST.MASK_BRANCH.NUM_CONVS = 4
+_C.MODEL.CONDINST.MASK_BRANCH.SEMANTIC_LOSS_ON = False
+
 # ---------------------------------------------------------------------------- #
 # TOP Module Options
 # ---------------------------------------------------------------------------- #

adet/modeling/__init__.py

@@ -6,6 +6,7 @@
 from .roi_heads.text_head import TextHead
 from .batext import BAText
 from .MEInst import MEInst
+from .condinst import condinst
 
 _EXCLUDE = {"torch", "ShapeSpec"}
 __all__ = [k for k in globals().keys() if k not in _EXCLUDE and not k.startswith("_")]

adet/modeling/backbone/__init__.py

@@ -2,3 +2,4 @@
 from .vovnet import build_vovnet_fpn_backbone, build_vovnet_backbone
 from .dla import build_fcos_dla_fpn_backbone
 from .resnet_lpf import build_resnet_lpf_backbone
+from .bifpn import build_fcos_resnet_bifpn_backbone