Pytorch 从Detectron2';COCOEvaluator
标准输出的度量比输出文件夹中的输出有用得多Pytorch 从Detectron2';COCOEvaluator,pytorch,Pytorch,标准输出的度量比输出文件夹中的输出有用得多 标准输出 metrics.json和coco\u实例\u结果.json 我如何获得ioU=0.5或每个类别bbox AP的平均感知等详细信息 def test(instance, cfg, trainer, test_instance): cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final.pth") cfg.MODEL.ROI_HEADS.SCORE_THRESH
标准输出
metrics.json
和coco\u实例\u结果.json
我如何获得ioU=0.5或每个类别bbox AP的平均感知等详细信息
def test(instance, cfg, trainer, test_instance):
cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.7 # TODO What is this
cfg.DATASETS.TEST = (test_instance, )
predictor = DefaultPredictor(cfg)
evaluator = COCOEvaluator(test_instance, cfg, False, output_dir=cfg.OUTPUT_DIR)
val_loader = build_detection_test_loader(cfg, test_instance)
result = inference_on_dataset(trainer.model, val_loader, evaluator)