Arulkumar03/Wheat_HEAD_Detection_Counting_ComputerVision_Model

1# -*- coding: utf-8 -*-2# Copyright (c) Facebook, Inc. and its affiliates.3 4from detectron2.config import CfgNode as CN5 6 7def add_tridentnet_config(cfg):8    """9    Add config for tridentnet.10    """11    _C = cfg12 13    _C.MODEL.TRIDENT = CN()14 15    # Number of branches for TridentNet.16    _C.MODEL.TRIDENT.NUM_BRANCH = 317    # Specify the dilations for each branch.18    _C.MODEL.TRIDENT.BRANCH_DILATIONS = [1, 2, 3]19    # Specify the stage for applying trident blocks. Default stage is Res4 according to the20    # TridentNet paper.21    _C.MODEL.TRIDENT.TRIDENT_STAGE = "res4"22    # Specify the test branch index TridentNet Fast inference:23    #   - use -1 to aggregate results of all branches during inference.24    #   - otherwise, only using specified branch for fast inference. Recommended setting is25    #     to use the middle branch.26    _C.MODEL.TRIDENT.TEST_BRANCH_IDX = 127