HeatmapHead
===========

.. currentmodule:: lightning_pose.models.heads

.. autoclass:: HeatmapHead
   :show-inheritance:

   .. rubric:: Methods Summary

   .. autosummary::

      ~HeatmapHead.forward
      ~HeatmapHead.run_subpixelmaxima

   .. rubric:: Methods Documentation

   .. automethod:: forward
   .. automethod:: run_subpixelmaxima