objectscope.model_export_utils

Attributes

Classes

OnnxModelExporter

Module Contents

objectscope.model_export_utils.fields

class objectscope.model_export_utils.OnnxModelExporter(cfg_path, model_path, registered_dataset_name)

Bases: object

model_path

registered_dataset_name

get_traceadapted_model(model: detectron2.engine.DefaultPredictor | None = None, inputs=None)

get_predictor(cfg=None)

get_sample_model_inputs(cfg=None, registered_dataset_name=None)

export_to_onnx(save_onnx_as, inputs=None, model=None, torchscript_formart: Literal['trace', 'script'] = 'trace')

create_script_model(model: detectron2.engine.DefaultPredictor | None = None, fields=fields)

save_model(format=Literal['trace', 'script'], model: detectron2.engine.DefaultPredictor | None = None, inputs=None, save_model_as='model.pt')