:py:mod:`rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn`
===============================================================

.. py:module:: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn

.. autodoc2-docstring:: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn
   :allowtitles:

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

Classes
~~~~~~~

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

   * - :py:obj:`TexeEmbedClassifier <rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.TexeEmbedClassifier>`
     -
   * - :py:obj:`VLMFastRCNNOutputLayers <rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.VLMFastRCNNOutputLayers>`
     -

API
~~~

.. py:class:: TexeEmbedClassifier(input_shape: detectron2.layers.ShapeSpec, zs_weight_dim: int = 1024, norm_weight: bool = True, norm_temperature: float = 50.0)
   :canonical: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.TexeEmbedClassifier

   Bases: :py:obj:`torch.nn.Module`

   .. py:method:: forward(x, text_embed)
      :canonical: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.TexeEmbedClassifier.forward

      .. autodoc2-docstring:: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.TexeEmbedClassifier.forward

.. py:class:: VLMFastRCNNOutputLayers(input_shape: detectron2.layers.ShapeSpec, box2box_transform, use_sigmoid_ce: bool = True, test_score_thresh: float = 0.0, test_nms_thresh: float = 0.5, test_topk_per_image: int = 100)
   :canonical: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.VLMFastRCNNOutputLayers

   Bases: :py:obj:`torch.nn.Module`

   .. py:method:: forward(x, text_embed)
      :canonical: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.VLMFastRCNNOutputLayers.forward

      .. autodoc2-docstring:: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.VLMFastRCNNOutputLayers.forward

   .. py:method:: predict_boxes(predictions, proposals)
      :canonical: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.VLMFastRCNNOutputLayers.predict_boxes

      .. autodoc2-docstring:: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.VLMFastRCNNOutputLayers.predict_boxes

   .. py:method:: predict_probs(predictions, proposals)
      :canonical: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.VLMFastRCNNOutputLayers.predict_probs

      .. autodoc2-docstring:: rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.VLMFastRCNNOutputLayers.predict_probs