MaskDINO icon indicating copy to clipboard operation
MaskDINO copied to clipboard

How to export maskdino into onnx format?

Open MeAmarP opened this issue 2 years ago • 7 comments

Hi Team, After initial evaluation am planning to export model into onnx format using detectron2 deploy tools.

But, Am running into following error, on build_model(cfg) call. KeyError: "No object named 'MaskDINO' found in 'META_ARCH' registry!"

I made following changes in setup_cfg() method of export_model.py

def setup_cfg(args):
    cfg = get_cfg()
    # cuda context is initialized before creating dataloader, so we don't fork anymore
    cfg.DATALOADER.NUM_WORKERS = 0
    add_pointrend_config(cfg)
    add_deeplab_config(cfg)                            # <---
    add_maskdino_config(cfg)                         # <---
    cfg.merge_from_file(args.config_file)
    cfg.merge_from_list(args.opts)
    cfg.freeze()
    return cfg

Is there any way I can resolve this? Please guide me here.

MeAmarP avatar Jan 11 '23 13:01 MeAmarP