Detectron2 trained model load (architecture and weights) from config and checkpoints

 

 

1. build model with config.yaml file

2. load weights with model.pth file

import torch

from detectron2.config import get_cfg
from detectron2.modeling import build_model
from detectron2.checkpoint import DetectionCheckpointer

config_path = 'config.yaml'
model_path = 'model_final.pth'

cfg = get_cfg()
cfg.merge_from_file(config_path)

model = build_model(cfg)
DetectionCheckpointer(model).load(model_path)