objectscope.model_export_utils
==============================

.. py:module:: objectscope.model_export_utils


Attributes
----------

.. autoapisummary::

   objectscope.model_export_utils.fields


Classes
-------

.. autoapisummary::

   objectscope.model_export_utils.OnnxModelExporter


Module Contents
---------------

.. py:data:: fields

.. py:class:: OnnxModelExporter(cfg_path, model_path, registered_dataset_name)

   Bases: :py:obj:`object`


   .. py:attribute:: model_path


   .. py:attribute:: registered_dataset_name


   .. py:method:: get_traceadapted_model(model: Union[detectron2.engine.DefaultPredictor, None] = None, inputs=None)


   .. py:method:: get_predictor(cfg=None)


   .. py:method:: get_sample_model_inputs(cfg=None, registered_dataset_name=None)


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


   .. py:method:: create_script_model(model: Union[detectron2.engine.DefaultPredictor, None] = None, fields=fields)


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