Models
======

This section documents the various scene graph generation models implemented in the project.

STTran Model
------------

.. automodule:: m3sgg.core.models.sttran.sttran
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.detectors.easg.sttran_EASG
   :members:
   :undoc-members:
   :show-inheritance:

SceneLLM Model
--------------

.. automodule:: m3sgg.core.models.scenellm.scenellm
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.scenellm.network
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.scenellm.llm
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.scenellm.vqvae
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.scenellm.ot
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.scenellm.sia
   :members:
   :undoc-members:
   :show-inheritance:

Tempura Model
-------------

.. automodule:: m3sgg.core.models.tempura.tempura
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.tempura.transformer_tempura
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.tempura.gmm_heads
   :members:
   :undoc-members:
   :show-inheritance:

STKET Model
-----------

.. automodule:: m3sgg.core.models.stket.stket
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.stket.transformer_stket
   :members:
   :undoc-members:
   :show-inheritance:

OED Model
---------

.. automodule:: m3sgg.core.models.oed.oed_multi
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.oed.oed_single
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.oed.transformer
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.oed.criterion
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.oed.postprocess
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: m3sgg.core.models.oed.utils
   :members:
   :undoc-members:
   :show-inheritance:

VLM Model
---------

.. automodule:: m3sgg.core.models.vlm.scene_graph_generator
   :members:
   :undoc-members:
   :show-inheritance:

Transformer Components
----------------------

.. automodule:: m3sgg.utils.transformer
   :members:
   :undoc-members:
   :show-inheritance: