You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
92 lines
3.4 KiB
92 lines
3.4 KiB
from detectron2.config.defaults import _C |
|
from detectron2.config import CfgNode as CN |
|
|
|
|
|
# ---------------------------------------------------------------------------- # |
|
# Additional Configs |
|
# ---------------------------------------------------------------------------- # |
|
_C.MODEL.MOBILENET = False |
|
_C.MODEL.BACKBONE.ANTI_ALIAS = False |
|
_C.MODEL.RESNETS.DEFORM_INTERVAL = 1 |
|
|
|
# ---------------------------------------------------------------------------- # |
|
# FCOS Head |
|
# ---------------------------------------------------------------------------- # |
|
_C.MODEL.FCOS = CN() |
|
|
|
# This is the number of foreground classes. |
|
_C.MODEL.FCOS.NUM_CLASSES = 80 |
|
_C.MODEL.FCOS.IN_FEATURES = ["p3", "p4", "p5", "p6", "p7"] |
|
_C.MODEL.FCOS.FPN_STRIDES = [8, 16, 32, 64, 128] |
|
_C.MODEL.FCOS.PRIOR_PROB = 0.01 |
|
_C.MODEL.FCOS.INFERENCE_TH_TRAIN = 0.05 |
|
_C.MODEL.FCOS.INFERENCE_TH_TEST = 0.05 |
|
_C.MODEL.FCOS.NMS_TH = 0.6 |
|
_C.MODEL.FCOS.PRE_NMS_TOPK_TRAIN = 1000 |
|
_C.MODEL.FCOS.PRE_NMS_TOPK_TEST = 1000 |
|
_C.MODEL.FCOS.POST_NMS_TOPK_TRAIN = 100 |
|
_C.MODEL.FCOS.POST_NMS_TOPK_TEST = 100 |
|
_C.MODEL.FCOS.TOP_LEVELS = 2 |
|
_C.MODEL.FCOS.NORM = "GN" # Support GN or none |
|
_C.MODEL.FCOS.USE_SCALE = True |
|
|
|
# Multiply centerness before threshold |
|
# This will affect the final performance by about 0.05 AP but save some time |
|
_C.MODEL.FCOS.THRESH_WITH_CTR = False |
|
|
|
# Focal loss parameters |
|
_C.MODEL.FCOS.LOSS_ALPHA = 0.25 |
|
_C.MODEL.FCOS.LOSS_GAMMA = 2.0 |
|
_C.MODEL.FCOS.SIZES_OF_INTEREST = [64, 128, 256, 512] |
|
_C.MODEL.FCOS.USE_RELU = True |
|
_C.MODEL.FCOS.USE_DEFORMABLE = False |
|
|
|
# the number of convolutions used in the cls and bbox tower |
|
_C.MODEL.FCOS.NUM_CLS_CONVS = 4 |
|
_C.MODEL.FCOS.NUM_BOX_CONVS = 4 |
|
_C.MODEL.FCOS.NUM_SHARE_CONVS = 0 |
|
_C.MODEL.FCOS.CENTER_SAMPLE = True |
|
_C.MODEL.FCOS.POS_RADIUS = 1.5 |
|
_C.MODEL.FCOS.LOC_LOSS_TYPE = 'giou' |
|
_C.MODEL.FCOS.YIELD_PROPOSAL = False |
|
|
|
# ---------------------------------------------------------------------------- # |
|
# VoVNet backbone |
|
# ---------------------------------------------------------------------------- # |
|
_C.MODEL.VOVNET = CN() |
|
_C.MODEL.VOVNET.CONV_BODY = "V-39-eSE" |
|
_C.MODEL.VOVNET.OUT_FEATURES = ["stage2", "stage3", "stage4", "stage5"] |
|
|
|
# Options: FrozenBN, GN, "SyncBN", "BN" |
|
_C.MODEL.VOVNET.NORM = "FrozenBN" |
|
_C.MODEL.VOVNET.OUT_CHANNELS = 256 |
|
_C.MODEL.VOVNET.BACKBONE_OUT_CHANNELS = 256 |
|
|
|
# ---------------------------------------------------------------------------- # |
|
# BlendMask Options |
|
# ---------------------------------------------------------------------------- # |
|
_C.MODEL.BLENDMASK = CN() |
|
_C.MODEL.BLENDMASK.ATTN_SIZE = 14 |
|
_C.MODEL.BLENDMASK.TOP_INTERP = "bilinear" |
|
_C.MODEL.BLENDMASK.BOTTOM_RESOLUTION = 56 |
|
_C.MODEL.BLENDMASK.POOLER_TYPE = "ROIAlignV2" |
|
_C.MODEL.BLENDMASK.POOLER_SAMPLING_RATIO = 1 |
|
_C.MODEL.BLENDMASK.POOLER_SCALES = (0.25,) |
|
_C.MODEL.BLENDMASK.INSTANCE_LOSS_WEIGHT = 1.0 |
|
_C.MODEL.BLENDMASK.VISUALIZE = False |
|
|
|
# ---------------------------------------------------------------------------- # |
|
# Basis Module Options |
|
# ---------------------------------------------------------------------------- # |
|
_C.MODEL.BASIS_MODULE = CN() |
|
_C.MODEL.BASIS_MODULE.NAME = "ProtoNet" |
|
_C.MODEL.BASIS_MODULE.NUM_BASES = 4 |
|
_C.MODEL.BASIS_MODULE.LOSS_ON = False |
|
_C.MODEL.BASIS_MODULE.ANN_SET = "coco" |
|
_C.MODEL.BASIS_MODULE.CONVS_DIM = 128 |
|
_C.MODEL.BASIS_MODULE.IN_FEATURES = ["p3", "p4", "p5"] |
|
_C.MODEL.BASIS_MODULE.NORM = "SyncBN" |
|
_C.MODEL.BASIS_MODULE.NUM_CONVS = 3 |
|
_C.MODEL.BASIS_MODULE.COMMON_STRIDE = 8 |
|
_C.MODEL.BASIS_MODULE.NUM_CLASSES = 80 |
|
_C.MODEL.BASIS_MODULE.LOSS_WEIGHT = 0.3 |