:py:mod:`rofunc.utils.visualab.segment.sam_seg`
===============================================

.. py:module:: rofunc.utils.visualab.segment.sam_seg

.. autodoc2-docstring:: rofunc.utils.visualab.segment.sam_seg
   :allowtitles:

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

Functions
~~~~~~~~~

.. list-table::
   :class: autosummary longtable
   :align: left

   * - :py:obj:`sam_generate <rofunc.utils.visualab.segment.sam_seg.sam_generate>`
     - .. autodoc2-docstring:: rofunc.utils.visualab.segment.sam_seg.sam_generate
          :summary:
   * - :py:obj:`sam_predict <rofunc.utils.visualab.segment.sam_seg.sam_predict>`
     - .. autodoc2-docstring:: rofunc.utils.visualab.segment.sam_seg.sam_predict
          :summary:

API
~~~

.. py:function:: sam_generate(image, sam_checkpoint='sam_vit_h_4b8939.pth', model_type='vit_h')
   :canonical: rofunc.utils.visualab.segment.sam_seg.sam_generate

   .. autodoc2-docstring:: rofunc.utils.visualab.segment.sam_seg.sam_generate

.. py:function:: sam_predict(image, use_point=False, use_box=False, choose_best_mask=True, sam_checkpoint='sam_vit_h_4b8939.pth', model_type='vit_h')
   :canonical: rofunc.utils.visualab.segment.sam_seg.sam_predict

   .. autodoc2-docstring:: rofunc.utils.visualab.segment.sam_seg.sam_predict