apply_overrides(cfg, args.opts) default_setup(cfg, args) if args.eval_only: # 实例化模型 model = instantiate(cfg.model) model.to(cfg.train.device) model = create_ddp_model(model) DetectionCheckpointer(model).load(cfg.train.init_checkpoint) print(do_test(cfg, model)) else: # 实例化部分...
default_setup(cfg, args):default_setup是detectron2/engine/default.py中提供的一个默认配置函数,具体是怎么配置的这里不详细说明了。不过需要知道的值这个文件中还提供了很多其他的配置函数,例如还提供了两个类:DefaultPredictor和DefaultTrainer。 Trainer...
from detectron2.engine import SimpleTrainer, DefaultTrainer, default_argument_parser, default_setup, launch from detectron2.evaluation import COCOEvaluator, verify_results from detectron2.solver.build import maybe_add_gradient_clipping from detectron2.utils.logger import setup_logger from detectron2.data....
default_setup(cfg, args) return cfg def main(args): cfg = setup(args) Register().register_dataset() # register my dataset if args.eval_only: model = Trainer.build_model(cfg) DetectionCheckpointer(model, save_dir=cfg.OUTPUT_DIR).resume_or_load( cfg.MODEL.WEIGHTS, resume=args.resume ) ...
jsonfromdetectron2.engineimportDefaultTrainer, default_argument_parser, default_setup, hooks, launchfromdetectron2.evaluationimport(CityscapesInstanceEvaluator,CityscapesSemSegEvaluator,COCOEvaluator,COCOPanopticEvaluator,DatasetEvaluators,LVISEvaluator,PascalVOCDetectionEvaluator,SemSegEvaluator,verify_results,)from...
1.3setup_cfg(args)函数 前面get_parser()函数返回的是一个参数解析器,主函数调用.parse_args()进行参数解析后得到args是一个namespace,指定了对应的配置文件,还需要将配置文件的参数进行解析。 首先构建了一个默认的CfgNode对象,主要用来存放基本的网络参数配置,通过调用.merge_from_file()和.merge_from_list()两...
detectron2from detectron2.utils.logger import setup_loggersetup_logger()# import some common librariesimport numpy as npimport cv2import randomfrom google.colab.patches import cv2_imshow# import some common detectron2 utilitiesfrom detectron2 import model_zoofrom detectron2.engine import DefaultPredictor...
zoofromdetectron2.engineimportDefaultPredictor#from detectron2.utils.logger import setup_logger#setup_...
detectron2.engine.default.DefaultTrainer在其__init__(self, cfg)函数中定义了解析cfg。如下面代码所示,cfg会作为参数倍若干个build_*方法解析,得到解析后的model,optimizer,data_loader等。 from detectron2.modeling import build_model class DefaultTrainer(SimpleTrainer): def __init__(self, cfg): """ ...
default_setup(cfg, args):default_setup是detectron2/engine/default.py中提供的一个默认配置函数,具体是怎么配置的这里不详细说明了。不过需要知道的值这个文件中还提供了很多其他的配置函数,例如还提供了两个类:DefaultPredictor和DefaultTrainer。 Trainer