mirror of https://github.com/RE-OWOD/RE-OWOD
27 lines
900 B
Python
27 lines
900 B
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
|
|
|
|
from detectron2.config import CfgNode as CN
|
|
|
|
|
|
def add_tridentnet_config(cfg):
|
|
"""
|
|
Add config for tridentnet.
|
|
"""
|
|
_C = cfg
|
|
|
|
_C.MODEL.TRIDENT = CN()
|
|
|
|
# Number of branches for TridentNet.
|
|
_C.MODEL.TRIDENT.NUM_BRANCH = 3
|
|
# Specify the dilations for each branch.
|
|
_C.MODEL.TRIDENT.BRANCH_DILATIONS = [1, 2, 3]
|
|
# Specify the stage for applying trident blocks. Default stage is Res4 according to the
|
|
# TridentNet paper.
|
|
_C.MODEL.TRIDENT.TRIDENT_STAGE = "res4"
|
|
# Specify the test branch index TridentNet Fast inference:
|
|
# - use -1 to aggregate results of all branches during inference.
|
|
# - otherwise, only using specified branch for fast inference. Recommended setting is
|
|
# to use the middle branch.
|
|
_C.MODEL.TRIDENT.TEST_BRANCH_IDX = 1
|