MaskDINO
MaskDINO copied to clipboard
How to export maskdino into onnx format?
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.