HeatmapDataset
==============

.. currentmodule:: lightning_pose.data.datasets

.. autoclass:: HeatmapDataset
   :show-inheritance:

   .. rubric:: Attributes Summary

   .. autosummary::

      ~HeatmapDataset.output_shape

   .. rubric:: Methods Summary

   .. autosummary::

      ~HeatmapDataset.compute_heatmap
      ~HeatmapDataset.compute_heatmaps

   .. rubric:: Attributes Documentation

   .. autoattribute:: output_shape

   .. rubric:: Methods Documentation

   .. automethod:: compute_heatmap
   .. automethod:: compute_heatmaps