o ĎiR@s.ddlmZGdddZd defddZdS) )Pathc@seZdZdZdZdZdS)Detectron2TestConstantsz+COCO-Detection/faster_rcnn_R_50_FPN_3x.yamlz)COCO-Detection/retinanet_R_50_FPN_3x.yamlz4COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yamlN)__name__ __module__ __qualname__FASTERCNN_MODEL_ZOO_NAMERETINANET_MODEL_ZOO_NAMEMASKRCNN_MODEL_ZOO_NAMEr r Q/home/jeff/fluffinator/venv/lib/python3.10/site-packages/sahi/utils/detectron2.pyrsr config.yaml export_pathcCsRt|jjdddt|d}||WddS1s"wYdS)a\Exports Detectron2 config object in yaml format so that it can be used later. Args: cfg (detectron2.config.CfgNode): Detectron2 config object. export_path (str): Path to export the Detectron2 config. Related Detectron2 doc: https://detectron2.readthedocs.io/en/stable/modules/config.html#detectron2.config.CfgNode.dump T)exist_okparentswN)rparentmkdiropenwritedump)cfgr fr r r export_cfg_as_yaml s "rN)r )pathlibrrstrrr r r r s