diff --git a/cyto_dl.api.html b/cyto_dl.api.html
index f10eb0c2..d6d26f1d 100644
--- a/cyto_dl.api.html
+++ b/cyto_dl.api.html
@@ -441,8 +441,8 @@
           </label>
         </div>
         <article role="main">
-          <section id="module-cyto_dl.api">
-<span id="cyto-dl-api-package"></span><h1>cyto_dl.api package<a class="headerlink" href="#module-cyto_dl.api" title="Permalink to this heading">#</a></h1>
+          <section id="cyto-dl-api-package">
+<h1>cyto_dl.api package<a class="headerlink" href="#cyto-dl-api-package" title="Permalink to this heading">#</a></h1>
 <section id="subpackages">
 <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to this heading">#</a></h2>
 <div class="toctree-wrapper compound">
diff --git a/genindex.html b/genindex.html
index 80ea69e1..92a091d8 100644
--- a/genindex.html
+++ b/genindex.html
@@ -612,13 +612,6 @@ <h2>C</h2>
 
         <ul>
           <li><a href="cyto_dl.html#module-cyto_dl">module</a>
-</li>
-        </ul></li>
-        <li>
-    cyto_dl.api
-
-        <ul>
-          <li><a href="cyto_dl.api.html#module-cyto_dl.api">module</a>
 </li>
         </ul></li>
         <li>
@@ -1020,8 +1013,6 @@ <h2>C</h2>
           <li><a href="cyto_dl.models.classification.classification.html#module-cyto_dl.models.classification.classification">module</a>
 </li>
         </ul></li>
-    </ul></td>
-    <td style="width: 33%; vertical-align: top;"><ul>
         <li>
     cyto_dl.models.classification.timepoint_classification
 
@@ -1029,6 +1020,8 @@ <h2>C</h2>
           <li><a href="cyto_dl.models.classification.timepoint_classification.html#module-cyto_dl.models.classification.timepoint_classification">module</a>
 </li>
         </ul></li>
+    </ul></td>
+    <td style="width: 33%; vertical-align: top;"><ul>
         <li>
     cyto_dl.models.contrastive
 
@@ -2259,8 +2252,6 @@ <h2>M</h2>
 
         <ul>
           <li><a href="cyto_dl.html#module-cyto_dl">cyto_dl</a>
-</li>
-          <li><a href="cyto_dl.api.html#module-cyto_dl.api">cyto_dl.api</a>
 </li>
           <li><a href="cyto_dl.api.cyto_dl_model.html#module-cyto_dl.api.cyto_dl_model">cyto_dl.api.cyto_dl_model</a>
 </li>
diff --git a/objects.inv b/objects.inv
index 5cd7cbf5..388ce4ac 100644
Binary files a/objects.inv and b/objects.inv differ
diff --git a/py-modindex.html b/py-modindex.html
index 87496dc5..1c2735a1 100644
--- a/py-modindex.html
+++ b/py-modindex.html
@@ -457,12 +457,6 @@ <h1>Python Module Index</h1>
         <a href="cyto_dl.html#module-cyto_dl"><code class="xref">cyto_dl</code></a></td><td>
     <em></em></td>
   </tr>
-  <tr class="cg-1">
-    <td></td>
-    <td>&#160;&#160;&#160;
-        <a href="cyto_dl.api.html#module-cyto_dl.api"><code class="xref">cyto_dl.api</code></a></td><td>
-    <em></em></td>
-  </tr>
   <tr class="cg-1">
     <td></td>
     <td>&#160;&#160;&#160;
diff --git a/searchindex.js b/searchindex.js
index c76d6d5b..3c1a3086 100644
--- a/searchindex.js
+++ b/searchindex.js
@@ -1 +1 @@
-Search.setIndex({"docnames": ["api", "cyto_dl", "cyto_dl.api", "cyto_dl.api.cyto_dl_model", "cyto_dl.api.cyto_dl_model.cyto_dl_base_model", "cyto_dl.api.cyto_dl_model.segmentation_plugin_model", "cyto_dl.api.data", "cyto_dl.api.model", "cyto_dl.callbacks", "cyto_dl.callbacks.callback_utils", "cyto_dl.callbacks.csv_saver", "cyto_dl.callbacks.image_saver", "cyto_dl.callbacks.latent_walk", "cyto_dl.callbacks.layer_freeze", "cyto_dl.callbacks.model_utils", "cyto_dl.callbacks.outlier_detection", "cyto_dl.compile", "cyto_dl.dataframe", "cyto_dl.dataframe.readers", "cyto_dl.dataframe.transforms", "cyto_dl.dataframe.transforms.filter", "cyto_dl.dataframe.transforms.group_cols", "cyto_dl.dataframe.transforms.misc", "cyto_dl.dataframe.transforms.split", "cyto_dl.datamodules", "cyto_dl.datamodules.array", "cyto_dl.datamodules.data_dict", "cyto_dl.datamodules.dataframe", "cyto_dl.datamodules.dataframe.dataframe_datamodule", "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule", "cyto_dl.datamodules.dataframe.utils", "cyto_dl.datamodules.folder", "cyto_dl.datamodules.multidim_image", "cyto_dl.datamodules.smartcache", "cyto_dl.datamodules.torchvision", "cyto_dl.eval", "cyto_dl.image", "cyto_dl.image.io", "cyto_dl.image.io.aicsimage_loader", "cyto_dl.image.io.monai_bio_reader", "cyto_dl.image.io.numpy_reader", "cyto_dl.image.io.ome_zarr_reader", "cyto_dl.image.io.polygon_loader", "cyto_dl.image.io.skimage_reader", "cyto_dl.image.transforms", "cyto_dl.image.transforms.add_meta", "cyto_dl.image.transforms.bright_sampler", "cyto_dl.image.transforms.clip", "cyto_dl.image.transforms.contrastadjust", "cyto_dl.image.transforms.generate_jepa_masks", "cyto_dl.image.transforms.merge", "cyto_dl.image.transforms.multiscale_cropper", "cyto_dl.image.transforms.pad", "cyto_dl.image.transforms.project", "cyto_dl.image.transforms.rotation_mask_transform", "cyto_dl.image.transforms.save", "cyto_dl.image.transforms.so2_random_rotation", "cyto_dl.image.transforms.track_transforms", "cyto_dl.loggers", "cyto_dl.loggers.mlflow", "cyto_dl.models", "cyto_dl.models.base_model", "cyto_dl.models.basic_model", "cyto_dl.models.classification", "cyto_dl.models.classification.classification", "cyto_dl.models.classification.timepoint_classification", "cyto_dl.models.contrastive", "cyto_dl.models.contrastive.contrastive", "cyto_dl.models.handlers", "cyto_dl.models.handlers.base_handler", "cyto_dl.models.handlers.image_handler", "cyto_dl.models.handlers.load_image_patch", "cyto_dl.models.im2im", "cyto_dl.models.im2im.gan", "cyto_dl.models.im2im.multi_task", "cyto_dl.models.im2im.utils", "cyto_dl.models.im2im.utils.instance_seg", "cyto_dl.models.im2im.utils.noise_annealer", "cyto_dl.models.im2im.utils.postprocessing", "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label", "cyto_dl.models.im2im.utils.postprocessing.auto_thresh", "cyto_dl.models.im2im.utils.postprocessing.dict_to_im", "cyto_dl.models.jepa", "cyto_dl.models.jepa.ijepa", "cyto_dl.models.jepa.iwm", "cyto_dl.models.jepa.jepa_base", "cyto_dl.models.utils", "cyto_dl.models.utils.mlflow", "cyto_dl.models.utils.utils", "cyto_dl.models.vae", "cyto_dl.models.vae.base_vae", "cyto_dl.models.vae.image_canon_vae", "cyto_dl.models.vae.image_encoder", "cyto_dl.models.vae.image_vae", "cyto_dl.models.vae.latent_loss_vae", "cyto_dl.models.vae.o2_spharm_vae", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_encoder", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_vae", "cyto_dl.models.vae.point_cloud_vae", "cyto_dl.models.vae.priors", "cyto_dl.models.vae.priors.abstract_prior", "cyto_dl.models.vae.priors.gaussian", "cyto_dl.models.vae.priors.identity_prior", "cyto_dl.models.vae.priors.joint_prior", "cyto_dl.models.vae.tabular_vae", "cyto_dl.nn", "cyto_dl.nn.discriminators", "cyto_dl.nn.discriminators.multi_scale_discriminator", "cyto_dl.nn.discriminators.n_layer_discriminator", "cyto_dl.nn.head", "cyto_dl.nn.head.base_head", "cyto_dl.nn.head.gan_head", "cyto_dl.nn.head.gan_head_superres", "cyto_dl.nn.head.mae_head", "cyto_dl.nn.head.mask_head", "cyto_dl.nn.head.res_blocks_head", "cyto_dl.nn.head.vic_reg", "cyto_dl.nn.hr_skip", "cyto_dl.nn.losses", "cyto_dl.nn.losses.adversarial_loss", "cyto_dl.nn.losses.chamfer_loss", "cyto_dl.nn.losses.continuous_bernoulli", "cyto_dl.nn.losses.cosine_loss", "cyto_dl.nn.losses.gan_loss", "cyto_dl.nn.losses.gaussian_nll_loss", "cyto_dl.nn.losses.geomloss", "cyto_dl.nn.losses.loss_wrapper", "cyto_dl.nn.losses.spharm_loss", "cyto_dl.nn.losses.threshold_loss", "cyto_dl.nn.losses.vic_reg", "cyto_dl.nn.losses.weibull", "cyto_dl.nn.losses.weighted_mse_loss", "cyto_dl.nn.mlp", "cyto_dl.nn.point_cloud", "cyto_dl.nn.point_cloud.dgcnn", "cyto_dl.nn.point_cloud.folding_net", "cyto_dl.nn.point_cloud.graph_functions", "cyto_dl.nn.point_cloud.vnn", "cyto_dl.nn.res_unit", "cyto_dl.nn.spatial_transformer", "cyto_dl.nn.track_sequence_predictor", "cyto_dl.nn.vits", "cyto_dl.nn.vits.blocks", "cyto_dl.nn.vits.blocks.cross_attention", "cyto_dl.nn.vits.blocks.intermediate_weigher", "cyto_dl.nn.vits.blocks.masked_unit_attention", "cyto_dl.nn.vits.blocks.patchify", "cyto_dl.nn.vits.blocks.patchify.patchify", "cyto_dl.nn.vits.blocks.patchify.patchify_base", "cyto_dl.nn.vits.blocks.patchify.patchify_hiera", "cyto_dl.nn.vits.decoder", "cyto_dl.nn.vits.encoder", "cyto_dl.nn.vits.mae", "cyto_dl.nn.vits.predictor", "cyto_dl.nn.vits.seg", "cyto_dl.nn.vits.utils", "cyto_dl.point_cloud", "cyto_dl.point_cloud.io", "cyto_dl.point_cloud.io.read_pcloud", "cyto_dl.train", "cyto_dl.utils", "cyto_dl.utils.array", "cyto_dl.utils.checkpoint", "cyto_dl.utils.config", "cyto_dl.utils.download_test_data", "cyto_dl.utils.dummy_dataset", "cyto_dl.utils.pylogger", "cyto_dl.utils.rich_utils", "cyto_dl.utils.rotation", "cyto_dl.utils.spharm", "cyto_dl.utils.spharm.reconstruction", "cyto_dl.utils.spharm.rotation", "cyto_dl.utils.template_utils", "index", "installation", "modules", "overview", "using_examples"], "filenames": ["api.rst", "cyto_dl.rst", "cyto_dl.api.rst", "cyto_dl.api.cyto_dl_model.rst", "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.rst", "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.rst", "cyto_dl.api.data.rst", "cyto_dl.api.model.rst", "cyto_dl.callbacks.rst", "cyto_dl.callbacks.callback_utils.rst", "cyto_dl.callbacks.csv_saver.rst", "cyto_dl.callbacks.image_saver.rst", "cyto_dl.callbacks.latent_walk.rst", "cyto_dl.callbacks.layer_freeze.rst", "cyto_dl.callbacks.model_utils.rst", "cyto_dl.callbacks.outlier_detection.rst", "cyto_dl.compile.rst", "cyto_dl.dataframe.rst", "cyto_dl.dataframe.readers.rst", "cyto_dl.dataframe.transforms.rst", "cyto_dl.dataframe.transforms.filter.rst", "cyto_dl.dataframe.transforms.group_cols.rst", "cyto_dl.dataframe.transforms.misc.rst", "cyto_dl.dataframe.transforms.split.rst", "cyto_dl.datamodules.rst", "cyto_dl.datamodules.array.rst", "cyto_dl.datamodules.data_dict.rst", "cyto_dl.datamodules.dataframe.rst", "cyto_dl.datamodules.dataframe.dataframe_datamodule.rst", "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule.rst", "cyto_dl.datamodules.dataframe.utils.rst", "cyto_dl.datamodules.folder.rst", "cyto_dl.datamodules.multidim_image.rst", "cyto_dl.datamodules.smartcache.rst", "cyto_dl.datamodules.torchvision.rst", "cyto_dl.eval.rst", "cyto_dl.image.rst", "cyto_dl.image.io.rst", "cyto_dl.image.io.aicsimage_loader.rst", "cyto_dl.image.io.monai_bio_reader.rst", "cyto_dl.image.io.numpy_reader.rst", "cyto_dl.image.io.ome_zarr_reader.rst", "cyto_dl.image.io.polygon_loader.rst", "cyto_dl.image.io.skimage_reader.rst", "cyto_dl.image.transforms.rst", "cyto_dl.image.transforms.add_meta.rst", "cyto_dl.image.transforms.bright_sampler.rst", "cyto_dl.image.transforms.clip.rst", "cyto_dl.image.transforms.contrastadjust.rst", "cyto_dl.image.transforms.generate_jepa_masks.rst", "cyto_dl.image.transforms.merge.rst", "cyto_dl.image.transforms.multiscale_cropper.rst", "cyto_dl.image.transforms.pad.rst", "cyto_dl.image.transforms.project.rst", "cyto_dl.image.transforms.rotation_mask_transform.rst", "cyto_dl.image.transforms.save.rst", "cyto_dl.image.transforms.so2_random_rotation.rst", "cyto_dl.image.transforms.track_transforms.rst", "cyto_dl.loggers.rst", "cyto_dl.loggers.mlflow.rst", "cyto_dl.models.rst", "cyto_dl.models.base_model.rst", "cyto_dl.models.basic_model.rst", "cyto_dl.models.classification.rst", "cyto_dl.models.classification.classification.rst", "cyto_dl.models.classification.timepoint_classification.rst", "cyto_dl.models.contrastive.rst", "cyto_dl.models.contrastive.contrastive.rst", "cyto_dl.models.handlers.rst", "cyto_dl.models.handlers.base_handler.rst", "cyto_dl.models.handlers.image_handler.rst", "cyto_dl.models.handlers.load_image_patch.rst", "cyto_dl.models.im2im.rst", "cyto_dl.models.im2im.gan.rst", "cyto_dl.models.im2im.multi_task.rst", "cyto_dl.models.im2im.utils.rst", "cyto_dl.models.im2im.utils.instance_seg.rst", "cyto_dl.models.im2im.utils.noise_annealer.rst", "cyto_dl.models.im2im.utils.postprocessing.rst", "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label.rst", "cyto_dl.models.im2im.utils.postprocessing.auto_thresh.rst", "cyto_dl.models.im2im.utils.postprocessing.dict_to_im.rst", "cyto_dl.models.jepa.rst", "cyto_dl.models.jepa.ijepa.rst", "cyto_dl.models.jepa.iwm.rst", "cyto_dl.models.jepa.jepa_base.rst", "cyto_dl.models.utils.rst", "cyto_dl.models.utils.mlflow.rst", "cyto_dl.models.utils.utils.rst", "cyto_dl.models.vae.rst", "cyto_dl.models.vae.base_vae.rst", "cyto_dl.models.vae.image_canon_vae.rst", "cyto_dl.models.vae.image_encoder.rst", "cyto_dl.models.vae.image_vae.rst", "cyto_dl.models.vae.latent_loss_vae.rst", "cyto_dl.models.vae.o2_spharm_vae.rst", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_encoder.rst", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_vae.rst", "cyto_dl.models.vae.point_cloud_vae.rst", "cyto_dl.models.vae.priors.rst", "cyto_dl.models.vae.priors.abstract_prior.rst", "cyto_dl.models.vae.priors.gaussian.rst", "cyto_dl.models.vae.priors.identity_prior.rst", "cyto_dl.models.vae.priors.joint_prior.rst", "cyto_dl.models.vae.tabular_vae.rst", "cyto_dl.nn.rst", "cyto_dl.nn.discriminators.rst", "cyto_dl.nn.discriminators.multi_scale_discriminator.rst", "cyto_dl.nn.discriminators.n_layer_discriminator.rst", "cyto_dl.nn.head.rst", "cyto_dl.nn.head.base_head.rst", "cyto_dl.nn.head.gan_head.rst", "cyto_dl.nn.head.gan_head_superres.rst", "cyto_dl.nn.head.mae_head.rst", "cyto_dl.nn.head.mask_head.rst", "cyto_dl.nn.head.res_blocks_head.rst", "cyto_dl.nn.head.vic_reg.rst", "cyto_dl.nn.hr_skip.rst", "cyto_dl.nn.losses.rst", "cyto_dl.nn.losses.adversarial_loss.rst", "cyto_dl.nn.losses.chamfer_loss.rst", "cyto_dl.nn.losses.continuous_bernoulli.rst", "cyto_dl.nn.losses.cosine_loss.rst", "cyto_dl.nn.losses.gan_loss.rst", "cyto_dl.nn.losses.gaussian_nll_loss.rst", "cyto_dl.nn.losses.geomloss.rst", "cyto_dl.nn.losses.loss_wrapper.rst", "cyto_dl.nn.losses.spharm_loss.rst", "cyto_dl.nn.losses.threshold_loss.rst", "cyto_dl.nn.losses.vic_reg.rst", "cyto_dl.nn.losses.weibull.rst", "cyto_dl.nn.losses.weighted_mse_loss.rst", "cyto_dl.nn.mlp.rst", "cyto_dl.nn.point_cloud.rst", "cyto_dl.nn.point_cloud.dgcnn.rst", "cyto_dl.nn.point_cloud.folding_net.rst", "cyto_dl.nn.point_cloud.graph_functions.rst", "cyto_dl.nn.point_cloud.vnn.rst", "cyto_dl.nn.res_unit.rst", "cyto_dl.nn.spatial_transformer.rst", "cyto_dl.nn.track_sequence_predictor.rst", "cyto_dl.nn.vits.rst", "cyto_dl.nn.vits.blocks.rst", "cyto_dl.nn.vits.blocks.cross_attention.rst", "cyto_dl.nn.vits.blocks.intermediate_weigher.rst", "cyto_dl.nn.vits.blocks.masked_unit_attention.rst", "cyto_dl.nn.vits.blocks.patchify.rst", "cyto_dl.nn.vits.blocks.patchify.patchify.rst", "cyto_dl.nn.vits.blocks.patchify.patchify_base.rst", "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.rst", "cyto_dl.nn.vits.decoder.rst", "cyto_dl.nn.vits.encoder.rst", "cyto_dl.nn.vits.mae.rst", "cyto_dl.nn.vits.predictor.rst", "cyto_dl.nn.vits.seg.rst", "cyto_dl.nn.vits.utils.rst", "cyto_dl.point_cloud.rst", "cyto_dl.point_cloud.io.rst", "cyto_dl.point_cloud.io.read_pcloud.rst", "cyto_dl.train.rst", "cyto_dl.utils.rst", "cyto_dl.utils.array.rst", "cyto_dl.utils.checkpoint.rst", "cyto_dl.utils.config.rst", "cyto_dl.utils.download_test_data.rst", "cyto_dl.utils.dummy_dataset.rst", "cyto_dl.utils.pylogger.rst", "cyto_dl.utils.rich_utils.rst", "cyto_dl.utils.rotation.rst", "cyto_dl.utils.spharm.rst", "cyto_dl.utils.spharm.reconstruction.rst", "cyto_dl.utils.spharm.rotation.rst", "cyto_dl.utils.template_utils.rst", "index.rst", "installation.rst", "modules.rst", "overview.md", "using_examples.rst"], "titles": ["Using the API", "cyto_dl package", "cyto_dl.api package", "cyto_dl.api.cyto_dl_model package", "cyto_dl.api.cyto_dl_model.cyto_dl_base_model module", "cyto_dl.api.cyto_dl_model.segmentation_plugin_model module", "cyto_dl.api.data module", "cyto_dl.api.model module", "cyto_dl.callbacks package", "cyto_dl.callbacks.callback_utils module", "cyto_dl.callbacks.csv_saver module", "cyto_dl.callbacks.image_saver module", "cyto_dl.callbacks.latent_walk module", "cyto_dl.callbacks.layer_freeze module", "cyto_dl.callbacks.model_utils module", "cyto_dl.callbacks.outlier_detection module", "cyto_dl.compile module", "cyto_dl.dataframe package", "cyto_dl.dataframe.readers module", "cyto_dl.dataframe.transforms package", "cyto_dl.dataframe.transforms.filter module", "cyto_dl.dataframe.transforms.group_cols module", "cyto_dl.dataframe.transforms.misc module", "cyto_dl.dataframe.transforms.split module", "cyto_dl.datamodules package", "cyto_dl.datamodules.array module", "cyto_dl.datamodules.data_dict module", "cyto_dl.datamodules.dataframe package", "cyto_dl.datamodules.dataframe.dataframe_datamodule module", "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule module", "cyto_dl.datamodules.dataframe.utils module", "cyto_dl.datamodules.folder module", "cyto_dl.datamodules.multidim_image module", "cyto_dl.datamodules.smartcache module", "cyto_dl.datamodules.torchvision module", "cyto_dl.eval module", "cyto_dl.image package", "cyto_dl.image.io package", "cyto_dl.image.io.aicsimage_loader module", "cyto_dl.image.io.monai_bio_reader module", "cyto_dl.image.io.numpy_reader module", "cyto_dl.image.io.ome_zarr_reader module", "cyto_dl.image.io.polygon_loader module", "cyto_dl.image.io.skimage_reader module", "cyto_dl.image.transforms package", "cyto_dl.image.transforms.add_meta module", "cyto_dl.image.transforms.bright_sampler module", "cyto_dl.image.transforms.clip module", "cyto_dl.image.transforms.contrastadjust module", "cyto_dl.image.transforms.generate_jepa_masks module", "cyto_dl.image.transforms.merge module", "cyto_dl.image.transforms.multiscale_cropper module", "cyto_dl.image.transforms.pad module", "cyto_dl.image.transforms.project module", "cyto_dl.image.transforms.rotation_mask_transform module", "cyto_dl.image.transforms.save module", "cyto_dl.image.transforms.so2_random_rotation module", "cyto_dl.image.transforms.track_transforms module", "cyto_dl.loggers package", "cyto_dl.loggers.mlflow module", "cyto_dl.models package", "cyto_dl.models.base_model module", "cyto_dl.models.basic_model module", "cyto_dl.models.classification package", "cyto_dl.models.classification.classification module", "cyto_dl.models.classification.timepoint_classification module", "cyto_dl.models.contrastive package", "cyto_dl.models.contrastive.contrastive module", "cyto_dl.models.handlers package", "cyto_dl.models.handlers.base_handler module", "cyto_dl.models.handlers.image_handler module", "cyto_dl.models.handlers.load_image_patch module", "cyto_dl.models.im2im package", "cyto_dl.models.im2im.gan module", "cyto_dl.models.im2im.multi_task module", "cyto_dl.models.im2im.utils package", "cyto_dl.models.im2im.utils.instance_seg module", "cyto_dl.models.im2im.utils.noise_annealer module", "cyto_dl.models.im2im.utils.postprocessing package", "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label module", "cyto_dl.models.im2im.utils.postprocessing.auto_thresh module", "cyto_dl.models.im2im.utils.postprocessing.dict_to_im module", "cyto_dl.models.jepa package", "cyto_dl.models.jepa.ijepa module", "cyto_dl.models.jepa.iwm module", "cyto_dl.models.jepa.jepa_base module", "cyto_dl.models.utils package", "cyto_dl.models.utils.mlflow module", "cyto_dl.models.utils.utils module", "cyto_dl.models.vae package", "cyto_dl.models.vae.base_vae module", "cyto_dl.models.vae.image_canon_vae module", "cyto_dl.models.vae.image_encoder module", "cyto_dl.models.vae.image_vae module", "cyto_dl.models.vae.latent_loss_vae module", "cyto_dl.models.vae.o2_spharm_vae package", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_encoder module", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_vae module", "cyto_dl.models.vae.point_cloud_vae module", "cyto_dl.models.vae.priors package", "cyto_dl.models.vae.priors.abstract_prior module", "cyto_dl.models.vae.priors.gaussian module", "cyto_dl.models.vae.priors.identity_prior module", "cyto_dl.models.vae.priors.joint_prior module", "cyto_dl.models.vae.tabular_vae module", "cyto_dl.nn package", "cyto_dl.nn.discriminators package", "cyto_dl.nn.discriminators.multi_scale_discriminator module", "cyto_dl.nn.discriminators.n_layer_discriminator module", "cyto_dl.nn.head package", "cyto_dl.nn.head.base_head module", "cyto_dl.nn.head.gan_head module", "cyto_dl.nn.head.gan_head_superres module", "cyto_dl.nn.head.mae_head module", "cyto_dl.nn.head.mask_head module", "cyto_dl.nn.head.res_blocks_head module", "cyto_dl.nn.head.vic_reg module", "cyto_dl.nn.hr_skip module", "cyto_dl.nn.losses package", "cyto_dl.nn.losses.adversarial_loss module", "cyto_dl.nn.losses.chamfer_loss module", "cyto_dl.nn.losses.continuous_bernoulli module", "cyto_dl.nn.losses.cosine_loss module", "cyto_dl.nn.losses.gan_loss module", "cyto_dl.nn.losses.gaussian_nll_loss module", "cyto_dl.nn.losses.geomloss module", "cyto_dl.nn.losses.loss_wrapper module", "cyto_dl.nn.losses.spharm_loss module", "cyto_dl.nn.losses.threshold_loss module", "cyto_dl.nn.losses.vic_reg module", "cyto_dl.nn.losses.weibull module", "cyto_dl.nn.losses.weighted_mse_loss module", "cyto_dl.nn.mlp module", "cyto_dl.nn.point_cloud package", "cyto_dl.nn.point_cloud.dgcnn module", "cyto_dl.nn.point_cloud.folding_net module", "cyto_dl.nn.point_cloud.graph_functions module", "cyto_dl.nn.point_cloud.vnn module", "cyto_dl.nn.res_unit module", "cyto_dl.nn.spatial_transformer module", "cyto_dl.nn.track_sequence_predictor module", "cyto_dl.nn.vits package", "cyto_dl.nn.vits.blocks package", "cyto_dl.nn.vits.blocks.cross_attention module", "cyto_dl.nn.vits.blocks.intermediate_weigher module", "cyto_dl.nn.vits.blocks.masked_unit_attention module", "cyto_dl.nn.vits.blocks.patchify package", "cyto_dl.nn.vits.blocks.patchify.patchify module", "cyto_dl.nn.vits.blocks.patchify.patchify_base module", "cyto_dl.nn.vits.blocks.patchify.patchify_hiera module", "cyto_dl.nn.vits.decoder module", "cyto_dl.nn.vits.encoder module", "cyto_dl.nn.vits.mae module", "cyto_dl.nn.vits.predictor module", "cyto_dl.nn.vits.seg module", "cyto_dl.nn.vits.utils module", "cyto_dl.point_cloud package", "cyto_dl.point_cloud.io package", "cyto_dl.point_cloud.io.read_pcloud module", "cyto_dl.train module", "cyto_dl.utils package", "cyto_dl.utils.array module", "cyto_dl.utils.checkpoint module", "cyto_dl.utils.config module", "cyto_dl.utils.download_test_data module", "cyto_dl.utils.dummy_dataset module", "cyto_dl.utils.pylogger module", "cyto_dl.utils.rich_utils module", "cyto_dl.utils.rotation module", "cyto_dl.utils.spharm package", "cyto_dl.utils.spharm.reconstruction module", "cyto_dl.utils.spharm.rotation module", "cyto_dl.utils.template_utils module", "Overview", "Installation", "cyto_dl", "Overview", "Modifying the Example Configs"], "terms": {"cytodl": [0, 5, 173, 176], "can": [0, 18, 20, 21, 28, 29, 31, 38, 40, 51, 52, 77, 90, 153, 155, 173, 176, 177], "import": [0, 138, 173, 176, 177], "python": [0, 21, 166, 172, 174], "packag": [0, 175], "programmat": 0, "cyto_dl": [0, 173, 176, 177], "cytodlmodel": [0, 1, 2, 7], "class": [0, 4, 5, 6, 7, 10, 11, 13, 14, 15, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 34, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 49, 50, 51, 52, 53, 55, 57, 59, 61, 62, 64, 65, 67, 73, 74, 76, 77, 79, 80, 81, 83, 84, 85, 90, 100, 101, 102, 103, 107, 108, 110, 111, 112, 113, 114, 115, 116, 117, 119, 120, 121, 122, 123, 124, 126, 128, 129, 130, 131, 132, 138, 139, 140, 143, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 165, 168], "addit": [0, 25, 26, 28, 29, 31, 33, 39, 61, 64, 67, 73, 74, 83, 84, 85, 90, 173, 176, 177], "command": [0, 166, 167, 172], "line": [0, 166, 167, 172, 177], "interfac": [0, 90], "we": [0, 173, 176, 177], "provid": [0, 7, 22, 32, 47, 48, 76, 126, 151, 167, 173, 176, 177], "configur": [0, 4, 5, 7, 167, 177], "file": [0, 4, 7, 18, 28, 29, 31, 32, 33, 40, 110, 172, 173, 176, 177], "model": [0, 1, 2, 4, 51, 115, 140, 161, 162, 167, 172, 175, 177], "make": [0, 14, 77, 172, 177], "from": [0, 4, 5, 7, 14, 18, 25, 26, 28, 29, 30, 31, 32, 38, 39, 43, 45, 46, 49, 51, 57, 62, 77, 81, 112, 115, 120, 123, 138, 143, 153, 161, 167, 168, 172, 177], "dictionari": [0, 7, 21, 25, 26, 28, 29, 30, 31, 32, 38, 40, 45, 47, 48, 51, 61, 67, 76, 81, 90, 151, 152, 161, 177], "cfg": [0, 4, 5, 35, 159, 163, 167, 172], "data": [0, 1, 2, 7, 22, 23, 25, 26, 28, 29, 30, 31, 33, 38, 39, 41, 43, 45, 79, 83, 84, 85, 115, 161, 167, 177], "load_config_from_dict": [0, 2, 7], "yaml": [0, 4], "load_config_from_yaml": 0, "path": [0, 4, 7, 11, 18, 26, 28, 29, 31, 32, 33, 38, 40, 42, 55, 62, 87, 145, 154, 167, 172, 173, 176, 177], "name": [0, 4, 13, 18, 21, 28, 29, 30, 31, 46, 47, 48, 50, 51, 52, 64, 153, 166, 173, 176, 177], "one": [0, 22, 28, 29, 30, 61, 172, 177], "our": [0, 177], "default": [0, 4, 5, 23, 28, 29, 41, 74, 79, 83, 84, 85, 138, 145, 154, 173, 176, 177], "avail": [0, 29, 30], "option": [0, 13, 18, 20, 21, 23, 26, 28, 29, 31, 32, 42, 49, 61, 62, 67, 74, 76, 83, 84, 85, 90, 115, 145, 154, 161, 167, 172, 174, 177], "ar": [0, 18, 20, 21, 22, 23, 25, 26, 28, 29, 31, 32, 38, 39, 42, 47, 48, 49, 61, 64, 73, 74, 76, 108, 115, 123, 148, 150, 151, 152, 153, 167, 172, 173, 176, 177], "gan": [0, 2, 6, 60, 72, 111, 112, 114, 123, 173, 176, 177], "instance_seg": [0, 2, 6, 72, 75], "labelfre": [0, 6, 173, 176, 177], "segmen": 0, "when": [0, 25, 26, 28, 29, 30, 31, 61, 76, 138, 173, 176], "experi": [0, 5, 177], "overrid": [0, 7, 173, 176], "list": [0, 7, 11, 13, 18, 20, 22, 25, 28, 29, 31, 32, 38, 40, 42, 43, 45, 50, 51, 53, 57, 62, 67, 74, 76, 79, 81, 90, 116, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 163, 172], "string": [0, 13, 18, 20, 21, 31, 61, 138], "format": [0, 38, 57, 177], "same": [0, 21, 57, 62, 90, 115, 123, 148, 150, 151, 152, 153, 155, 177], "wai": [0, 173, 176], "cli": 0, "load_default_experi": [0, 2, 7], "batch_siz": [0, 25, 26, 28, 29, 30, 31, 84, 165, 173, 176, 177], "16": [0, 6, 152, 154], "onc": [0, 177], "i": [0, 4, 18, 20, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 42, 49, 51, 52, 53, 61, 62, 76, 79, 81, 83, 84, 85, 90, 115, 123, 126, 138, 144, 150, 151, 152, 153, 154, 155, 163, 165, 168, 172, 173, 176, 177], "attribut": 0, "overridden": [0, 177], "override_config": [0, 2, 7], "which": [0, 7, 20, 21, 25, 26, 28, 29, 31, 32, 40, 61, 77, 107, 168, 172, 173, 176, 177], "take": [0, 18, 51, 53], "valu": [0, 6, 20, 21, 22, 23, 25, 28, 32, 38, 47, 48, 49, 79, 81, 83, 84, 85, 128, 138, 143, 145, 163, 165, 172, 173, 176, 177], "made": [0, 28, 61, 177], "async": 0, "pass": [0, 25, 26, 28, 29, 31, 32, 33, 38, 39, 51, 61, 64, 67, 73, 74, 77, 83, 84, 85, 90, 107, 173, 176, 177], "run_async": [0, 7], "true": [0, 7, 23, 28, 30, 32, 38, 39, 42, 45, 47, 51, 59, 76, 87, 90, 110, 111, 113, 114, 116, 128, 138, 145, 147, 148, 150, 151, 152, 153, 154, 155, 173, 176], "await": 0, "The": [0, 20, 23, 28, 29, 32, 49, 61, 62, 83, 84, 85, 123, 138, 168, 173, 176, 177], "To": [0, 173, 176], "examin": [0, 46], "print_config": [0, 2, 7], "download": 0, "exampl": [0, 32, 51, 61, 77, 115, 138, 151, 173, 176], "download_example_data": [0, 2, 7], "thi": [0, 4, 14, 23, 28, 29, 30, 32, 49, 51, 61, 62, 77, 165, 173, 176, 177], "api": [1, 175], "cyto_dl_model": [1, 2], "modul": [1, 2, 3, 8, 17, 19, 24, 27, 36, 37, 44, 58, 60, 63, 66, 68, 72, 75, 78, 82, 86, 89, 95, 99, 105, 106, 109, 118, 133, 141, 142, 146, 156, 157, 160, 169, 175], "autothresholdmethod": [1, 2, 6], "experimenttyp": [1, 2, 6], "hardwaretyp": [1, 2, 5, 6], "patchsiz": [1, 2, 5, 6], "callback": [1, 167, 172, 173, 175, 176], "callback_util": [1, 8], "csv_saver": [1, 8], "csvsaver": [1, 8, 10], "jepasav": [1, 8, 10], "image_sav": [1, 8], "imagesav": [1, 8, 11], "latent_walk": [1, 8], "layer_freez": [1, 8], "layerfreez": [1, 8, 13], "model_util": [1, 8], "getembed": [1, 8, 14], "get_all_embed": [1, 8, 14], "save_predictions_classifi": [1, 8, 14], "outlier_detect": [1, 8], "outlierdetect": [1, 8, 15], "datafram": [1, 24, 33, 173, 175, 176], "transform": [1, 17, 25, 26, 28, 29, 30, 31, 32, 33, 36, 38, 42, 43, 76, 79, 115, 140, 150, 151, 152, 153, 173, 176, 177], "reader": [1, 17], "filter_column": [1, 17, 18, 19, 20], "read_csv": [1, 17, 18], "read_datafram": [1, 17, 18], "read_h5ad": [1, 17, 18], "read_parquet": [1, 17, 18], "datamodul": [1, 14, 173, 175, 176], "arrai": [1, 24, 32, 42, 81, 115, 126, 148, 160, 173, 176], "make_array_dataload": [1, 24, 25], "data_dict": [1, 24], "make_data_dict_dataload": [1, 24, 26], "folder": [1, 24, 167, 173, 176], "make_folder_dataload": [1, 24, 31], "multidim_imag": [1, 24], "multidimimagedataset": [1, 24, 32], "make_multidim_image_dataload": [1, 24, 32], "smartcach": [1, 24], "smartcachedatamodul": [1, 24, 33], "torchvis": [1, 24], "cifar10": [1, 24, 34], "cifar100": [1, 24, 34], "clevrclassif": [1, 24, 34], "caltech101": [1, 24, 34], "caltech256": [1, 24, 34], "celeba": [1, 24, 34], "cityscap": [1, 24, 34], "cococapt": [1, 24, 34], "cocodetect": [1, 24, 34], "country211": [1, 24, 34], "dtd": [1, 24, 34], "datasetfold": [1, 24, 34], "emnist": [1, 24, 34], "eurosat": [1, 24, 34], "fer2013": [1, 24, 34], "fgvcaircraft": [1, 24, 34], "fakedata": [1, 24, 34], "fashionmnist": [1, 24, 34], "flickr30k": [1, 24, 34], "flickr8k": [1, 24, 34], "flowers102": [1, 24, 34], "food101": [1, 24, 34], "gtsrb": [1, 24, 34], "hmdb51": [1, 24, 34], "inaturalist": [1, 24, 34], "imagefold": [1, 24, 34], "imagenet": [1, 24, 34], "kmnist": [1, 24, 34], "kinet": [1, 24, 34], "kitti": [1, 24, 34], "lfwpair": [1, 24, 34], "lfwpeopl": [1, 24, 34], "lsun": [1, 24, 34], "lsunclass": [1, 24, 34], "mnist": [1, 24, 34], "movingmnist": [1, 24, 34], "omniglot": [1, 24, 34], "oxfordiiitpet": [1, 24, 34], "pcam": [1, 24, 34], "phototour": [1, 24, 34], "places365": [1, 24, 34], "qmnist": [1, 24, 34], "renderedsst2": [1, 24, 34], "sbdataset": [1, 24, 34], "sbu": [1, 24, 34], "semeion": [1, 24, 34], "stl10": [1, 24, 34], "sun397": [1, 24, 34], "svhn": [1, 24, 34], "stanfordcar": [1, 24, 34], "ucf101": [1, 24, 34], "usp": [1, 24, 34], "vocdetect": [1, 24, 34], "vocsegment": [1, 24, 34], "visiondataset": [1, 24, 34], "widerfac": [1, 24, 34], "imag": [1, 11, 26, 29, 30, 32, 33, 64, 67, 73, 74, 76, 79, 81, 83, 84, 108, 110, 111, 112, 113, 114, 115, 116, 123, 126, 140, 147, 148, 149, 150, 151, 152, 153, 173, 175, 176, 177], "io": [1, 18, 36, 73, 74, 115, 150, 156, 173, 176], "logger": [1, 166, 167, 172, 173, 175, 176], "mlflow": [1, 58, 60, 86, 173, 176], "mlflowlogg": [1, 58, 59], "classif": [1, 60], "contrast": [1, 48, 60], "handler": [1, 60], "im2im": [1, 60], "jepa": [1, 49, 60, 153], "util": [1, 24, 25, 26, 27, 28, 29, 31, 60, 72, 105, 141, 173, 175, 176, 177], "vae": [1, 60], "base_model": [1, 60], "basemodel": [1, 60, 61, 62, 64, 67, 73, 74, 83, 84, 85, 90], "basemodelmeta": [1, 60, 61], "basic_model": [1, 60], "basicmodel": [1, 60, 62], "nn": [1, 62, 64, 67, 73, 74, 79, 83, 84, 85, 90, 175], "discrimin": [1, 73, 77, 105, 111, 119, 123, 173, 176], "head": [1, 64, 67, 73, 74, 105, 145, 150, 151, 152, 153, 154, 161, 173, 176, 177], "loss": [1, 61, 62, 76, 90, 102, 105, 110, 111, 112, 113, 114, 115, 116, 173, 176, 177], "point_cloud": [1, 105, 175], "vit": [1, 105], "hr_skip": [1, 105], "hrskip": [1, 105, 117], "mlp": [1, 105, 116, 141, 142, 143, 145], "res_unit": [1, 105], "residualunit": [1, 105, 138], "spatial_transform": [1, 105], "convpoolrelu": [1, 105, 139], "stn": [1, 105, 139], "track_sequence_predictor": [1, 105], "trackclassifi": [1, 105, 140], "positionalencoding1d": [1, 105, 140], "spharm": [1, 160], "create_dataload": [1, 160, 161], "extract_array_predict": [1, 160, 161], "checkpoint": [1, 4, 15, 87, 154, 160], "load_checkpoint": [1, 160, 162], "config": [1, 4, 7, 160, 161, 167, 172, 173, 176], "is_config": [1, 160, 163], "kv_to_dict": [1, 160, 163], "remove_aux_kei": [1, 160, 163], "download_test_data": [1, 160], "delete_test_data": [1, 160, 164], "parse_s3_path": [1, 160, 164], "setup_path": [1, 160, 164], "dummy_dataset": [1, 160], "dummydatamodul": [1, 160, 165], "dummydataset": [1, 160, 165], "pylogg": [1, 160], "get_pylogg": [1, 160, 166], "rich_util": [1, 160], "enforce_tag": [1, 160, 167], "print_config_tre": [1, 160, 167], "rotat": [1, 160, 169, 177], "rotationmodul": [1, 160, 168], "get_rotation_matrix_so2": [1, 160, 168], "get_rotation_matrix_so3": [1, 160, 168], "template_util": [1, 160], "close_logg": [1, 160, 172], "extra": [1, 148, 149, 151, 152, 160, 167, 172, 174], "get_metric_valu": [1, 160, 172], "instantiate_callback": [1, 160, 172], "instantiate_logg": [1, 160, 172], "log_hyperparamet": [1, 160, 172], "save_fil": [1, 160, 172], "task_wrapp": [1, 160, 172], "compil": [1, 64, 73, 74, 173, 175, 176], "eval": [1, 175], "main": [1, 35, 120, 159, 175], "train": [1, 2, 3, 4, 5, 6, 7, 11, 23, 25, 26, 28, 29, 30, 31, 32, 49, 51, 59, 61, 62, 64, 73, 74, 76, 83, 84, 85, 87, 114, 154, 173, 175, 176], "cyto_dl_base_model": [2, 3], "cytodlbasemodel": [2, 3, 4, 5], "segmentation_plugin_model": [2, 3], "segmentationpluginmodel": [2, 3, 5], "isodata": [2, 6], "li": [2, 6], "local": [2, 6], "mean": [2, 6, 46, 76, 101, 119, 121, 122], "minimum": [2, 6, 33], "multiotsu": [2, 6], "niblack": [2, 6], "otsu": [2, 6], "sauvola": [2, 6], "triangl": [2, 6], "yen": [2, 6], "label_fre": [2, 6], "segment": [2, 6, 52, 76, 77, 154, 173, 176, 177], "segmentation_arrai": [2, 6], "segmentation_plugin": [2, 5, 6], "cpu": [2, 6, 61, 81], "gpu": [2, 6, 166, 172, 177], "larg": [2, 6, 33, 177], "medium": [2, 6], "small": [2, 6, 76], "load_config_from_fil": [2, 7], "predict": [2, 3, 4, 5, 7, 10, 25, 28, 29, 32, 61, 62, 64, 67, 83, 84, 85, 110, 111, 112, 113, 114, 115, 123, 126, 153, 161, 173, 176], "save_config": [2, 3, 4, 7], "from_default_config": [3, 4], "from_existing_config": [3, 4], "get_config": [3, 4, 60, 86, 87], "get_experiment_nam": [3, 4], "set_experiment_nam": [3, 4], "get_hardware_typ": [3, 5], "get_input_channel": [3, 5], "get_manifest_column_nam": [3, 5], "get_patch_s": [3, 5], "get_raw_image_channel": [3, 5], "get_split_column": [3, 5], "remove_split_column": [3, 5], "set_hardware_typ": [3, 5], "set_input_channel": [3, 5], "set_manifest_column_nam": [3, 5], "set_patch_s": [3, 5], "set_raw_image_channel": [3, 5], "set_split_column": [3, 5], "dictconfig": [4, 5, 26, 35, 159, 163, 167, 172], "sourc": [4, 5, 6, 7, 10, 11, 13, 14, 15, 18, 20, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 35, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 49, 50, 51, 52, 53, 55, 57, 59, 61, 62, 64, 65, 67, 73, 74, 76, 77, 79, 80, 81, 83, 84, 85, 87, 88, 90, 100, 101, 102, 103, 107, 108, 110, 111, 112, 113, 114, 115, 116, 117, 119, 120, 121, 122, 123, 124, 126, 128, 129, 130, 131, 132, 138, 139, 140, 143, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 159, 161, 162, 163, 164, 165, 166, 167, 168, 172], "base": [4, 5, 6, 7, 10, 11, 13, 14, 15, 21, 22, 26, 28, 29, 30, 31, 32, 33, 34, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 49, 50, 51, 52, 53, 55, 57, 59, 61, 62, 64, 65, 67, 73, 74, 76, 77, 79, 80, 81, 83, 84, 85, 90, 100, 101, 102, 103, 107, 108, 110, 111, 112, 113, 114, 115, 116, 117, 119, 120, 121, 122, 123, 124, 126, 128, 129, 130, 131, 132, 138, 139, 140, 143, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 165, 168, 173, 176, 177], "abc": [4, 79, 110, 148, 152], "A": [4, 5, 18, 20, 22, 26, 28, 29, 31, 62, 123, 138, 168], "us": [4, 5, 14, 18, 20, 21, 22, 23, 25, 26, 28, 29, 31, 32, 33, 38, 39, 47, 48, 49, 50, 51, 52, 61, 62, 64, 67, 73, 74, 76, 77, 83, 84, 85, 90, 110, 112, 114, 115, 123, 138, 148, 150, 151, 152, 153, 165, 167, 173, 174, 176, 177], "run": [4, 32, 74, 110, 113, 114, 116, 173, 176, 177], "cyto": [4, 174], "dl": [4, 174], "Not": [4, 114], "intend": 4, "direct": [4, 139], "client": 4, "pleas": 4, "see": [4, 18, 20, 25, 26, 28, 29, 31, 115, 138, 173, 176], "classmethod": [4, 101], "factori": 4, "method": [4, 61, 80, 115], "instead": [4, 38, 84, 152], "spatial_dim": [4, 32, 33, 49, 112, 115, 138, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 168, 177], "int": [4, 5, 7, 11, 22, 23, 28, 29, 30, 32, 33, 46, 48, 49, 51, 52, 53, 57, 67, 76, 77, 79, 83, 84, 85, 100, 102, 107, 108, 112, 115, 116, 126, 132, 138, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 165, 168], "return": [4, 18, 23, 32, 51, 61, 76, 107, 123, 140, 144, 163, 168], "paramet": [4, 7, 11, 13, 18, 20, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 38, 40, 42, 45, 46, 47, 48, 49, 50, 51, 52, 53, 55, 57, 62, 64, 67, 73, 74, 76, 77, 79, 81, 83, 84, 85, 90, 101, 107, 108, 110, 111, 112, 113, 114, 115, 116, 123, 126, 128, 140, 145, 148, 150, 151, 152, 153, 154, 163, 168, 172, 173, 176, 177], "dimens": [4, 32, 33, 49, 76, 115, 116, 138, 140, 144, 145, 148, 149, 150, 151, 152, 153, 154, 155], "e": [4, 18, 32, 90, 153, 173, 174, 176, 177], "g": [4, 32, 153, 173, 176, 177], "2": [4, 29, 32, 49, 51, 76, 107, 108, 117, 138, 139, 140, 145, 151, 154, 168], "config_filepath": 4, "an": [4, 6, 18, 25, 26, 29, 32, 49, 51, 53, 61, 79, 101, 126, 138, 148, 151, 173, 176, 177], "exist": [4, 23], "basi": 4, "must": [4, 13, 18, 20, 23, 25, 31, 32, 45, 51, 61, 173, 176], "gener": [4, 18, 22, 33, 49, 51, 57, 62, 76, 77, 79, 83, 84, 85, 108, 110, 111, 112, 155, 165, 173, 176], "subclass": [4, 30, 90], "want": [4, 173, 176, 177], "str": [4, 5, 7, 11, 13, 14, 15, 18, 20, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 50, 51, 52, 53, 55, 57, 59, 62, 64, 67, 73, 74, 76, 79, 80, 81, 83, 84, 85, 90, 114, 123, 126, 138, 147, 148, 149, 153, 154, 167, 172], "manifest_path": 4, "output_dir": [4, 7, 14], "none": [4, 5, 7, 11, 13, 14, 18, 20, 21, 22, 23, 26, 28, 29, 30, 31, 32, 33, 35, 41, 43, 51, 52, 59, 61, 62, 64, 74, 76, 79, 80, 90, 101, 108, 110, 111, 112, 113, 114, 115, 116, 117, 124, 131, 132, 138, 143, 147, 148, 149, 150, 151, 152, 153, 154, 159, 161, 165, 167, 172], "max_epoch": [4, 83, 84, 85], "handl": [5, 32, 42, 61, 123], "tupl": [5, 32, 39, 41, 43, 49, 61, 79, 138, 150, 155], "hardware_typ": 5, "input_channel": [5, 147, 148, 149, 151, 152], "target1": 5, "target2": 5, "merge_mask": 5, "exclude_mask": 5, "base_imag": 5, "patch_siz": [5, 140, 147, 148, 149, 150, 151, 152, 177], "channel": [5, 32, 33, 38, 41, 43, 46, 52, 57, 79, 108, 115, 126, 138, 148, 149, 151, 152, 154, 173, 176, 177], "split_column": [5, 28, 29, 30, 173, 176], "enum": 6, "enumer": [6, 33, 38], "threshold_isodata": 6, "threshold_li": 6, "threshold_loc": 6, "threshold_mean": 6, "threshold_minimum": 6, "threshold_multiotsu": 6, "threshold_niblack": 6, "threshold_otsu": 6, "threshold_sauvola": 6, "threshold_triangl": 6, "threshold_yen": 6, "patch": [6, 46, 49, 51, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 177], "size": [6, 30, 49, 51, 76, 123, 138, 148, 149, 150, 151, 152, 154, 155, 177], "respect": [6, 18, 90, 177], "shape": [6, 42, 148, 165], "128": [6, 140], "64": [6, 108, 139, 149, 151, 152], "32": [6, 139, 154], "object": [7, 18, 32, 46, 76, 77, 79, 80, 81, 108, 123, 154, 168], "dict": [7, 20, 21, 25, 26, 28, 29, 32, 39, 41, 43, 51, 52, 59, 62, 73, 74, 76, 83, 84, 85, 90, 151, 152, 165, 172], "load": [7, 18, 26, 28, 29, 31, 32, 33, 38, 39, 40, 62, 173, 176, 177], "config_path": 7, "experiment_typ": 7, "directori": [7, 11, 26, 28, 29, 31, 64, 67, 73, 74, 84, 177], "float": [7, 23, 33, 46, 47, 48, 49, 76, 77, 79, 80, 83, 84, 85, 90, 123, 126, 128, 138, 145, 152, 159, 168, 172], "bool": [7, 11, 20, 23, 28, 29, 30, 32, 38, 39, 40, 41, 42, 43, 45, 47, 48, 51, 52, 53, 55, 57, 76, 79, 81, 84, 112, 115, 123, 126, 128, 138, 145, 147, 148, 150, 151, 152, 153, 154, 155, 167], "fals": [7, 11, 20, 23, 28, 29, 30, 32, 38, 39, 40, 45, 47, 48, 51, 53, 55, 57, 64, 73, 74, 76, 79, 81, 84, 90, 100, 101, 103, 108, 112, 114, 115, 117, 119, 138, 143, 145, 148, 150, 151, 152, 153, 167], "save": [7, 8, 11, 32, 36, 44, 57, 62, 64, 67, 73, 74, 84, 110, 113, 114, 116, 151, 172, 173, 176, 177], "current": [7, 29, 32, 42], "on_predict_epoch_end": [8, 10], "pred_to_csv": [8, 10], "save_feat": [8, 10], "on_predict_batch_end": [8, 11, 15], "on_test_batch_end": [8, 11, 15], "on_train_batch_end": [8, 11], "on_validation_batch_end": [8, 11], "setup": [8, 13, 24, 33, 172], "on_test_epoch_end": [8, 14], "calculate_mahalanobis_hook": [8, 15], "flatten_activ": [8, 15], "on_load_checkpoint": [8, 15], "on_predict_epoch_start": [8, 15], "on_save_checkpoint": [8, 15], "on_test_epoch_start": [8, 15], "on_train_epoch_start": [8, 15], "update_covariance_hook": [8, 15], "save_dir": [10, 11, 15, 59, 64, 67, 73, 74, 84, 177], "meta_kei": [10, 45, 67], "trainer": [10, 11, 13, 14, 15, 167, 173, 176], "pl_modul": [10, 11, 13, 14, 15], "pred": [10, 14, 61, 107, 120], "stage": [10, 11, 13, 33, 61, 62, 64, 67, 73, 74, 83, 84, 85, 90, 110, 111, 113, 114, 116], "save_every_n_epoch": 11, "1": [11, 23, 29, 30, 33, 49, 51, 53, 64, 73, 74, 76, 79, 90, 101, 110, 111, 112, 113, 115, 116, 123, 126, 129, 131, 132, 138, 145, 147, 148, 149, 151, 152, 164, 177], "val": [11, 23, 28, 29, 61, 62, 88], "save_input": 11, "after": [11, 45, 76, 79, 115, 143, 172], "postprocess": [11, 72, 75, 110, 111, 112, 113, 114, 115, 116, 173, 176, 177], "ead": 11, "union": [11, 13, 18, 20, 21, 25, 26, 28, 29, 31, 32, 33, 40, 42, 50, 53, 62, 74, 76, 79, 81], "frequenc": [11, 64, 67, 73, 74], "whether": [11, 23, 28, 29, 30, 32, 38, 39, 40, 42, 46, 47, 48, 51, 52, 53, 57, 64, 73, 74, 76, 79, 81, 84, 114, 115, 128, 138, 148, 150, 151, 152, 165, 167, 168, 177], "input": [11, 20, 22, 23, 25, 26, 28, 29, 31, 40, 62, 64, 73, 74, 76, 83, 85, 108, 110, 111, 112, 113, 114, 115, 116, 119, 121, 122, 123, 124, 131, 138, 145, 148, 149, 151, 152, 153, 163, 167, 173, 176, 177], "output": [11, 15, 32, 38, 62, 79, 90, 110, 115, 116, 123, 138, 145, 151, 154, 161, 167, 172, 173, 176], "batch": [11, 15, 21, 26, 28, 29, 30, 31, 32, 45, 55, 61, 62, 64, 65, 67, 73, 74, 83, 84, 85, 90, 101, 110, 111, 113, 114, 116, 145, 149, 165, 168, 177], "batch_idx": [11, 15, 61, 62, 64, 65, 67, 73, 74, 83, 84, 85, 90], "dataloader_idx": [11, 15], "0": [11, 15, 33, 41, 46, 47, 48, 49, 51, 76, 77, 83, 84, 85, 90, 101, 112, 115, 117, 123, 126, 128, 138, 143, 145, 147, 148, 149, 151, 152, 154, 168], "step": [11, 32, 59, 61, 77, 177], "contain": [13, 18, 20, 21, 25, 26, 28, 31, 32, 33, 42, 177], "exclud": [13, 18, 20, 31, 76], "search": 13, "within": [13, 48, 151], "layer": [13, 108, 115, 116, 123, 138, 144, 145, 150, 151, 152, 153, 154], "freez": 13, "x_label": [14, 62, 90], "id_label": [14, 61, 90], "arg": [14, 25, 26, 28, 29, 31, 34, 38, 40, 61, 62, 64, 65, 67, 73, 74, 83, 84, 85, 90, 138, 165, 167], "id_field": 14, "lightningmodul": [14, 61, 172], "train_dataload": [14, 24, 27, 28, 33, 160, 165], "val_dataload": [14, 24, 27, 28, 33, 160, 165], "test_dataload": [14, 24, 27, 28, 33, 160, 165], "todo": [14, 23, 28, 29], "better": [14, 148, 150, 151, 152, 153, 177], "mayb": 14, "vol": 14, "predictor": [14, 83, 84, 85, 105, 141], "code": [14, 173, 176], "drop": [14, 30, 143], "unnecessari": 14, "index": [14, 53, 148, 149, 150, 155], "n_epoch": 15, "layer_nam": 15, "act": [15, 138], "set": [15, 28, 48, 172], "forward": [15, 60, 61, 62, 63, 64, 66, 67, 72, 74, 82, 85, 89, 90, 99, 100, 101, 102, 103, 105, 106, 107, 108, 109, 110, 111, 112, 115, 117, 118, 119, 120, 121, 122, 124, 129, 130, 131, 132, 138, 139, 140, 141, 142, 143, 144, 145, 146, 148, 150, 151, 152, 153, 154], "hook": 15, "callabl": [15, 25, 26, 31, 32, 51, 62, 79, 112, 115, 172], "filter": [17, 18, 19, 31, 108, 154], "filter_row": [17, 19, 20], "group_col": [17, 19], "groupcol": [17, 19, 21], "misc": [17, 19], "append_class_weight": [17, 19, 22], "append_labels_to_integ": [17, 19, 22], "append_one_hot": [17, 19, 22], "make_random_df": [17, 19, 22], "split": [17, 19, 28, 29, 30, 33, 61, 173, 176, 177], "sample_n_each": [17, 19, 23], "split_datafram": [17, 19, 23], "columns_to_filt": 18, "sequenc": [18, 20, 22, 25, 26, 28, 29, 30, 31, 39, 40, 41, 43, 45, 51, 55, 62, 76, 79, 90, 138, 140, 147, 148, 149, 155, 167, 173, 176], "regex": [18, 20, 31], "startswith": [18, 20, 31], "endswith": [18, 20, 31], "column": [18, 20, 21, 22, 23, 28, 29, 30, 31, 32, 33, 173, 176, 177], "combin": [18, 30, 173, 176, 177], "differ": [18, 20, 23, 38, 123, 126, 153, 154, 173, 176], "queri": [18, 143, 145, 151, 152], "pattern": 18, "If": [18, 20, 22, 25, 28, 29, 30, 31, 32, 38, 39, 45, 62, 115, 145, 148, 150, 151, 152, 153, 155, 173, 176, 177], "suppli": [18, 20], "preced": 18, "remain": [18, 20, 21], "argument": [18, 22, 25, 26, 28, 29, 31, 33, 39, 64, 67, 73, 74, 83, 84, 85, 90, 107, 138, 173, 176], "ignor": [18, 20, 28, 29, 42, 172], "otherwis": [18, 62, 177], "logic": [18, 61], "AND": 18, "appli": [18, 25, 26, 31, 33, 51, 52, 53, 57, 79, 115, 144, 172, 173, 176], "all": [18, 23, 28, 29, 32, 33, 51, 74, 79, 150, 153, 172, 173, 176, 177], "condit": 18, "regular": [18, 20, 31], "express": [18, 20, 31], "match": [18, 20, 31, 61, 155], "substr": [18, 20, 31], "start": [18, 20, 31, 32, 33, 51, 57, 172, 173, 176], "end": [18, 20, 31], "include_column": 18, "read": [18, 32, 33, 37, 38, 39, 41, 43], "store": [18, 26, 28, 29, 31, 62, 173, 176], "csv": [18, 29, 32, 33, 38, 67, 173, 176, 177], "includ": [18, 20, 23, 148, 149, 151, 152], "onli": [18, 20, 25, 26, 28, 29, 30, 31, 32, 42, 52, 62, 67, 76, 110, 138, 150, 152, 172, 177], "given": [18, 20, 22, 23, 28, 29, 46, 61, 102, 128, 155, 173, 176], "upath": [18, 26, 28, 29, 31], "desir": 18, "result": [18, 20, 61, 62, 177], "type": [18, 21, 29, 30, 32, 38, 61, 79, 112, 115, 123, 138, 145, 163, 168, 173, 176, 177], "pd": [18, 20, 22, 23], "required_column": 18, "parquet": [18, 28, 29], "assert": 18, "expect": [18, 26, 173, 176, 177], "requir": [18, 173, 174, 176, 177], "either": [18, 20, 21, 23, 28, 61, 155, 168], "support": [18, 29, 42, 77, 123], "aren": 18, "t": [18, 33, 38, 102, 174], "found": [18, 173, 176], "valueerror": 18, "thrown": 18, "those": [18, 173, 176], "get": [18, 33, 173, 176], "append": [18, 22, 23], "without": [18, 155], "duplic": 18, "back": 18, "anndata": 18, "h5ad": 18, "r": [18, 174], "": [18, 20, 28, 29, 30, 73, 74, 76, 77, 90, 111, 112, 168, 173, 176], "doc": [18, 25, 26, 28, 29, 31, 73, 74, 115, 173, 176], "detail": [18, 20], "http": [18, 25, 26, 28, 29, 31, 39, 43, 73, 74, 115, 120, 121, 150, 173, 174, 176], "readthedoc": 18, "en": [18, 73, 74, 115, 173, 176], "latest": 18, "html": [18, 25, 26, 28, 29, 31, 73, 74, 115, 173, 176], "select": [20, 23, 30, 46, 173, 176], "dataset": [20, 25, 28, 29, 30, 32, 33, 165, 173, 176], "more": [20, 25, 26, 28, 29, 31, 172, 173, 176], "oper": 20, "It": [20, 28, 101, 177], "panda": [20, 23], "case": [20, 21, 23, 28, 29, 61, 173, 176, 177], "explicit": [20, 130], "keep": [20, 76, 148, 173, 176], "row": [20, 22, 23], "where": [20, 21, 29, 32, 61, 79, 150, 154, 163, 165, 173, 176], "group": [21, 29, 149, 168], "dtype": [21, 38, 79, 126], "kei": [21, 26, 28, 29, 30, 31, 32, 38, 40, 42, 45, 46, 47, 48, 50, 51, 52, 53, 55, 57, 61, 62, 64, 67, 73, 74, 76, 81, 83, 84, 85, 90, 143, 145, 151, 152, 163, 165, 173, 176], "becom": [21, 165, 173, 176], "kwarg": [21, 33, 34, 38, 40, 61, 62, 64, 65, 67, 73, 74, 83, 84, 85, 90, 100, 101, 102, 103, 107, 120, 165], "col": 21, "filter_col": 21, "singl": [21, 25, 28, 32, 151, 155, 173, 176], "np": [21, 25, 79], "numpi": [21, 25, 32, 38, 42, 79, 81, 161], "recogn": 21, "turn": 21, "add": [22, 45, 61, 145], "weight": [22, 62, 76, 83, 84, 85, 90, 111, 112, 126, 131, 144, 151, 154], "modifi": [22, 23, 29, 107], "its": [22, 46, 167], "integ": [22, 32, 51], "encod": [22, 49, 83, 84, 85, 89, 90, 105, 140, 141, 143, 148, 149, 150, 152], "convert": [22, 32, 42, 76, 81, 147, 148, 149, 151, 161], "hot": 22, "call": [22, 28, 29, 61, 77, 172], "function": [22, 32, 51, 62, 76, 90, 110, 111, 112, 113, 114, 115, 116, 126, 128, 172, 173, 176, 177], "one_hot_encod": 22, "b": [22, 130], "c": [22, 38], "d": 22, "n_row": 22, "100": [22, 51, 76, 83, 84, 85, 111, 112], "random": [22, 23, 28, 29, 51], "test": [22, 23, 28, 29, 32, 61, 62, 173, 176], "wrangl": 22, "pipelin": 22, "creat": [22, 25, 26, 28, 29, 31, 32, 42, 64, 76, 123, 154, 161, 173, 174, 176, 177], "number": [22, 23, 28, 29, 33, 38, 49, 51, 52, 57, 77, 83, 84, 85, 107, 108, 115, 126, 138, 145, 148, 149, 150, 151, 152, 153, 154, 155, 172, 177], "forc": 23, "seed": [23, 28, 29], "42": [23, 28, 29], "have": [23, 45, 51, 61, 76, 138, 150, 151, 152], "equal": [23, 126], "per": [23, 28, 29, 51, 76, 126, 145, 161], "In": [23, 61, 168, 173, 176, 177], "ha": [23, 77, 123, 151, 155], "less": 23, "than": [23, 138], "correspond": [23, 29, 31, 51, 165], "sampl": [23, 25, 26, 28, 29, 30, 31, 46, 49, 51, 99, 100, 101, 103, 165], "replac": [23, 33], "uniqu": [23, 173, 176], "toggl": [23, 105, 139], "upsampl": [23, 115, 154, 173, 176], "smaller": 23, "train_frac": 23, "val_frac": 23, "return_split": 23, "perform": [23, 115, 151], "three": [23, 28], "identifi": 23, "each": [23, 25, 26, 28, 29, 30, 31, 33, 49, 57, 61, 64, 76, 90, 138, 140, 144, 148, 149, 150, 151, 152, 153, 154, 155, 165, 173, 176, 177], "belong": 23, "extend": 23, "enabl": 23, "balanc": 23, "stratifi": 23, "fraction": [23, 152], "valid": [23, 33, 61, 64, 73, 74, 173, 176], "By": [23, 177], "half": 23, "between": [23, 29, 30, 32, 64, 73, 74, 101, 110, 115, 126, 153], "separ": [23, 32, 38, 76], "reproduc": 23, "dataframe_datamodul": [24, 27], "dataframedatamodul": [24, 27, 28, 29], "grouped_dataframe_datamodul": [24, 27], "groupeddataframedatamodul": [24, 27, 29], "alternatingbatchsampl": [24, 27, 29, 30], "removenankeysd": [24, 27, 30], "get_canonical_split_nam": [24, 27, 30], "get_dataset": [24, 27, 28, 30], "make_multiple_dataframe_split": [24, 27, 30], "make_single_dataframe_split": [24, 27, 30], "parse_transform": [24, 27, 30], "create_metatensor": [24, 32], "get_per_file_arg": [24, 32, 33], "is_batch": [24, 32], "make_dataload": [24, 27, 28, 29, 33], "predict_dataload": [24, 27, 28, 33, 160, 165], "prepare_data": [24, 33], "ndarrai": [25, 39, 41, 43, 76, 81], "source_kei": [25, 84], "dataloader_kwarg": [25, 26, 28, 29, 31, 32], "dataload": [25, 26, 28, 29, 31, 32, 33, 62, 161, 173, 176], "element": [25, 61, 138, 155, 165], "keyword": [25, 26, 28, 29, 31, 39], "torch": [25, 26, 28, 29, 30, 31, 42, 62, 64, 73, 74, 79, 90, 101, 108, 111, 112, 115, 123, 143, 144, 145, 168, 177], "instanti": [25, 26, 28, 29, 31, 90, 172, 173, 176], "asid": [25, 26, 28, 29, 31], "shuffl": [25, 26, 28, 29, 30, 31], "among": [25, 26, 28, 29, 31], "num_work": [25, 26, 28, 29, 31, 33, 173, 176], "etc": [25, 26, 28, 29, 31, 173, 176, 177], "pytorch": [25, 26, 28, 29, 30, 31, 62, 90, 173, 176], "info": [25, 26, 28, 29, 31], "org": [25, 26, 28, 29, 31, 121], "stabl": [25, 26, 28, 29, 31, 73, 74, 115, 173, 176], "cache_dir": [26, 28, 29, 30, 31, 173, 176], "usual": 26, "cach": [26, 28, 29, 31, 33, 173, 176], "acceler": [26, 28, 29, 31], "get_dataload": [27, 28, 160, 165], "split_map": [28, 29, 30], "just_infer": [28, 29, 30], "subsampl": [28, 29], "refresh_subsampl": [28, 29], "smartcache_arg": [28, 29, 30], "lightningdatamodul": [28, 33, 165], "lightn": [28, 62, 172, 173, 176, 177], "fold": 28, "addition": [28, 172], "go": 28, "flag": 28, "so": [28, 29, 77, 168, 173, 176, 177], "whole": [28, 29], "predict_datamodul": 28, "simpli": [28, 61, 173, 176], "construct": [28, 173, 176], "regardless": 28, "specif": [28, 29, 61, 73, 74, 90, 115, 173, 176, 177], "everyth": [28, 29, 48], "document": [28, 29, 173, 176], "just": [28, 29, 84, 151], "infer": [28, 29, 61, 73, 74, 90, 100, 101, 103, 168], "epoch": [28, 29, 83, 84, 85], "refresh": [28, 29], "time": [28, 29, 30, 77, 151, 152, 172], "smartcachedataset": [28, 29], "target_column": [29, 30], "grouping_column": [29, 30], "should": [29, 32, 33, 40, 51, 61, 115, 126, 138, 151, 152, 154, 177], "some": [29, 32, 138, 173, 176, 177], "criterion": 29, "leverag": 29, "two": [29, 50, 101, 173, 176], "multitask": 29, "ground": [29, 30, 76, 115, 123, 173, 176, 177], "truth": [29, 30, 76, 115, 123, 173, 176, 177], "task": [29, 30, 64, 67, 73, 74, 77, 110, 111, 113, 114, 115, 116, 147, 148, 149, 172, 173, 176, 177], "characterist": 29, "altern": [29, 30, 177], "dure": [29, 32, 64, 67, 73, 74, 114, 154, 172, 173, 176], "factor": [29, 51, 154, 177], "homogen": 29, "across": [29, 53, 76, 173, 176], "subset": 30, "drop_last": 30, "sampler": 30, "subsetrandomsampl": 30, "batchsampl": 30, "mutual": 30, "exclus": 30, "monai": [30, 33, 51, 53, 73, 74, 115, 138, 173, 176, 177], "wrap": [30, 62, 172, 173, 176], "repres": [30, 138], "last": [30, 52, 87, 123, 138, 144, 151], "incomplet": 30, "randomli": [30, 49, 52, 138], "follow": [30, 151, 152, 177], "order": [30, 138, 148, 149, 152, 154, 167], "remov": [30, 49, 76, 152], "nan": 30, "ad": [30, 45, 49, 177], "allow_missing_kei": [30, 38, 47, 48, 51, 52, 53, 55, 57, 76, 81], "allow": [30, 38, 52, 55, 57, 177], "multi": [30, 32, 151, 166, 172, 173, 176, 177], "target": [30, 49, 61, 67, 77, 84, 119, 121, 122, 123, 124, 128, 130, 131, 153, 173, 176, 177], "split_path": 30, "dataframe_path": 30, "orig_fnam": 31, "origin": [31, 45], "filenam": [31, 39, 41, 43], "extens": 31, "out": [31, 64, 73, 74, 138, 152, 154, 173, 176], "rule": [31, 177], "csv_path": [32, 33], "img_path_column": [32, 33], "channel_column": [32, 33], "out_kei": [32, 38], "3": [32, 33, 49, 51, 76, 77, 107, 108, 112, 115, 117, 138, 145, 147, 148, 149, 150, 151, 152, 153, 154, 168, 174, 177], "scene_column": 32, "scene": [32, 33, 38, 39], "time_start_column": 32, "time_stop_column": 32, "stop": 32, "time_step_column": 32, "dict_meta": 32, "dask_load": [32, 38, 39], "dimension": [32, 100, 101, 102, 151], "timelaps": 32, "metadata": [32, 45, 165], "timepoint": [32, 33, 38, 39], "extract": [32, 38, 39, 83, 84, 85, 152], "spatial": [32, 33, 49, 76, 107, 115, 126, 138, 145, 148, 149, 151, 152, 153, 154, 177], "yx": [32, 49, 148, 149, 152, 177], "zyx": [32, 49, 148, 149, 152, 154, 177], "specifi": [32, 33, 38, 39, 48, 53, 81, 90, 149, 151, 152, 173, 176, 177], "multipl": [32, 51, 107, 108, 138, 153, 173, 176], "thei": [32, 45, 77], "comma": [32, 38], "scene1": 32, "scene2": 32, "ani": [32, 59, 79, 126, 173, 176, 177], "start_column": 32, "stop_column": 32, "step_column": 32, "For": [32, 51, 115, 138, 151, 173, 176, 177], "everi": [32, 62, 163], "other": [32, 173, 176, 177], "version": [32, 107, 177], "accept": [32, 126], "normal": [32, 108, 138, 143, 144, 145, 173, 176, 177], "could": 32, "here": [32, 33, 61, 121, 173, 176, 177], "dask": [32, 38, 39], "full": [32, 38, 39], "memori": [32, 33, 38, 39, 177], "befor": [32, 38, 39, 51, 57, 128, 172], "img": [32, 39, 41, 43, 57, 76, 81, 140, 148, 151, 152, 154], "meta": 32, "df": [32, 33], "x": [32, 61, 62, 64, 74, 85, 90, 107, 108, 110, 111, 112, 115, 117, 120, 129, 132, 138, 139, 143, 144, 145, 149, 151, 177], "listconfig": [32, 52, 57, 163], "multidimimag": 32, "cannot": 32, "dim": [32, 76, 107, 108, 116, 132, 145], "multiscen": 32, "alreadi": [32, 173, 176], "compos": [33, 167], "img_data": 33, "n_val": 33, "20": 33, "pct_val": 33, "raw": [33, 38, 51, 173, 176, 177], "ch": [33, 46, 79], "num_neighbor": 33, "4": [33, 140, 143, 145, 150, 151, 152, 154], "cache_r": 33, "5": [33, 49, 51, 76, 129], "replace_r": 33, "czi": 33, "don": 33, "fit": 33, "bioio": 33, "n_imag": 33, "percentag": 33, "neighbor": [33, 76], "worker": 33, "most": [33, 61], "schedul": 33, "parallel": 33, "submodul": [36, 156, 175], "aicsimage_load": [36, 37], "aicsimageloaderd": [36, 37, 38], "monai_bio_read": [36, 37, 38], "monaibioread": [36, 37, 39], "numpy_read": [36, 37], "readnumpyfil": [36, 37, 40], "ome_zarr_read": [36, 37], "omezarrread": [36, 37, 41], "polygon_load": [36, 37], "polygonloaderd": [36, 37, 42], "skimage_read": [36, 37], "skimageread": [36, 37, 43], "add_meta": [36, 44], "addmeta": [36, 44, 45], "bright_sampl": [36, 44], "brightsampl": [36, 44, 46], "clip": [36, 44, 48], "clipd": [36, 44, 47], "contrastadjust": [36, 44], "contrastadjustd": [36, 44, 48], "generate_jepa_mask": [36, 44], "jepamaskgener": [36, 44, 49], "merg": [36, 44], "multiscale_cropp": [36, 44], "randommultiscalecropd": [36, 44, 51], "pad": [36, 44, 76, 138], "padzd": [36, 44, 52], "project": [36, 44, 145, 173, 176], "maxprojectd": [36, 44, 53], "rotation_mask_transform": [36, 44], "so2_random_rot": [36, 44], "track_transform": [36, 44], "cropres": [36, 44, 57], "generatetracklabel": [36, 44, 57], "perchannel": [36, 44, 57], "splittrackd": [36, 44, 57], "split_arg": [37, 38], "get_data": [37, 39, 41, 43], "verify_suffix": [37, 39, 41, 43], "path_kei": 38, "scene_kei": 38, "kwargs_kei": 38, "dimension_order_out": [38, 177], "float16": 38, "channel_kei": 38, "timepoint_kei": 38, "fix": [38, 148, 150, 151, 152, 153, 155], "initi": [38, 77, 115, 123, 166], "bioimag": [38, 39], "get_image_dask_data": [38, 39], "miss": [38, 42, 47, 48, 52, 53, 55, 57, 76, 81], "adapt": [39, 43, 120], "github": [39, 43, 120, 150, 174], "com": [39, 43, 120, 174], "mmv": [39, 43], "lab": [39, 43], "mmv_im2im": [39, 43], "reader_kwarg": 39, "imageread": [39, 41, 43], "get_image_data": 39, "pathlik": [39, 41, 43], "maptransform": 40, "thereof": 40, "interpret": 40, "point": [40, 51, 76, 138, 177], "cloud": [40, 120], "remot": 40, "fsspec": 40, "locat": [40, 173, 176], "level": [41, 149, 173, 176, 177], "image_nam": 41, "shape_reference_kei": 42, "propagate_3d": 42, "missing_key_mod": 42, "rais": [42, 53, 76, 81], "polygon": 42, "vertic": 42, "tensor": [42, 81, 84, 85, 101, 123, 124, 131, 138, 145, 147, 154, 168, 177], "mask": [42, 49, 50, 85, 114, 145, 147, 148, 149, 150, 151, 152, 154], "propag": 42, "2d": [42, 49, 83, 115, 138, 148, 149, 168, 173, 176, 177], "3d": [42, 49, 76, 83, 115, 138, 148, 149, 168, 173, 176, 177], "how": [42, 115, 165, 173, 176], "keyerror": 42, "do": [42, 115, 123, 173, 176], "noth": 42, "new": [42, 45, 173, 176, 177], "blank": 42, "remove_excess_pixel": [44, 49], "generate_slic": [44, 51], "image_kei": [45, 50, 52, 76], "delet": 45, "updat": [45, 61], "metatensor": [45, 165], "been": [45, 151], "threshold": [46, 79, 128], "base_prob": 46, "decid": [46, 51], "intens": [46, 47, 48, 79, 177], "background": [46, 48], "rate": [46, 138, 145], "probabilii": 46, "check": [46, 52], "against": 46, "low": [47, 48, 49], "01": 47, "high": [47, 48, 49], "99": 47, "percentil": [47, 48], "absolut": [47, 48], "lower": [47, 48], "bound": [47, 48], "upper": [47, 48], "per_channel": 47, "resiz": [47, 48, 51, 57, 177], "fail": [47, 48], "adjust": 48, "rang": [48, 49, 79, 173, 176, 177], "outsid": 48, "everywher": 48, "mask_siz": 49, "12": [49, 145, 151, 152, 153, 154], "block_aspect_ratio": 49, "num_patch": [49, 148, 150, 151, 152, 153, 154, 155], "6": [49, 76, 154], "24": 49, "mask_ratio": [49, 147, 148, 149, 151, 152, 154], "9": [49, 174], "randomizabletransform": [49, 51, 57, 76], "block": [49, 105, 115, 138, 141, 151, 152], "contigu": 49, "work": [49, 101, 148, 150, 151, 152, 153, 174, 177], "until": [49, 177], "met": 49, "exceed": [49, 177], "exterior": 49, "exactli": 49, "determin": [49, 115, 138, 167, 173, 176, 177], "aspect": 49, "ratio": [49, 115, 138, 145, 152, 154], "proport": 49, "pixel": [49, 57, 76, 126, 148, 149, 151, 152, 177], "along": 49, "boundari": 49, "reach": 49, "mask_kei": [50, 114], "base_image_kei": 50, "output_nam": 50, "serv": 50, "patch_shap": [51, 139, 177], "scales_dict": 51, "patch_per_imag": 51, "selection_fn": 51, "max_attempt": 51, "style": [51, 53, 152, 177], "slice": [51, 52, 76, 177], "scale": [51, 107, 111, 112, 123, 126], "superresolut": 51, "x_kei": [51, 64, 73, 74, 83, 85], "network": [51, 62, 64, 73, 74, 90, 115, 138, 177], "Its": 51, "resolut": [51, 112, 115, 151], "len": 51, "map": [51, 110, 138, 177], "seg": [51, 76, 105, 141, 173, 176], "would": [51, 61], "kept": 51, "discard": 51, "max": [51, 53], "attempt": 51, "try": [51, 177], "quit": 51, "image_dict": 51, "segmentation_kei": 52, "pad_amount": 52, "pad_kei": 52, "segmentation_ch": 52, "top": [52, 154, 177], "bottom": 52, "crop": 52, "repeat": [52, 151, 152, 155], "first": [52, 101, 108, 151, 168, 173, 176, 177], "present": [52, 76], "offset": 52, "presenc": 52, "projection_dim": 53, "nczyx": 53, "comput": [53, 101, 128, 168], "error": [53, 76, 81], "save_path": 55, "disk": 55, "debug": 55, "max_shift": 57, "8": [57, 101, 140, 143, 145, 149, 151, 152, 154, 173, 176], "maximum": [57, 83, 84, 85], "shift": 57, "img_kei": 57, "formation_kei": 57, "breakdown_kei": 57, "breakdown": 57, "track_start_kei": 57, "track_start": 57, "label_kei": [57, 76], "label": [57, 64, 67, 76, 79, 123], "track": 57, "after_save_checkpoint": [58, 59], "log_hyperparam": [58, 59], "log_metr": [58, 59], "experiment_nam": [59, 173, 176], "lightning_log": 59, "run_nam": [59, 173, 176], "tracking_uri": [59, 87], "tag": [59, 153, 167, 172], "mlrun": 59, "prefix": 59, "artifact_loc": 59, "run_id": [59, 87], "fault_toler": 59, "ckpt_callback": 59, "param": [59, 110], "namespac": 59, "mode": [59, 87, 100, 101, 102, 103, 117, 121, 130, 172], "metric": [59, 61, 110, 113, 114, 116, 172], "timepoint_classif": [60, 63], "timepointclassif": [60, 63, 65], "base_handl": [60, 68], "image_handl": [60, 68], "load_image_patch": [60, 68], "multi_task": [60, 72], "multitaskim2im": [60, 72, 73, 74], "ijepa": [60, 82, 85], "iwm": [60, 82, 85], "jepa_bas": [60, 82], "jepabas": [60, 82, 83, 84, 85], "load_model_from_checkpoint": [60, 86, 87], "find_indic": [60, 86, 88], "o2_spharm_va": [60, 89], "prior": [60, 89, 90], "base_va": [60, 89], "baseva": [60, 89, 90], "image_canon_va": [60, 89], "image_encod": [60, 89], "image_va": [60, 89], "latent_loss_va": [60, 89], "point_cloud_va": [60, 89], "tabular_va": [60, 89], "compute_metr": [60, 61], "configure_optim": [60, 61, 72, 73, 74, 82, 85], "model_step": [60, 61, 62, 63, 64, 66, 67, 72, 73, 74, 82, 83, 84, 85, 89, 90], "on_train_start": [60, 61], "parse_batch": [60, 61], "predict_step": [60, 61, 63, 64, 65, 66, 67, 72, 73, 74, 82, 84, 85], "test_step": [60, 61, 72, 74, 82, 84], "training_step": [60, 61, 72, 74], "validation_step": [60, 61, 72, 74], "log": [61, 62, 101, 172, 173, 176, 177], "assumpt": 61, "_step": 61, "whose": 61, "mai": 61, "self": [61, 76, 83, 84, 151, 152], "structur": [61, 167, 177], "part": [61, 90, 172, 173, 176], "arbitrari": 61, "denot": 61, "you": [61, 173, 176, 177], "implement": [61, 173, 176, 177], "process": [61, 172, 177], "variabl": [61, 173, 176], "hparam": 61, "id": 61, "detach": [61, 75, 78, 81, 110, 111, 112, 113, 114, 115], "consist": [61, 154, 173, 176], "your": [61, 173, 176, 177], "might": 61, "wish": 61, "post": 61, "minim": 62, "wrapper": [62, 126, 128], "around": [62, 148, 149, 151, 152], "assum": [62, 168, 173, 176, 177], "gt": [62, 120], "optim": [62, 111, 112, 173, 176, 177], "retriev": [62, 172], "adam": 62, "save_predict": 62, "serotini": [62, 173, 176], "fields_to_log": 62, "field": [62, 167, 177], "pretrained_weight": 62, "pretrain": [62, 154], "via": [62, 151], "load_state_dict": 62, "save_imag": [63, 64], "should_save_imag": [63, 64], "share": [64, 73, 74, 177], "save_images_every_n_epoch": [64, 73, 74], "base_kwarg": [64, 67, 73, 74, 83, 84, 85, 90], "logit": [64, 115, 173, 176], "text": 64, "overlaid": 64, "plot_class": [66, 67], "plot_neighbor": [66, 67], "backbon": [67, 73, 74, 110, 113, 114, 115, 116, 154, 173, 176, 177], "task_head": [67, 73, 74, 161, 177], "anchor_kei": 67, "anchor": 67, "positive_kei": 67, "posit": [67, 140, 147, 148, 149, 150, 151, 152, 153, 155], "target_kei": [67, 84, 90], "visual": [67, 173, 176], "viz_freq": 67, "x1": 67, "x2": 67, "embedding1": 67, "embedding2": 67, "instancesegclust": [72, 75, 76], "instancesegloss": [72, 75, 76], "instancesegpreprocessd": [72, 75, 76], "instancesegrandflipd": [72, 75, 76], "pad_slic": [72, 75, 76], "noise_anneal": [72, 75, 108], "noiseanneal": [72, 75, 77], "get_n_postprocess_imag": [72, 74], "get_per_head": [72, 74], "run_forward": [72, 74, 110, 111, 113, 114, 116], "basic": [73, 90, 173, 176, 177], "inference_arg": [73, 74], "slide": [73, 74], "window": [73, 74], "inference_head": 74, "run_head": [74, 109, 110, 111, 112, 113, 114, 116], "n_postprocess": [74, 110, 111, 113, 114, 116], "act_thresh_label": [75, 78], "actthreshlabel": [75, 78, 79], "auto_thresh": [75, 78], "autothreshold": [75, 78, 80], "dict_to_im": [75, 78], "dicttoim": [75, 78, 81], "cluster_object": [75, 76], "kd_cluster": [75, 76], "remove_small_skeleton": [75, 76], "embed_from_skel": [75, 76], "keep_largest_cc": [75, 76], "label_2d": [75, 76], "shrink": [75, 76], "skeleton_tal": [75, 76], "smooth_embed": [75, 76], "topology_preserving_thin": [75, 76], "update_nois": [75, 77], "anisotropi": 76, "skel_threshold": 76, "semantic_threshold": 76, "min_siz": 76, "1000": 76, "distance_threshold": 76, "progress": 76, "cluster": 76, "instanceseg": 76, "find": 76, "skeleton": 76, "assign": 76, "semant": 76, "embed": [76, 84, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155], "nearest": [76, 117], "distanc": [76, 177], "skel": 76, "closest": 76, "below": [76, 128, 173, 176], "touch": 76, "edg": [76, 177], "compon": [76, 167], "kernel_s": [76, 138, 139], "thin": 76, "keep_largest": 76, "instanc": [76, 77, 138, 154, 177], "kernel": [76, 138], "gaussian": [76, 77, 89, 99], "smooth": 76, "flow": 76, "amount": [76, 115, 138], "psuedo": 76, "largest": 76, "connect": [76, 115, 138, 154], "iseg": 76, "vector": [76, 168, 177], "z": [76, 90, 100, 101, 102, 152, 168, 173, 176], "im": [76, 108], "topologi": 76, "preserv": 76, "binari": 76, "max_label": 76, "increas": [76, 77, 177], "thick": 76, "convolv": 76, "non": [76, 151], "bw": 76, "bridg": 76, "gap": 76, "eros": 76, "spatial_axi": 76, "prob": [76, 121], "flip": [76, 177], "augment": [76, 173, 176, 177], "sign": 76, "gradient": 76, "chang": [76, 173, 176, 177], "axi": [76, 168], "NOT": 76, "probabl": 76, "constraint": 76, "annealing_step": 77, "5000": 77, "init_vari": 77, "anneal": [77, 108], "varianc": [77, 101], "nois": [77, 108], "real": [77, 107, 123], "fake": [77, 123], "harder": 77, "distribut": [77, 102], "overlap": 77, "nice": 77, "theoret": 77, "implic": 77, "qualiti": 77, "also": [77, 101, 138, 173, 176, 177], "curriculum": 77, "learn": [77, 83, 84, 177], "techniqu": 77, "iter": 77, "unblur": 77, "over": 77, "linearli": 77, "activ": [79, 107, 115, 138, 143, 145, 174], "ident": [79, 112, 115, 138, 173, 176], "_type": 79, "_dtype_lik": 79, "_supportsdtyp": 79, "supportsindex": 79, "collect": 79, "_dtypedict": 79, "uint8": 79, "rescale_dtyp": 79, "purpos": 79, "rescal": [79, 173, 176], "dtypelik": 79, "multichannel": 81, "concaten": 81, "cuda": 81, "extract_embed": [82, 84], "get_predict_mask": [82, 84], "get_context_embed": [82, 85], "get_mask": [82, 85, 146, 148], "get_target_embed": [82, 85], "remove_first_dim": [82, 85], "update_teach": [82, 85], "supervis": [83, 84], "featur": [83, 84, 85, 110, 113, 114, 116, 123, 138, 143, 144, 145, 150, 151, 152, 153, 154, 173, 176], "access": [83, 84, 85], "momentum": [83, 84, 85], "exponenti": [83, 84, 85], "move": [83, 84, 85], "averag": [83, 84, 85, 126], "998": [83, 84, 85], "world": 84, "translat": 84, "latent": [84, 90], "space": [84, 90], "target_domain_kei": 84, "domain": [84, 153], "predict_target": 84, "devic": [84, 155], "tmp_dir": 87, "strict": 87, "ckpt": [87, 177], "lst": 88, "o2_spharm_encod": [89, 95], "abstract_prior": [89, 99], "diagonalgaussianprior": [89, 99, 101], "isotropicgaussianprior": [89, 99, 101], "compute_tc_penalti": [89, 99, 101], "identity_prior": [89, 99], "identityprior": [89, 99, 102], "joint_prior": [89, 99], "jointprior": [89, 99, 103], "calculate_elbo": [89, 90], "calculate_rcl": [89, 90], "calculate_rcl_dict": [89, 90], "decod": [89, 90, 105, 141, 143, 152, 154], "sample_z": [89, 90], "beta": [90, 130], "kld": 90, "term": [90, 138], "reconstruction_loss": [90, 111, 112], "reconstruct": [90, 111, 112, 160, 169], "_loss": [90, 119, 121, 122, 124, 130, 131], "abstractprior": 90, "decoder_latent_part": 90, "decoer": 90, "what": [90, 167], "depend": [90, 174], "xhat": 90, "input_kei": 90, "return_param": 90, "z_parts_param": 90, "kl_diverg": [99, 100, 101, 103], "param_s": [99, 100, 101, 102, 103], "kl": [100, 101, 102, 103], "properti": [100, 101, 102, 103, 147, 148, 149, 152], "logvar": 101, "learn_mean": 101, "learn_logvar": 101, "clamp_logvar": 101, "tc_penalty_weight": 101, "mu1": 101, "mu2": 101, "logvar1": 101, "logvar2": 101, "reduct": [101, 103, 119, 121, 122, 130, 131], "sum": [101, 103, 130, 144, 151, 154], "kullback": 101, "leibler": 101, "diverg": 101, "diagon": 101, "necessarili": 101, "isotrop": 101, "wise": 101, "second": [101, 151], "doesn": [102, 174], "contribut": 102, "effect": 102, "dirac": 102, "delta": 102, "z_param": 103, "multi_scale_discrimin": [105, 106], "multiscalediscrimin": [105, 106, 107], "n_layer_discrimin": [105, 106], "nlayerdiscrimin": [105, 106, 107, 108], "base_head": [105, 109], "basehead": [105, 109, 110, 111, 113, 114, 115, 116], "gan_head": [105, 109], "ganhead": [105, 109, 111, 112], "gan_head_superr": [105, 109], "ganhead_res": [105, 109, 112], "mae_head": [105, 109], "maehead": [105, 109, 113], "mask_head": [105, 109], "maskhead": [105, 109, 114], "res_blocks_head": [105, 109], "resblockshead": [105, 109, 112, 115], "vic_reg": [105, 109, 118], "vicreghead": [105, 109, 116], "adversarial_loss": [105, 118], "adversarialloss": [105, 118, 119], "chamfer_loss": [105, 118], "chamferloss": [105, 118, 120], "continuous_bernoulli": [105, 118], "cblogloss": [105, 118, 121], "cosine_loss": [105, 118], "cosineloss": [105, 118, 122], "gan_loss": [105, 111, 112, 118], "ganloss": [105, 111, 112, 118, 123], "pix2pixhd": [105, 107, 111, 112, 118, 123], "gaussian_nll_loss": [105, 118], "gaussiannllloss": [105, 118, 124], "geomloss": [105, 118], "loss_wrapp": [105, 118], "cmap_loss": [105, 118, 126], "losswrapp": [105, 118, 126], "spharm_loss": [105, 118], "threshold_loss": [105, 118], "thresholdloss": [105, 118, 128], "vicregloss": [105, 118, 129], "off_diagon": [105, 118, 129], "weibul": [105, 118], "weibulllogloss": [105, 118, 130], "weibull_log_prob": [105, 118, 130], "weighted_mse_loss": [105, 118], "weightedmseloss": [105, 118, 131], "dgcnn": [105, 133], "folding_net": [105, 133], "graph_funct": [105, 133], "vnn": [105, 133], "crossmae_decod": [105, 141, 150], "mae_decod": [105, 141, 150], "hieraencod": [105, 141, 151], "jepaencod": [105, 141, 151], "mae_encod": [105, 141, 151], "spatialmerg": [105, 141, 151], "mae": [105, 141], "hierama": [105, 141, 152], "mae_bas": [105, 141, 152], "iwmpredictor": [105, 141, 153], "jepapredictor": [105, 141, 153], "encodedskip": [105, 141, 154], "seg_vit": [105, 141, 154], "superresdecod": [105, 141, 154], "get_positional_embed": [105, 141, 155], "match_tuple_dimens": [105, 141, 155], "random_index": [105, 141, 155], "take_index": [105, 141, 155], "n_scale": 107, "input_nc": 108, "ndf": 108, "n_layer": 108, "norm_lay": [108, 143, 144, 145], "instancenorm": 108, "instancenorm3d": 108, "defin": [108, 123], "patchgan": 108, "n_channel": 108, "input_im": 108, "model_output": 108, "conv_fn": 108, "later": 108, "requires_featur": 108, "standard": 108, "generate_io_map": [109, 110], "update_param": [109, 110], "input_filenam": 110, "backbone_featur": [110, 111, 113, 114, 116], "y_hat": [110, 111, 113, 114, 116], "calcul": [110, 113, 114, 116, 126], "bcewithlogitsloss": [111, 112, 123], "feature_matching_loss": [111, 112], "l1loss": [111, 112], "mseloss": [111, 112], "reconstruction_loss_weight": [111, 112], "in_channel": [112, 115, 117, 138, 177], "out_channel": [112, 115, 117, 138, 154, 177], "final_act": [112, 115], "lr": [112, 115], "n_conv": [112, 115], "dropout": [112, 115, 117, 138, 145], "upsample_method": [112, 115], "pixelshuffl": [112, 115], "upsample_ratio": [112, 115], "first_lay": [112, 115], "dens": [112, 115], "inherit": 112, "__init__": 112, "calculate_metr": 114, "convolut": [115, 138, 148, 149, 151, 152, 154, 173, 176], "final": 115, "done": 115, "hr": 115, "length": [115, 126, 140, 155], "convprojectionlay": 115, "2048": 116, "8192": 116, "hidden_lay": [116, 132], "projectornetwork": 116, "hidden": [116, 145], "scale_factor": 117, "align_corn": 117, "antialia": 117, "batch_pairwise_dist": [118, 120], "batch_pairwise_dist2": [118, 120], "get_target_tensor": [118, 123], "get_feature_matching_loss": [118, 123], "get_gan_loss": [118, 123], "argmax": 119, "squeez": 119, "return_pr": 119, "sentinal4d": 120, "cellshap": 120, "blob": 120, "cellshape_cloud": 120, "vendor": 120, "chamfer_dist": 120, "py": [120, 177], "licens": 120, "license_antao": 120, "y": [120, 129, 143], "continu": 121, "bernoulli": 121, "propos": 121, "arxiv": 121, "ab": 121, "1907": 121, "06845": 121, "gan_mod": 123, "vanilla": 123, "target_real_label": 123, "target_fake_label": 123, "abstract": [123, 148, 152, 173, 176], "awai": 123, "need": [123, 173, 176, 177], "lsgan": 123, "wgangp": 123, "note": [123, 177], "sigmoid": [123, 173, 176], "target_is_r": 123, "fill": 123, "loss_weight": 123, "fm": 123, "10": [123, 124, 130, 138], "feature_typ": 123, "mean_dim": 124, "ep": [124, 130, 138, 168], "1e": [124, 130, 138, 168], "costmap": 126, "differenti": [126, 173, 176], "emphas": 126, "throughout": 126, "loss_fn": [126, 128], "channel_weight": 126, "_supportsarrai": 126, "_nestedsequ": 126, "complex": [126, 173, 176], "byte": 126, "loss_scal": 126, "arraylik": 126, "abov": 128, "num_featur": 129, "sim_coeff": 129, "std_coeff": 129, "cov_coeff": 129, "alpha": 130, "256": [132, 139], "scale_output": 132, "stride": [138, 145, 151, 152], "subunit": 138, "adn_ord": 138, "nda": 138, "prelu": 138, "norm": [138, 144], "dropout_dim": 138, "dilat": 138, "bia": [138, 145], "last_conv_onli": 138, "residu": 138, "conv": 138, "init": 138, "normalized_shap": 138, "print": [138, 167, 172], "sequenti": 138, "unit0": 138, "conv3d": 138, "adn": 138, "num_paramet": 138, "n": [138, 173, 174, 176], "layernorm": [138, 143, 144, 145], "05": 138, "elementwise_affin": 138, "unit1": 138, "zero": [138, 172], "entir": 138, "larger": 138, "control": [138, 172], "implicit": 138, "both": [138, 177], "side": 138, "in_filt": 139, "out_filt": 139, "n_input_ch": 139, "512": 139, "n_conv_filt": 139, "pos_embedding_length": 140, "120": 140, "emb_dim": [140, 143, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155], "num_lay": [140, 144, 150, 151, 153, 154], "num_head": [140, 143, 145, 150, 151, 152, 153], "num_class": 140, "token": [140, 145, 147, 149, 150, 155], "classifi": 140, "them": [140, 173, 176], "d_model": 140, "matrix": [140, 168], "patchifi": [141, 142, 151], "cross_attent": [141, 142], "crossattent": [141, 142, 143], "crossattentionblock": [141, 142, 143], "crossselfblock": [141, 142, 143], "intermediate_weigh": [141, 142], "intermediateweigh": [141, 142, 144], "masked_unit_attent": [141, 142], "hierablock": [141, 142, 145], "maskunitattent": [141, 142, 145], "add_mask_token": [141, 150], "adjust_indices_for_cl": [141, 150], "init_weight": [141, 150, 151, 153], "init_decod": [141, 152], "init_encod": [141, 152], "predict_target_featur": [141, 153], "patchify_bas": [142, 146], "patchifybas": [142, 146, 147, 148, 149], "patchify_hiera": [142, 146], "patchifyhiera": [142, 146, 149], "take_indexes_mask": [142, 146, 149], "encoder_dim": 143, "decoder_dim": [143, 152], "qkv_bia": [143, 145], "qk_scale": 143, "attn_drop": [143, 145], "proj_drop": [143, 145], "mlp_ratio": [143, 145], "drop_path": [143, 145], "act_lay": [143, 145], "gelu": [143, 145], "in_featur": 143, "hidden_featur": 143, "out_featur": 143, "embed_dim": [144, 151], "n_output": 144, "intermedi": [144, 151, 154], "dim_out": 145, "q_stride": [145, 151, 152], "patches_per_mask_unit": 145, "attent": [145, 151, 152], "unit": [145, 149, 151, 152], "learnabl": [145, 148, 150, 151, 152, 153, 155], "create_img2token": [146, 147, 149], "extract_visible_token": [146, 147, 148, 149], "get_mask_arg": [146, 147, 148, 149], "img2token": [146, 147, 148, 149], "create_conv": [146, 148], "create_patch2img": [146, 148], "n_patch": [147, 148, 149], "context_pixel": [147, 148, 149, 151, 152], "learnable_pos_embed": [147, 148, 150, 151, 152, 153], "rearrang": 147, "forward_index": [147, 149, 150], "n_visible_patch": [147, 148, 149], "pix": [148, 149], "sin": [148, 150, 151, 152, 153, 168], "co": [148, 150, 151, 152, 153, 168], "empir": [148, 150, 151, 152, 153], "brightfield": [148, 150, 151, 152, 153], "boolean": [148, 177], "mask_units_per_dim": 149, "mask_unit": 149, "enc_dim": 150, "768": [150, 152, 154], "192": [150, 151, 152, 153, 154], "has_cls_token": 150, "inspir": 150, "crossma": [150, 152], "attend": 150, "visibl": 150, "cl": [150, 155], "backward_index": 150, "num_mask_unit": [151, 152], "architectur": [151, 152, 177], "save_lay": 151, "multihead": [151, 152], "self_attent": [151, 152], "On": 151, "doubl": 151, "pool": 151, "doe": [151, 177], "while": [151, 177], "produc": [151, 173, 176], "context": [151, 153], "n_intermediate_weight": 151, "75": [151, 152, 154], "downsample_factor": 151, "in_dim": 151, "out_dim": 151, "hiera": 151, "lowest": 151, "decoder_lay": [152, 154], "decoder_head": 152, "use_crossma": 152, "features_onli": 152, "encoder_lay": [152, 154], "encoder_head": [152, 154], "input_dim": 153, "special": 153, "condition": 153, "fluoresc": 153, "context_emb": 153, "target_mask": 153, "target_domain": 153, "n_decoder_filt": 154, "base_patch_s": 154, "upsample_factor": 154, "6134": 154, "5005": 154, "encoder_ckpt": 154, "freeze_encod": 154, "encoder_kwarg": 154, "simpl": [154, 173, 176], "aic": 154, "20x": 154, "100x": 154, "unet": 154, "like": [154, 177], "fed": [154, 173, 176], "skip": 154, "use_cls_token": 155, "ensur": [155, 177], "read_pcloud": [156, 157], "generate_img": [160, 165], "compute_rotation_matrix": [160, 168], "data_cfg": 161, "load_param": 162, "kv": 163, "item": 163, "nest": 163, "form": [163, 168], "interpol": [163, 173, 176], "limit": [164, 173, 176, 177], "fn": 164, "num_sampl": 165, "dummy_metadata": 165, "10000": 165, "mani": [165, 173, 176, 177], "k": 165, "friendli": 166, "save_to_fil": 167, "prompt": 167, "user": [167, 173, 176], "print_ord": 167, "resolv": 167, "content": [167, 172], "rich": [167, 172], "librari": 167, "tree": 167, "hydra": [167, 173, 176, 177], "refer": [167, 173, 176], "export": 167, "pad_valu": 168, "06": 168, "pose": 168, "about": 168, "equivari": [168, 174], "theta": 168, "indic": 168, "pair": 168, "precis": 168, "sure": 172, "close": 172, "properli": 172, "prevent": 172, "failur": 172, "multirun": 172, "warn": 172, "metric_dict": 172, "metric_nam": 172, "safe": 172, "callbacks_cfg": 172, "logger_cfg": 172, "object_dict": 172, "rank": 172, "task_func": 172, "decor": 172, "resist": 172, "finish": 172, "except": 172, "occur": 172, "total": 172, "execut": 172, "dir": 172, "aim": [173, 176], "common": [173, 176], "manner": [173, 176], "easi": [173, 176], "beginn": [173, 176], "flexibl": [173, 176, 177], "custom": [173, 176], "templat": [173, 176, 177], "conveni": [173, 176], "syntax": [173, 176], "repo": [173, 176], "best": [173, 176], "familiar": [173, 176, 177], "mechan": [173, 176], "topic": [173, 176], "cover": [173, 176], "becaus": [173, 176], "help": [173, 176], "These": [173, 176], "reli": [173, 176], "heavili": [173, 176], "omegaconf": [173, 176], "expos": [173, 176], "_aux": [173, 176], "section": [173, 176], "persist": [173, 176], "speed": [173, 176], "up": [173, 176, 177], "divid": [173, 176], "manifest": [173, 176], "preprocess": [173, 176], "offer": [173, 176], "wide": [173, 176], "often": [173, 176, 177], "primari": [173, 176], "due": [173, 176], "score": [173, 176], "v": [173, 176], "binar": [173, 176, 177], "own": [173, 176], "thing": [173, 176], "pin_memori": [173, 176], "workhors": [173, 176], "source_col": [173, 176, 177], "target_col": [173, 176], "organ": [173, 176], "arrang": [173, 176], "through": [173, 176], "further": [173, 176], "lr_schedul": [173, 176], "alwai": [173, 176], "well": [173, 176], "_target_": [173, 176], "_partial_": [173, 176], "sigmoid_rescal": [173, 176], "tell": [173, 176], "bit": [173, 176], "encourag": [173, 176], "few": [173, 176, 177], "git": 174, "clone": 174, "allencellmodel": 174, "cd": 174, "conda": 174, "environ": 174, "myenv": 174, "pip": 174, "txt": 174, "relat": 174, "equiv": 174, "upgrad": 174, "subpackag": 175, "outlin": 177, "modif": 177, "suit": 177, "aicsimageio": 177, "recommend": 177, "suitabl": 177, "robust": 177, "again": 177, "excel": 177, "naiv": 177, "incorrect": 177, "neural": 177, "signal": 177, "cut": 177, "crucial": 177, "bottleneck": 177, "suggest": 177, "look": 177, "consider": 177, "primarili": 177, "As": 177, "thumb": 177, "enough": 177, "entiti": 177, "establish": 177, "too": 177, "downsampl": 177, "decreas": 177, "usag": 177, "ckpt_path": 177, "dictat": 177, "omit": 177}, "objects": {"": [[1, 0, 0, "-", "cyto_dl"]], "cyto_dl": [[2, 0, 0, "-", "api"], [8, 0, 0, "-", "callbacks"], [17, 0, 0, "-", "dataframe"], [24, 0, 0, "-", "datamodules"], [35, 0, 0, "-", "eval"], [36, 0, 0, "-", "image"], [58, 0, 0, "-", "loggers"], [60, 0, 0, "-", "models"], [105, 0, 0, "-", "nn"], [156, 0, 0, "-", "point_cloud"], [159, 0, 0, "-", "train"], [160, 0, 0, "-", "utils"]], "cyto_dl.api": [[3, 0, 0, "-", "cyto_dl_model"], [6, 0, 0, "-", "data"], [7, 0, 0, "-", "model"]], "cyto_dl.api.cyto_dl_model": [[4, 0, 0, "-", "cyto_dl_base_model"], [5, 0, 0, "-", "segmentation_plugin_model"]], "cyto_dl.api.cyto_dl_model.cyto_dl_base_model": [[4, 1, 1, "", "CytoDLBaseModel"]], "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel": [[4, 2, 1, "", "from_default_config"], [4, 2, 1, "", "from_existing_config"], [4, 2, 1, "", "get_config"], [4, 2, 1, "", "get_experiment_name"], [4, 2, 1, "", "predict"], [4, 2, 1, "", "save_config"], [4, 2, 1, "", "set_experiment_name"], [4, 2, 1, "", "train"]], "cyto_dl.api.cyto_dl_model.segmentation_plugin_model": [[5, 1, 1, "", "SegmentationPluginModel"]], "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel": [[5, 2, 1, "", "get_hardware_type"], [5, 2, 1, "", "get_input_channel"], [5, 2, 1, "", "get_manifest_column_names"], [5, 2, 1, "", "get_patch_size"], [5, 2, 1, "", "get_raw_image_channels"], [5, 2, 1, "", "get_split_column"], [5, 2, 1, "", "remove_split_column"], [5, 2, 1, "", "set_hardware_type"], [5, 2, 1, "", "set_input_channel"], [5, 2, 1, "", "set_manifest_column_names"], [5, 2, 1, "", "set_patch_size"], [5, 2, 1, "", "set_raw_image_channels"], [5, 2, 1, "", "set_split_column"]], "cyto_dl.api.data": [[6, 1, 1, "", "AutoThresholdMethod"], [6, 1, 1, "", "ExperimentType"], [6, 1, 1, "", "HardwareType"], [6, 1, 1, "", "PatchSize"]], "cyto_dl.api.data.AutoThresholdMethod": [[6, 3, 1, "", "ISODATA"], [6, 3, 1, "", "LI"], [6, 3, 1, "", "LOCAL"], [6, 3, 1, "", "MEAN"], [6, 3, 1, "", "MINIMUM"], [6, 3, 1, "", "MULTIOTSU"], [6, 3, 1, "", "NIBLACK"], [6, 3, 1, "", "OTSU"], [6, 3, 1, "", "SAUVOLA"], [6, 3, 1, "", "TRIANGLE"], [6, 3, 1, "", "YEN"]], "cyto_dl.api.data.ExperimentType": [[6, 3, 1, "", "GAN"], [6, 3, 1, "", "INSTANCE_SEG"], [6, 3, 1, "", "LABEL_FREE"], [6, 3, 1, "", "SEGMENTATION"], [6, 3, 1, "", "SEGMENTATION_ARRAY"], [6, 3, 1, "", "SEGMENTATION_PLUGIN"]], "cyto_dl.api.data.HardwareType": [[6, 3, 1, "", "CPU"], [6, 3, 1, "", "GPU"]], "cyto_dl.api.data.PatchSize": [[6, 3, 1, "", "LARGE"], [6, 3, 1, "", "MEDIUM"], [6, 3, 1, "", "SMALL"]], "cyto_dl.api.model": [[7, 1, 1, "", "CytoDLModel"]], "cyto_dl.api.model.CytoDLModel": [[7, 2, 1, "", "download_example_data"], [7, 2, 1, "", "load_config_from_dict"], [7, 2, 1, "", "load_config_from_file"], [7, 2, 1, "", "load_default_experiment"], [7, 2, 1, "", "override_config"], [7, 2, 1, "", "predict"], [7, 2, 1, "", "print_config"], [7, 2, 1, "", "save_config"], [7, 2, 1, "", "train"]], "cyto_dl.callbacks": [[10, 0, 0, "-", "csv_saver"], [11, 0, 0, "-", "image_saver"], [13, 0, 0, "-", "layer_freeze"], [14, 0, 0, "-", "model_utils"], [15, 0, 0, "-", "outlier_detection"]], "cyto_dl.callbacks.csv_saver": [[10, 1, 1, "", "CSVSaver"], [10, 1, 1, "", "JEPASaver"]], "cyto_dl.callbacks.csv_saver.CSVSaver": [[10, 2, 1, "", "on_predict_epoch_end"], [10, 2, 1, "", "pred_to_csv"], [10, 2, 1, "", "save_feats"]], "cyto_dl.callbacks.csv_saver.JEPASaver": [[10, 2, 1, "", "pred_to_csv"]], "cyto_dl.callbacks.image_saver": [[11, 1, 1, "", "ImageSaver"]], "cyto_dl.callbacks.image_saver.ImageSaver": [[11, 2, 1, "", "on_predict_batch_end"], [11, 2, 1, "", "on_test_batch_end"], [11, 2, 1, "", "on_train_batch_end"], [11, 2, 1, "", "on_validation_batch_end"], [11, 2, 1, "", "save"]], "cyto_dl.callbacks.layer_freeze": [[13, 1, 1, "", "LayerFreeze"]], "cyto_dl.callbacks.layer_freeze.LayerFreeze": [[13, 2, 1, "", "setup"]], "cyto_dl.callbacks.model_utils": [[14, 1, 1, "", "GetEmbeddings"], [14, 4, 1, "", "get_all_embeddings"], [14, 4, 1, "", "save_predictions_classifier"]], "cyto_dl.callbacks.model_utils.GetEmbeddings": [[14, 2, 1, "", "on_test_epoch_end"]], "cyto_dl.callbacks.outlier_detection": [[15, 1, 1, "", "OutlierDetection"]], "cyto_dl.callbacks.outlier_detection.OutlierDetection": [[15, 2, 1, "", "calculate_mahalanobis_hook"], [15, 2, 1, "", "flatten_activations"], [15, 2, 1, "", "on_load_checkpoint"], [15, 2, 1, "", "on_predict_batch_end"], [15, 2, 1, "", "on_predict_epoch_start"], [15, 2, 1, "", "on_save_checkpoint"], [15, 2, 1, "", "on_test_batch_end"], [15, 2, 1, "", "on_test_epoch_start"], [15, 2, 1, "", "on_train_epoch_start"], [15, 2, 1, "", "update_covariance_hook"]], "cyto_dl.dataframe": [[18, 0, 0, "-", "readers"], [19, 0, 0, "-", "transforms"]], "cyto_dl.dataframe.readers": [[18, 4, 1, "", "filter_columns"], [18, 4, 1, "", "read_csv"], [18, 4, 1, "", "read_dataframe"], [18, 4, 1, "", "read_h5ad"], [18, 4, 1, "", "read_parquet"]], "cyto_dl.dataframe.transforms": [[20, 0, 0, "-", "filter"], [21, 0, 0, "-", "group_cols"], [22, 0, 0, "-", "misc"], [23, 0, 0, "-", "split"]], "cyto_dl.dataframe.transforms.filter": [[20, 4, 1, "", "filter_columns"], [20, 4, 1, "", "filter_rows"]], "cyto_dl.dataframe.transforms.group_cols": [[21, 1, 1, "", "GroupCols"]], "cyto_dl.dataframe.transforms.misc": [[22, 4, 1, "", "append_class_weights"], [22, 4, 1, "", "append_labels_to_integers"], [22, 4, 1, "", "append_one_hot"], [22, 4, 1, "", "make_random_df"]], "cyto_dl.dataframe.transforms.split": [[23, 4, 1, "", "sample_n_each"], [23, 4, 1, "", "split_dataframe"]], "cyto_dl.datamodules": [[25, 0, 0, "-", "array"], [26, 0, 0, "-", "data_dict"], [27, 0, 0, "-", "dataframe"], [31, 0, 0, "-", "folder"], [32, 0, 0, "-", "multidim_image"], [33, 0, 0, "-", "smartcache"], [34, 0, 0, "-", "torchvision"]], "cyto_dl.datamodules.array": [[25, 4, 1, "", "make_array_dataloader"]], "cyto_dl.datamodules.data_dict": [[26, 4, 1, "", "make_data_dict_dataloader"]], "cyto_dl.datamodules.dataframe": [[28, 0, 0, "-", "dataframe_datamodule"], [29, 0, 0, "-", "grouped_dataframe_datamodule"], [30, 0, 0, "-", "utils"]], "cyto_dl.datamodules.dataframe.dataframe_datamodule": [[28, 1, 1, "", "DataframeDatamodule"]], "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule": [[28, 2, 1, "", "get_dataloader"], [28, 2, 1, "", "get_dataset"], [28, 2, 1, "", "make_dataloader"], [28, 2, 1, "", "predict_dataloader"], [28, 2, 1, "", "test_dataloader"], [28, 2, 1, "", "train_dataloader"], [28, 2, 1, "", "val_dataloader"]], "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule": [[29, 1, 1, "", "GroupedDataframeDatamodule"]], "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule.GroupedDataframeDatamodule": [[29, 2, 1, "", "make_dataloader"]], "cyto_dl.datamodules.dataframe.utils": [[30, 1, 1, "", "AlternatingBatchSampler"], [30, 1, 1, "", "RemoveNaNKeysd"], [30, 4, 1, "", "get_canonical_split_name"], [30, 4, 1, "", "get_dataset"], [30, 4, 1, "", "make_multiple_dataframe_splits"], [30, 4, 1, "", "make_single_dataframe_splits"], [30, 4, 1, "", "parse_transforms"]], "cyto_dl.datamodules.folder": [[31, 4, 1, "", "make_folder_dataloader"]], "cyto_dl.datamodules.multidim_image": [[32, 1, 1, "", "MultiDimImageDataset"], [32, 4, 1, "", "make_multidim_image_dataloader"]], "cyto_dl.datamodules.multidim_image.MultiDimImageDataset": [[32, 2, 1, "", "create_metatensor"], [32, 2, 1, "", "get_per_file_args"], [32, 2, 1, "", "is_batch"]], "cyto_dl.datamodules.smartcache": [[33, 1, 1, "", "SmartcacheDatamodule"]], "cyto_dl.datamodules.smartcache.SmartcacheDatamodule": [[33, 2, 1, "", "get_per_file_args"], [33, 2, 1, "", "make_dataloader"], [33, 2, 1, "", "predict_dataloader"], [33, 2, 1, "", "prepare_data"], [33, 2, 1, "", "setup"], [33, 2, 1, "", "test_dataloader"], [33, 2, 1, "", "train_dataloader"], [33, 2, 1, "", "val_dataloader"]], "cyto_dl.datamodules.torchvision": [[34, 1, 1, "", "CIFAR10"], [34, 1, 1, "", "CIFAR100"], [34, 1, 1, "", "CLEVRClassification"], [34, 1, 1, "", "Caltech101"], [34, 1, 1, "", "Caltech256"], [34, 1, 1, "", "CelebA"], [34, 1, 1, "", "Cityscapes"], [34, 1, 1, "", "CocoCaptions"], [34, 1, 1, "", "CocoDetection"], [34, 1, 1, "", "Country211"], [34, 1, 1, "", "DTD"], [34, 1, 1, "", "DatasetFolder"], [34, 1, 1, "", "EMNIST"], [34, 1, 1, "", "EuroSAT"], [34, 1, 1, "", "FER2013"], [34, 1, 1, "", "FGVCAircraft"], [34, 1, 1, "", "FakeData"], [34, 1, 1, "", "FashionMNIST"], [34, 1, 1, "", "Flickr30k"], [34, 1, 1, "", "Flickr8k"], [34, 1, 1, "", "Flowers102"], [34, 1, 1, "", "Food101"], [34, 1, 1, "", "GTSRB"], [34, 1, 1, "", "HMDB51"], [34, 1, 1, "", "INaturalist"], [34, 1, 1, "", "ImageFolder"], [34, 1, 1, "", "ImageNet"], [34, 1, 1, "", "KMNIST"], [34, 1, 1, "", "Kinetics"], [34, 1, 1, "", "Kitti"], [34, 1, 1, "", "LFWPairs"], [34, 1, 1, "", "LFWPeople"], [34, 1, 1, "", "LSUN"], [34, 1, 1, "", "LSUNClass"], [34, 1, 1, "", "MNIST"], [34, 1, 1, "", "MovingMNIST"], [34, 1, 1, "", "Omniglot"], [34, 1, 1, "", "OxfordIIITPet"], [34, 1, 1, "", "PCAM"], [34, 1, 1, "", "PhotoTour"], [34, 1, 1, "", "Places365"], [34, 1, 1, "", "QMNIST"], [34, 1, 1, "", "RenderedSST2"], [34, 1, 1, "", "SBDataset"], [34, 1, 1, "", "SBU"], [34, 1, 1, "", "SEMEION"], [34, 1, 1, "", "STL10"], [34, 1, 1, "", "SUN397"], [34, 1, 1, "", "SVHN"], [34, 1, 1, "", "StanfordCars"], [34, 1, 1, "", "UCF101"], [34, 1, 1, "", "USPS"], [34, 1, 1, "", "VOCDetection"], [34, 1, 1, "", "VOCSegmentation"], [34, 1, 1, "", "VisionDataset"], [34, 1, 1, "", "WIDERFace"]], "cyto_dl.eval": [[35, 4, 1, "", "main"]], "cyto_dl.image": [[37, 0, 0, "-", "io"], [44, 0, 0, "-", "transforms"]], "cyto_dl.image.io": [[38, 0, 0, "-", "aicsimage_loader"], [39, 0, 0, "-", "monai_bio_reader"], [40, 0, 0, "-", "numpy_reader"], [41, 0, 0, "-", "ome_zarr_reader"], [42, 0, 0, "-", "polygon_loader"], [43, 0, 0, "-", "skimage_reader"]], "cyto_dl.image.io.aicsimage_loader": [[38, 1, 1, "", "AICSImageLoaderd"]], "cyto_dl.image.io.aicsimage_loader.AICSImageLoaderd": [[38, 2, 1, "", "split_args"]], "cyto_dl.image.io.monai_bio_reader": [[39, 1, 1, "", "MonaiBioReader"]], "cyto_dl.image.io.monai_bio_reader.MonaiBioReader": [[39, 2, 1, "", "get_data"], [39, 2, 1, "", "read"], [39, 2, 1, "", "verify_suffix"]], "cyto_dl.image.io.numpy_reader": [[40, 1, 1, "", "ReadNumpyFile"]], "cyto_dl.image.io.ome_zarr_reader": [[41, 1, 1, "", "OmeZarrReader"]], "cyto_dl.image.io.ome_zarr_reader.OmeZarrReader": [[41, 2, 1, "", "get_data"], [41, 2, 1, "", "read"], [41, 2, 1, "", "verify_suffix"]], "cyto_dl.image.io.polygon_loader": [[42, 1, 1, "", "PolygonLoaderd"]], "cyto_dl.image.io.skimage_reader": [[43, 1, 1, "", "SkimageReader"]], "cyto_dl.image.io.skimage_reader.SkimageReader": [[43, 2, 1, "", "get_data"], [43, 2, 1, "", "read"], [43, 2, 1, "", "verify_suffix"]], "cyto_dl.image.transforms": [[45, 0, 0, "-", "add_meta"], [46, 0, 0, "-", "bright_sampler"], [47, 0, 0, "-", "clip"], [48, 0, 0, "-", "contrastadjust"], [49, 0, 0, "-", "generate_jepa_masks"], [50, 0, 0, "-", "merge"], [51, 0, 0, "-", "multiscale_cropper"], [52, 0, 0, "-", "pad"], [53, 0, 0, "-", "project"], [55, 0, 0, "-", "save"], [57, 0, 0, "-", "track_transforms"]], "cyto_dl.image.transforms.add_meta": [[45, 1, 1, "", "AddMeta"]], "cyto_dl.image.transforms.bright_sampler": [[46, 1, 1, "", "BrightSampler"]], "cyto_dl.image.transforms.clip": [[47, 1, 1, "", "Clip"], [47, 1, 1, "", "Clipd"]], "cyto_dl.image.transforms.contrastadjust": [[48, 1, 1, "", "ContrastAdjust"], [48, 1, 1, "", "ContrastAdjustd"]], "cyto_dl.image.transforms.generate_jepa_masks": [[49, 1, 1, "", "JEPAMaskGenerator"]], "cyto_dl.image.transforms.generate_jepa_masks.JEPAMaskGenerator": [[49, 2, 1, "", "remove_excess_pixels"]], "cyto_dl.image.transforms.merge": [[50, 1, 1, "", "Merged"]], "cyto_dl.image.transforms.multiscale_cropper": [[51, 1, 1, "", "RandomMultiScaleCropd"]], "cyto_dl.image.transforms.multiscale_cropper.RandomMultiScaleCropd": [[51, 2, 1, "", "generate_slices"]], "cyto_dl.image.transforms.pad": [[52, 1, 1, "", "PadZd"]], "cyto_dl.image.transforms.project": [[53, 1, 1, "", "MaxProjectd"]], "cyto_dl.image.transforms.save": [[55, 1, 1, "", "Save"], [55, 1, 1, "", "Saved"]], "cyto_dl.image.transforms.track_transforms": [[57, 1, 1, "", "CropResize"], [57, 1, 1, "", "GenerateTrackLabels"], [57, 1, 1, "", "PerChannel"], [57, 1, 1, "", "SplitTrackd"]], "cyto_dl.loggers": [[59, 0, 0, "-", "mlflow"]], "cyto_dl.loggers.mlflow": [[59, 1, 1, "", "MLFlowLogger"]], "cyto_dl.loggers.mlflow.MLFlowLogger": [[59, 2, 1, "", "after_save_checkpoint"], [59, 2, 1, "", "log_hyperparams"], [59, 2, 1, "", "log_metrics"]], "cyto_dl.models": [[61, 0, 0, "-", "base_model"], [62, 0, 0, "-", "basic_model"], [63, 0, 0, "-", "classification"], [66, 0, 0, "-", "contrastive"], [72, 0, 0, "-", "im2im"], [82, 0, 0, "-", "jepa"], [86, 0, 0, "-", "utils"]], "cyto_dl.models.base_model": [[61, 1, 1, "", "BaseModel"], [61, 1, 1, "", "BaseModelMeta"]], "cyto_dl.models.base_model.BaseModel": [[61, 2, 1, "", "compute_metrics"], [61, 2, 1, "", "configure_optimizers"], [61, 2, 1, "", "forward"], [61, 2, 1, "", "model_step"], [61, 2, 1, "", "on_train_start"], [61, 2, 1, "", "parse_batch"], [61, 2, 1, "", "predict_step"], [61, 2, 1, "", "test_step"], [61, 2, 1, "", "training_step"], [61, 2, 1, "", "validation_step"]], "cyto_dl.models.basic_model": [[62, 1, 1, "", "BasicModel"]], "cyto_dl.models.basic_model.BasicModel": [[62, 2, 1, "", "forward"], [62, 2, 1, "", "model_step"]], "cyto_dl.models.classification": [[64, 0, 0, "-", "classification"], [65, 0, 0, "-", "timepoint_classification"]], "cyto_dl.models.classification.classification": [[64, 1, 1, "", "Classification"]], "cyto_dl.models.classification.classification.Classification": [[64, 2, 1, "", "forward"], [64, 2, 1, "", "model_step"], [64, 2, 1, "", "predict_step"], [64, 2, 1, "", "save_images"], [64, 2, 1, "", "should_save_image"]], "cyto_dl.models.classification.timepoint_classification": [[65, 1, 1, "", "TimepointClassification"]], "cyto_dl.models.classification.timepoint_classification.TimepointClassification": [[65, 2, 1, "", "predict_step"]], "cyto_dl.models.contrastive": [[67, 0, 0, "-", "contrastive"]], "cyto_dl.models.contrastive.contrastive": [[67, 1, 1, "", "Contrastive"]], "cyto_dl.models.contrastive.contrastive.Contrastive": [[67, 2, 1, "", "forward"], [67, 2, 1, "", "model_step"], [67, 2, 1, "", "plot_classes"], [67, 2, 1, "", "plot_neighbors"], [67, 2, 1, "", "predict_step"]], "cyto_dl.models.im2im": [[73, 0, 0, "-", "gan"], [74, 0, 0, "-", "multi_task"], [75, 0, 0, "-", "utils"]], "cyto_dl.models.im2im.gan": [[73, 1, 1, "", "GAN"]], "cyto_dl.models.im2im.gan.GAN": [[73, 2, 1, "", "configure_optimizers"], [73, 2, 1, "", "model_step"], [73, 2, 1, "", "predict_step"]], "cyto_dl.models.im2im.multi_task": [[74, 1, 1, "", "MultiTaskIm2Im"]], "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im": [[74, 2, 1, "", "configure_optimizers"], [74, 2, 1, "", "forward"], [74, 2, 1, "", "get_n_postprocess_image"], [74, 2, 1, "", "get_per_head"], [74, 2, 1, "", "model_step"], [74, 2, 1, "", "predict_step"], [74, 2, 1, "", "run_forward"], [74, 2, 1, "", "test_step"], [74, 2, 1, "", "training_step"], [74, 2, 1, "", "validation_step"]], "cyto_dl.models.im2im.utils": [[76, 0, 0, "-", "instance_seg"], [77, 0, 0, "-", "noise_annealer"], [78, 0, 0, "-", "postprocessing"]], "cyto_dl.models.im2im.utils.instance_seg": [[76, 1, 1, "", "InstanceSegCluster"], [76, 1, 1, "", "InstanceSegLoss"], [76, 1, 1, "", "InstanceSegPreprocessd"], [76, 1, 1, "", "InstanceSegRandFlipd"], [76, 4, 1, "", "pad_slice"]], "cyto_dl.models.im2im.utils.instance_seg.InstanceSegCluster": [[76, 2, 1, "", "cluster_object"], [76, 2, 1, "", "kd_clustering"], [76, 2, 1, "", "remove_small_skeletons"]], "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd": [[76, 2, 1, "", "embed_from_skel"], [76, 2, 1, "", "keep_largest_cc"], [76, 2, 1, "", "label_2d"], [76, 2, 1, "", "shrink"], [76, 2, 1, "", "skeleton_tall"], [76, 2, 1, "", "smooth_embedding"], [76, 2, 1, "", "topology_preserving_thinning"]], "cyto_dl.models.im2im.utils.noise_annealer": [[77, 1, 1, "", "NoiseAnnealer"]], "cyto_dl.models.im2im.utils.noise_annealer.NoiseAnnealer": [[77, 2, 1, "", "update_noise"]], "cyto_dl.models.im2im.utils.postprocessing": [[79, 0, 0, "-", "act_thresh_label"], [80, 0, 0, "-", "auto_thresh"], [81, 0, 0, "-", "dict_to_im"]], "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label": [[79, 1, 1, "", "ActThreshLabel"]], "cyto_dl.models.im2im.utils.postprocessing.auto_thresh": [[80, 1, 1, "", "AutoThreshold"]], "cyto_dl.models.im2im.utils.postprocessing.dict_to_im": [[81, 1, 1, "", "DictToIm"], [81, 4, 1, "", "detach"]], "cyto_dl.models.jepa": [[83, 0, 0, "-", "ijepa"], [84, 0, 0, "-", "iwm"], [85, 0, 0, "-", "jepa_base"]], "cyto_dl.models.jepa.ijepa": [[83, 1, 1, "", "IJEPA"]], "cyto_dl.models.jepa.ijepa.IJEPA": [[83, 2, 1, "", "model_step"]], "cyto_dl.models.jepa.iwm": [[84, 1, 1, "", "IWM"]], "cyto_dl.models.jepa.iwm.IWM": [[84, 2, 1, "", "extract_embeddings"], [84, 2, 1, "", "get_predict_masks"], [84, 2, 1, "", "model_step"], [84, 2, 1, "", "predict_step"], [84, 2, 1, "", "test_step"]], "cyto_dl.models.jepa.jepa_base": [[85, 1, 1, "", "JEPABase"]], "cyto_dl.models.jepa.jepa_base.JEPABase": [[85, 2, 1, "", "configure_optimizers"], [85, 2, 1, "", "forward"], [85, 2, 1, "", "get_context_embeddings"], [85, 2, 1, "", "get_mask"], [85, 2, 1, "", "get_target_embeddings"], [85, 2, 1, "", "model_step"], [85, 2, 1, "", "predict_step"], [85, 2, 1, "", "remove_first_dim"], [85, 2, 1, "", "update_teacher"]], "cyto_dl.models.utils": [[87, 0, 0, "-", "mlflow"], [88, 0, 0, "-", "utils"]], "cyto_dl.models.utils.mlflow": [[87, 4, 1, "", "get_config"], [87, 4, 1, "", "load_model_from_checkpoint"]], "cyto_dl.models.utils.utils": [[88, 4, 1, "", "find_indices"]], "cyto_dl.models.vae": [[90, 0, 0, "-", "base_vae"], [99, 0, 0, "-", "priors"]], "cyto_dl.models.vae.base_vae": [[90, 1, 1, "", "BaseVAE"]], "cyto_dl.models.vae.base_vae.BaseVAE": [[90, 2, 1, "", "calculate_elbo"], [90, 2, 1, "", "calculate_rcl"], [90, 2, 1, "", "calculate_rcl_dict"], [90, 2, 1, "", "decode"], [90, 2, 1, "", "encode"], [90, 2, 1, "", "forward"], [90, 2, 1, "", "model_step"], [90, 2, 1, "", "sample_z"]], "cyto_dl.models.vae.priors": [[100, 0, 0, "-", "abstract_prior"], [101, 0, 0, "-", "gaussian"], [102, 0, 0, "-", "identity_prior"], [103, 0, 0, "-", "joint_prior"]], "cyto_dl.models.vae.priors.abstract_prior": [[100, 1, 1, "", "Prior"]], "cyto_dl.models.vae.priors.abstract_prior.Prior": [[100, 2, 1, "", "forward"], [100, 2, 1, "", "kl_divergence"], [100, 5, 1, "", "param_size"], [100, 2, 1, "", "sample"]], "cyto_dl.models.vae.priors.gaussian": [[101, 1, 1, "", "DiagonalGaussianPrior"], [101, 1, 1, "", "IsotropicGaussianPrior"], [101, 4, 1, "", "compute_tc_penalty"]], "cyto_dl.models.vae.priors.gaussian.DiagonalGaussianPrior": [[101, 2, 1, "", "forward"], [101, 2, 1, "", "kl_divergence"], [101, 5, 1, "", "param_size"]], "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior": [[101, 2, 1, "", "forward"], [101, 2, 1, "", "kl_divergence"], [101, 5, 1, "", "param_size"], [101, 2, 1, "", "sample"]], "cyto_dl.models.vae.priors.identity_prior": [[102, 1, 1, "", "IdentityPrior"]], "cyto_dl.models.vae.priors.identity_prior.IdentityPrior": [[102, 2, 1, "", "forward"], [102, 5, 1, "", "param_size"]], "cyto_dl.models.vae.priors.joint_prior": [[103, 1, 1, "", "JointPrior"]], "cyto_dl.models.vae.priors.joint_prior.JointPrior": [[103, 2, 1, "", "forward"], [103, 2, 1, "", "kl_divergence"], [103, 5, 1, "", "param_size"], [103, 2, 1, "", "sample"]], "cyto_dl.nn": [[106, 0, 0, "-", "discriminators"], [109, 0, 0, "-", "head"], [117, 0, 0, "-", "hr_skip"], [118, 0, 0, "-", "losses"], [132, 0, 0, "-", "mlp"], [138, 0, 0, "-", "res_unit"], [139, 0, 0, "-", "spatial_transformer"], [140, 0, 0, "-", "track_sequence_predictor"], [141, 0, 0, "-", "vits"]], "cyto_dl.nn.discriminators": [[107, 0, 0, "-", "multi_scale_discriminator"], [108, 0, 0, "-", "n_layer_discriminator"]], "cyto_dl.nn.discriminators.multi_scale_discriminator": [[107, 1, 1, "", "MultiScaleDiscriminator"]], "cyto_dl.nn.discriminators.multi_scale_discriminator.MultiScaleDiscriminator": [[107, 2, 1, "", "forward"]], "cyto_dl.nn.discriminators.n_layer_discriminator": [[108, 1, 1, "", "NLayerDiscriminator"]], "cyto_dl.nn.discriminators.n_layer_discriminator.NLayerDiscriminator": [[108, 2, 1, "", "forward"]], "cyto_dl.nn.head": [[110, 0, 0, "-", "base_head"], [111, 0, 0, "-", "gan_head"], [112, 0, 0, "-", "gan_head_superres"], [113, 0, 0, "-", "mae_head"], [114, 0, 0, "-", "mask_head"], [115, 0, 0, "-", "res_blocks_head"], [116, 0, 0, "-", "vic_reg"]], "cyto_dl.nn.head.base_head": [[110, 1, 1, "", "BaseHead"]], "cyto_dl.nn.head.base_head.BaseHead": [[110, 2, 1, "", "forward"], [110, 2, 1, "", "generate_io_map"], [110, 2, 1, "", "run_head"], [110, 2, 1, "", "update_params"]], "cyto_dl.nn.head.gan_head": [[111, 1, 1, "", "GANHead"]], "cyto_dl.nn.head.gan_head.GANHead": [[111, 2, 1, "", "forward"], [111, 2, 1, "", "run_head"]], "cyto_dl.nn.head.gan_head_superres": [[112, 1, 1, "", "GANHead_resize"]], "cyto_dl.nn.head.gan_head_superres.GANHead_resize": [[112, 2, 1, "", "forward"]], "cyto_dl.nn.head.mae_head": [[113, 1, 1, "", "MAEHead"]], "cyto_dl.nn.head.mae_head.MAEHead": [[113, 2, 1, "", "run_head"]], "cyto_dl.nn.head.mask_head": [[114, 1, 1, "", "MaskHead"]], "cyto_dl.nn.head.mask_head.MaskHead": [[114, 2, 1, "", "run_head"]], "cyto_dl.nn.head.res_blocks_head": [[115, 1, 1, "", "ResBlocksHead"]], "cyto_dl.nn.head.res_blocks_head.ResBlocksHead": [[115, 2, 1, "", "forward"]], "cyto_dl.nn.head.vic_reg": [[116, 1, 1, "", "VICRegHead"]], "cyto_dl.nn.head.vic_reg.VICRegHead": [[116, 2, 1, "", "run_head"]], "cyto_dl.nn.hr_skip": [[117, 1, 1, "", "HRSkip"]], "cyto_dl.nn.hr_skip.HRSkip": [[117, 2, 1, "", "forward"]], "cyto_dl.nn.losses": [[119, 0, 0, "-", "adversarial_loss"], [120, 0, 0, "-", "chamfer_loss"], [121, 0, 0, "-", "continuous_bernoulli"], [122, 0, 0, "-", "cosine_loss"], [123, 0, 0, "-", "gan_loss"], [124, 0, 0, "-", "gaussian_nll_loss"], [126, 0, 0, "-", "loss_wrapper"], [128, 0, 0, "-", "threshold_loss"], [129, 0, 0, "-", "vic_reg"], [130, 0, 0, "-", "weibull"], [131, 0, 0, "-", "weighted_mse_loss"]], "cyto_dl.nn.losses.adversarial_loss": [[119, 1, 1, "", "AdversarialLoss"]], "cyto_dl.nn.losses.adversarial_loss.AdversarialLoss": [[119, 2, 1, "", "forward"]], "cyto_dl.nn.losses.chamfer_loss": [[120, 1, 1, "", "ChamferLoss"]], "cyto_dl.nn.losses.chamfer_loss.ChamferLoss": [[120, 2, 1, "", "batch_pairwise_dist"], [120, 2, 1, "", "batch_pairwise_dist2"], [120, 2, 1, "", "forward"]], "cyto_dl.nn.losses.continuous_bernoulli": [[121, 1, 1, "", "CBLogLoss"]], "cyto_dl.nn.losses.continuous_bernoulli.CBLogLoss": [[121, 2, 1, "", "forward"]], "cyto_dl.nn.losses.cosine_loss": [[122, 1, 1, "", "CosineLoss"]], "cyto_dl.nn.losses.cosine_loss.CosineLoss": [[122, 2, 1, "", "forward"]], "cyto_dl.nn.losses.gan_loss": [[123, 1, 1, "", "GANLoss"], [123, 1, 1, "", "Pix2PixHD"]], "cyto_dl.nn.losses.gan_loss.GANLoss": [[123, 2, 1, "", "get_target_tensor"]], "cyto_dl.nn.losses.gan_loss.Pix2PixHD": [[123, 2, 1, "", "get_feature_matching_loss"], [123, 2, 1, "", "get_gan_loss"]], "cyto_dl.nn.losses.gaussian_nll_loss": [[124, 1, 1, "", "GaussianNLLLoss"]], "cyto_dl.nn.losses.gaussian_nll_loss.GaussianNLLLoss": [[124, 2, 1, "", "forward"]], "cyto_dl.nn.losses.loss_wrapper": [[126, 1, 1, "", "CMAP_loss"], [126, 1, 1, "", "LossWrapper"]], "cyto_dl.nn.losses.threshold_loss": [[128, 1, 1, "", "ThresholdLoss"]], "cyto_dl.nn.losses.vic_reg": [[129, 1, 1, "", "VICRegLoss"], [129, 4, 1, "", "off_diagonal"]], "cyto_dl.nn.losses.vic_reg.VICRegLoss": [[129, 2, 1, "", "forward"]], "cyto_dl.nn.losses.weibull": [[130, 1, 1, "", "WeibullLogLoss"], [130, 4, 1, "", "weibull_log_probs"]], "cyto_dl.nn.losses.weibull.WeibullLogLoss": [[130, 2, 1, "", "forward"]], "cyto_dl.nn.losses.weighted_mse_loss": [[131, 1, 1, "", "WeightedMSELoss"]], "cyto_dl.nn.losses.weighted_mse_loss.WeightedMSELoss": [[131, 2, 1, "", "forward"]], "cyto_dl.nn.mlp": [[132, 1, 1, "", "MLP"]], "cyto_dl.nn.mlp.MLP": [[132, 2, 1, "", "forward"]], "cyto_dl.nn.res_unit": [[138, 1, 1, "", "ResidualUnit"]], "cyto_dl.nn.res_unit.ResidualUnit": [[138, 2, 1, "", "forward"]], "cyto_dl.nn.spatial_transformer": [[139, 1, 1, "", "ConvPoolReLU"], [139, 1, 1, "", "STN"]], "cyto_dl.nn.spatial_transformer.ConvPoolReLU": [[139, 2, 1, "", "forward"]], "cyto_dl.nn.spatial_transformer.STN": [[139, 2, 1, "", "forward"], [139, 2, 1, "", "toggle"]], "cyto_dl.nn.track_sequence_predictor": [[140, 1, 1, "", "TrackClassifier"], [140, 4, 1, "", "positionalencoding1d"]], "cyto_dl.nn.track_sequence_predictor.TrackClassifier": [[140, 2, 1, "", "forward"]], "cyto_dl.nn.vits": [[142, 0, 0, "-", "blocks"], [150, 0, 0, "-", "decoder"], [151, 0, 0, "-", "encoder"], [152, 0, 0, "-", "mae"], [153, 0, 0, "-", "predictor"], [154, 0, 0, "-", "seg"], [155, 0, 0, "-", "utils"]], "cyto_dl.nn.vits.blocks": [[143, 0, 0, "-", "cross_attention"], [144, 0, 0, "-", "intermediate_weigher"], [145, 0, 0, "-", "masked_unit_attention"], [146, 0, 0, "-", "patchify"]], "cyto_dl.nn.vits.blocks.cross_attention": [[143, 1, 1, "", "CrossAttention"], [143, 1, 1, "", "CrossAttentionBlock"], [143, 1, 1, "", "CrossSelfBlock"], [143, 1, 1, "", "Mlp"]], "cyto_dl.nn.vits.blocks.cross_attention.CrossAttention": [[143, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.cross_attention.CrossAttentionBlock": [[143, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.cross_attention.CrossSelfBlock": [[143, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.cross_attention.Mlp": [[143, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.intermediate_weigher": [[144, 1, 1, "", "IntermediateWeigher"]], "cyto_dl.nn.vits.blocks.intermediate_weigher.IntermediateWeigher": [[144, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.masked_unit_attention": [[145, 1, 1, "", "HieraBlock"], [145, 1, 1, "", "MaskUnitAttention"]], "cyto_dl.nn.vits.blocks.masked_unit_attention.HieraBlock": [[145, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.masked_unit_attention.MaskUnitAttention": [[145, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.patchify": [[147, 0, 0, "-", "patchify"], [148, 0, 0, "-", "patchify_base"], [149, 0, 0, "-", "patchify_hiera"]], "cyto_dl.nn.vits.blocks.patchify.patchify": [[147, 1, 1, "", "Patchify"]], "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify": [[147, 2, 1, "", "create_img2token"], [147, 2, 1, "", "extract_visible_tokens"], [147, 2, 1, "", "get_mask_args"], [147, 5, 1, "", "img2token"]], "cyto_dl.nn.vits.blocks.patchify.patchify_base": [[148, 1, 1, "", "PatchifyBase"]], "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase": [[148, 2, 1, "", "create_conv"], [148, 2, 1, "", "create_patch2img"], [148, 2, 1, "", "extract_visible_tokens"], [148, 2, 1, "", "forward"], [148, 2, 1, "", "get_mask"], [148, 2, 1, "", "get_mask_args"], [148, 5, 1, "", "img2token"]], "cyto_dl.nn.vits.blocks.patchify.patchify_hiera": [[149, 1, 1, "", "PatchifyHiera"], [149, 4, 1, "", "take_indexes_mask"]], "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera": [[149, 2, 1, "", "create_img2token"], [149, 2, 1, "", "extract_visible_tokens"], [149, 2, 1, "", "get_mask_args"], [149, 5, 1, "", "img2token"]], "cyto_dl.nn.vits.decoder": [[150, 1, 1, "", "CrossMAE_Decoder"], [150, 1, 1, "", "MAE_Decoder"]], "cyto_dl.nn.vits.decoder.CrossMAE_Decoder": [[150, 2, 1, "", "forward"]], "cyto_dl.nn.vits.decoder.MAE_Decoder": [[150, 2, 1, "", "add_mask_tokens"], [150, 2, 1, "", "adjust_indices_for_cls"], [150, 2, 1, "", "forward"], [150, 2, 1, "", "init_weight"]], "cyto_dl.nn.vits.encoder": [[151, 1, 1, "", "HieraEncoder"], [151, 1, 1, "", "JEPAEncoder"], [151, 1, 1, "", "MAE_Encoder"], [151, 1, 1, "", "SpatialMerger"]], "cyto_dl.nn.vits.encoder.HieraEncoder": [[151, 2, 1, "", "forward"]], "cyto_dl.nn.vits.encoder.JEPAEncoder": [[151, 2, 1, "", "forward"]], "cyto_dl.nn.vits.encoder.MAE_Encoder": [[151, 2, 1, "", "forward"], [151, 2, 1, "", "init_weight"]], "cyto_dl.nn.vits.encoder.SpatialMerger": [[151, 2, 1, "", "forward"]], "cyto_dl.nn.vits.mae": [[152, 1, 1, "", "HieraMAE"], [152, 1, 1, "", "MAE"], [152, 1, 1, "", "MAE_Base"]], "cyto_dl.nn.vits.mae.HieraMAE": [[152, 5, 1, "", "decoder"], [152, 5, 1, "", "encoder"]], "cyto_dl.nn.vits.mae.MAE": [[152, 5, 1, "", "decoder"], [152, 5, 1, "", "encoder"]], "cyto_dl.nn.vits.mae.MAE_Base": [[152, 5, 1, "", "decoder"], [152, 5, 1, "", "encoder"], [152, 2, 1, "", "forward"], [152, 2, 1, "", "init_decoder"], [152, 2, 1, "", "init_encoder"]], "cyto_dl.nn.vits.predictor": [[153, 1, 1, "", "IWMPredictor"], [153, 1, 1, "", "JEPAPredictor"]], "cyto_dl.nn.vits.predictor.IWMPredictor": [[153, 2, 1, "", "forward"]], "cyto_dl.nn.vits.predictor.JEPAPredictor": [[153, 2, 1, "", "forward"], [153, 2, 1, "", "init_weight"], [153, 2, 1, "", "predict_target_features"]], "cyto_dl.nn.vits.seg": [[154, 1, 1, "", "EncodedSkip"], [154, 1, 1, "", "Seg_ViT"], [154, 1, 1, "", "SuperresDecoder"]], "cyto_dl.nn.vits.seg.EncodedSkip": [[154, 2, 1, "", "forward"]], "cyto_dl.nn.vits.seg.Seg_ViT": [[154, 2, 1, "", "forward"]], "cyto_dl.nn.vits.seg.SuperresDecoder": [[154, 2, 1, "", "forward"]], "cyto_dl.nn.vits.utils": [[155, 4, 1, "", "get_positional_embedding"], [155, 4, 1, "", "match_tuple_dimensions"], [155, 4, 1, "", "random_indexes"], [155, 4, 1, "", "take_indexes"]], "cyto_dl.train": [[159, 4, 1, "", "main"]], "cyto_dl.utils": [[161, 0, 0, "-", "array"], [162, 0, 0, "-", "checkpoint"], [163, 0, 0, "-", "config"], [164, 0, 0, "-", "download_test_data"], [165, 0, 0, "-", "dummy_dataset"], [166, 0, 0, "-", "pylogger"], [167, 0, 0, "-", "rich_utils"], [168, 0, 0, "-", "rotation"], [172, 0, 0, "-", "template_utils"]], "cyto_dl.utils.array": [[161, 4, 1, "", "create_dataloader"], [161, 4, 1, "", "extract_array_predictions"]], "cyto_dl.utils.checkpoint": [[162, 4, 1, "", "load_checkpoint"]], "cyto_dl.utils.config": [[163, 4, 1, "", "is_config"], [163, 4, 1, "", "kv_to_dict"], [163, 4, 1, "", "remove_aux_key"]], "cyto_dl.utils.download_test_data": [[164, 4, 1, "", "delete_test_data"], [164, 4, 1, "", "download_test_data"], [164, 4, 1, "", "parse_s3_path"], [164, 4, 1, "", "setup_paths"]], "cyto_dl.utils.dummy_dataset": [[165, 1, 1, "", "DummyDatamodule"], [165, 1, 1, "", "DummyDataset"]], "cyto_dl.utils.dummy_dataset.DummyDatamodule": [[165, 2, 1, "", "get_dataloader"], [165, 2, 1, "", "predict_dataloader"], [165, 2, 1, "", "test_dataloader"], [165, 2, 1, "", "train_dataloader"], [165, 2, 1, "", "val_dataloader"]], "cyto_dl.utils.dummy_dataset.DummyDataset": [[165, 2, 1, "", "generate_img"]], "cyto_dl.utils.pylogger": [[166, 4, 1, "", "get_pylogger"]], "cyto_dl.utils.rich_utils": [[167, 4, 1, "", "enforce_tags"], [167, 4, 1, "", "print_config_tree"]], "cyto_dl.utils.rotation": [[168, 1, 1, "", "RotationModule"], [168, 4, 1, "", "get_rotation_matrix_so2"], [168, 4, 1, "", "get_rotation_matrix_so3"]], "cyto_dl.utils.rotation.RotationModule": [[168, 2, 1, "", "compute_rotation_matrix"]], "cyto_dl.utils.template_utils": [[172, 4, 1, "", "close_loggers"], [172, 4, 1, "", "extras"], [172, 4, 1, "", "get_metric_value"], [172, 4, 1, "", "instantiate_callbacks"], [172, 4, 1, "", "instantiate_loggers"], [172, 4, 1, "", "log_hyperparameters"], [172, 4, 1, "", "save_file"], [172, 4, 1, "", "task_wrapper"]]}, "objtypes": {"0": "py:module", "1": "py:class", "2": "py:method", "3": "py:attribute", "4": "py:function", "5": "py:property"}, "objnames": {"0": ["py", "module", "Python module"], "1": ["py", "class", "Python class"], "2": ["py", "method", "Python method"], "3": ["py", "attribute", "Python attribute"], "4": ["py", "function", "Python function"], "5": ["py", "property", "Python property"]}, "titleterms": {"us": 0, "api": [0, 2, 3, 4, 5, 6, 7], "load": 0, "config": [0, 163, 177], "train": [0, 159, 177], "predict": [0, 177], "util": [0, 30, 75, 76, 77, 78, 79, 80, 81, 86, 87, 88, 155, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172], "cyto_dl": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 175], "packag": [1, 2, 3, 8, 17, 19, 24, 27, 36, 37, 44, 58, 60, 63, 66, 68, 72, 75, 78, 82, 86, 89, 95, 99, 105, 106, 109, 118, 133, 141, 142, 146, 156, 157, 160, 169], "subpackag": [1, 2, 17, 24, 36, 60, 72, 75, 89, 105, 141, 142, 156, 160], "submodul": [1, 2, 3, 8, 17, 19, 24, 27, 37, 44, 58, 60, 63, 66, 68, 72, 75, 78, 82, 86, 89, 95, 99, 105, 106, 109, 118, 133, 141, 142, 146, 157, 160, 169], "cyto_dl_model": [3, 4, 5], "cyto_dl_base_model": 4, "modul": [4, 5, 6, 7, 9, 10, 11, 12, 13, 14, 15, 16, 18, 20, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 34, 35, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 59, 61, 62, 64, 65, 67, 69, 70, 71, 73, 74, 76, 77, 79, 80, 81, 83, 84, 85, 87, 88, 90, 91, 92, 93, 94, 96, 97, 98, 100, 101, 102, 103, 104, 107, 108, 110, 111, 112, 113, 114, 115, 116, 117, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 137, 138, 139, 140, 143, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 158, 159, 161, 162, 163, 164, 165, 166, 167, 168, 170, 171, 172], "segmentation_plugin_model": 5, "data": [6, 173, 176], "model": [7, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 173, 176], "callback": [8, 9, 10, 11, 12, 13, 14, 15], "callback_util": 9, "csv_saver": 10, "image_sav": 11, "latent_walk": 12, "layer_freez": 13, "model_util": 14, "outlier_detect": 15, "compil": 16, "datafram": [17, 18, 19, 20, 21, 22, 23, 27, 28, 29, 30], "reader": 18, "transform": [19, 20, 21, 22, 23, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57], "filter": 20, "group_col": 21, "misc": 22, "split": 23, "datamodul": [24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34], "arrai": [25, 161], "data_dict": 26, "dataframe_datamodul": 28, "grouped_dataframe_datamodul": 29, "folder": 31, "multidim_imag": 32, "smartcach": 33, "torchvis": 34, "eval": 35, "imag": [36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57], "io": [37, 38, 39, 40, 41, 42, 43, 157, 158], "aicsimage_load": 38, "monai_bio_read": 39, "numpy_read": 40, "ome_zarr_read": 41, "polygon_load": 42, "skimage_read": 43, "add_meta": 45, "bright_sampl": 46, "clip": 47, "contrastadjust": 48, "generate_jepa_mask": 49, "merg": 50, "multiscale_cropp": 51, "pad": 52, "project": 53, "rotation_mask_transform": 54, "save": 55, "so2_random_rot": 56, "track_transform": 57, "logger": [58, 59], "mlflow": [59, 87], "base_model": 61, "basic_model": 62, "classif": [63, 64, 65], "timepoint_classif": 65, "contrast": [66, 67], "handler": [68, 69, 70, 71], "base_handl": 69, "image_handl": 70, "load_image_patch": 71, "im2im": [72, 73, 74, 75, 76, 77, 78, 79, 80, 81], "gan": 73, "multi_task": 74, "instance_seg": 76, "noise_anneal": 77, "postprocess": [78, 79, 80, 81], "act_thresh_label": 79, "auto_thresh": 80, "dict_to_im": 81, "jepa": [82, 83, 84, 85], "ijepa": 83, "iwm": 84, "jepa_bas": 85, "vae": [89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104], "base_va": 90, "image_canon_va": 91, "image_encod": 92, "image_va": 93, "latent_loss_va": 94, "o2_spharm_va": [95, 96, 97], "o2_spharm_encod": 96, "point_cloud_va": 98, "prior": [99, 100, 101, 102, 103], "abstract_prior": 100, "gaussian": 101, "identity_prior": 102, "joint_prior": 103, "tabular_va": 104, "nn": [105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155], "discrimin": [106, 107, 108], "multi_scale_discrimin": 107, "n_layer_discrimin": 108, "head": [109, 110, 111, 112, 113, 114, 115, 116], "base_head": 110, "gan_head": 111, "gan_head_superr": 112, "mae_head": 113, "mask_head": 114, "res_blocks_head": 115, "vic_reg": [116, 129], "hr_skip": 117, "loss": [118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131], "adversarial_loss": 119, "chamfer_loss": 120, "continuous_bernoulli": 121, "cosine_loss": 122, "gan_loss": 123, "gaussian_nll_loss": 124, "geomloss": 125, "loss_wrapp": 126, "spharm_loss": 127, "threshold_loss": 128, "weibul": 130, "weighted_mse_loss": 131, "mlp": 132, "point_cloud": [133, 134, 135, 136, 137, 156, 157, 158], "dgcnn": 134, "folding_net": 135, "graph_funct": 136, "vnn": 137, "res_unit": 138, "spatial_transform": 139, "track_sequence_predictor": 140, "vit": [141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155], "block": [142, 143, 144, 145, 146, 147, 148, 149], "cross_attent": 143, "intermediate_weigh": 144, "masked_unit_attent": 145, "patchifi": [146, 147, 148, 149], "patchify_bas": 148, "patchify_hiera": 149, "decod": 150, "encod": 151, "mae": 152, "predictor": 153, "seg": 154, "read_pcloud": 158, "checkpoint": 162, "download_test_data": 164, "dummy_dataset": 165, "pylogg": 166, "rich_util": 167, "rotat": [168, 171], "spharm": [169, 170, 171], "reconstruct": 170, "template_util": 172, "overview": [173, 176], "experi": [173, 176], "instal": 174, "modifi": 177, "exampl": 177, "test": 177}, "envversion": {"sphinx.domains.c": 2, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 8, "sphinx.domains.index": 1, "sphinx.domains.javascript": 2, "sphinx.domains.math": 2, "sphinx.domains.python": 3, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.viewcode": 1, "sphinx": 57}, "alltitles": {"Using the API": [[0, "using-the-api"]], "Loading Configs": [[0, "loading-configs"]], "Training and Prediction": [[0, "training-and-prediction"]], "Utils": [[0, "utils"]], "cyto_dl package": [[1, "module-cyto_dl"]], "Subpackages": [[1, "subpackages"], [2, "subpackages"], [17, "subpackages"], [24, "subpackages"], [36, "subpackages"], [60, "subpackages"], [72, "subpackages"], [75, "subpackages"], [89, "subpackages"], [105, "subpackages"], [141, "subpackages"], [142, "subpackages"], [156, "subpackages"], [160, "subpackages"]], "Submodules": [[1, "submodules"], [2, "submodules"], [3, "submodules"], [8, "submodules"], [17, "submodules"], [19, "submodules"], [24, "submodules"], [27, "submodules"], [37, "submodules"], [44, "submodules"], [58, "submodules"], [60, "submodules"], [63, "submodules"], [66, "submodules"], [68, "submodules"], [72, "submodules"], [75, "submodules"], [78, "submodules"], [82, "submodules"], [86, "submodules"], [89, "submodules"], [95, "submodules"], [99, "submodules"], [105, "submodules"], [106, "submodules"], [109, "submodules"], [118, "submodules"], [133, "submodules"], [141, "submodules"], [142, "submodules"], [146, "submodules"], [157, "submodules"], [160, "submodules"], [169, "submodules"]], "cyto_dl.api package": [[2, "module-cyto_dl.api"]], "cyto_dl.api.cyto_dl_model package": [[3, "module-cyto_dl.api.cyto_dl_model"]], "cyto_dl.api.cyto_dl_model.cyto_dl_base_model module": [[4, "module-cyto_dl.api.cyto_dl_model.cyto_dl_base_model"]], "cyto_dl.api.cyto_dl_model.segmentation_plugin_model module": [[5, "module-cyto_dl.api.cyto_dl_model.segmentation_plugin_model"]], "cyto_dl.api.data module": [[6, "module-cyto_dl.api.data"]], "cyto_dl.api.model module": [[7, "module-cyto_dl.api.model"]], "cyto_dl.callbacks package": [[8, "module-cyto_dl.callbacks"]], "cyto_dl.callbacks.callback_utils module": [[9, "cyto-dl-callbacks-callback-utils-module"]], "cyto_dl.callbacks.csv_saver module": [[10, "module-cyto_dl.callbacks.csv_saver"]], "cyto_dl.callbacks.image_saver module": [[11, "module-cyto_dl.callbacks.image_saver"]], "cyto_dl.callbacks.latent_walk module": [[12, "cyto-dl-callbacks-latent-walk-module"]], "cyto_dl.callbacks.layer_freeze module": [[13, "module-cyto_dl.callbacks.layer_freeze"]], "cyto_dl.callbacks.model_utils module": [[14, "module-cyto_dl.callbacks.model_utils"]], "cyto_dl.callbacks.outlier_detection module": [[15, "module-cyto_dl.callbacks.outlier_detection"]], "cyto_dl.compile module": [[16, "cyto-dl-compile-module"]], "cyto_dl.dataframe package": [[17, "module-cyto_dl.dataframe"]], "cyto_dl.dataframe.readers module": [[18, "module-cyto_dl.dataframe.readers"]], "cyto_dl.dataframe.transforms package": [[19, "module-cyto_dl.dataframe.transforms"]], "cyto_dl.dataframe.transforms.filter module": [[20, "module-cyto_dl.dataframe.transforms.filter"]], "cyto_dl.dataframe.transforms.group_cols module": [[21, "module-cyto_dl.dataframe.transforms.group_cols"]], "cyto_dl.dataframe.transforms.misc module": [[22, "module-cyto_dl.dataframe.transforms.misc"]], "cyto_dl.dataframe.transforms.split module": [[23, "module-cyto_dl.dataframe.transforms.split"]], "cyto_dl.datamodules package": [[24, "module-cyto_dl.datamodules"]], "cyto_dl.datamodules.array module": [[25, "module-cyto_dl.datamodules.array"]], "cyto_dl.datamodules.data_dict module": [[26, "module-cyto_dl.datamodules.data_dict"]], "cyto_dl.datamodules.dataframe package": [[27, "module-cyto_dl.datamodules.dataframe"]], "cyto_dl.datamodules.dataframe.dataframe_datamodule module": [[28, "module-cyto_dl.datamodules.dataframe.dataframe_datamodule"]], "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule module": [[29, "module-cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule"]], "cyto_dl.datamodules.dataframe.utils module": [[30, "module-cyto_dl.datamodules.dataframe.utils"]], "cyto_dl.datamodules.folder module": [[31, "module-cyto_dl.datamodules.folder"]], "cyto_dl.datamodules.multidim_image module": [[32, "module-cyto_dl.datamodules.multidim_image"]], "cyto_dl.datamodules.smartcache module": [[33, "module-cyto_dl.datamodules.smartcache"]], "cyto_dl.datamodules.torchvision module": [[34, "module-cyto_dl.datamodules.torchvision"]], "cyto_dl.eval module": [[35, "module-cyto_dl.eval"]], "cyto_dl.image package": [[36, "module-cyto_dl.image"]], "cyto_dl.image.io package": [[37, "module-cyto_dl.image.io"]], "cyto_dl.image.io.aicsimage_loader module": [[38, "module-cyto_dl.image.io.aicsimage_loader"]], "cyto_dl.image.io.monai_bio_reader module": [[39, "module-cyto_dl.image.io.monai_bio_reader"]], "cyto_dl.image.io.numpy_reader module": [[40, "module-cyto_dl.image.io.numpy_reader"]], "cyto_dl.image.io.ome_zarr_reader module": [[41, "module-cyto_dl.image.io.ome_zarr_reader"]], "cyto_dl.image.io.polygon_loader module": [[42, "module-cyto_dl.image.io.polygon_loader"]], "cyto_dl.image.io.skimage_reader module": [[43, "module-cyto_dl.image.io.skimage_reader"]], "cyto_dl.image.transforms package": [[44, "module-cyto_dl.image.transforms"]], "cyto_dl.image.transforms.add_meta module": [[45, "module-cyto_dl.image.transforms.add_meta"]], "cyto_dl.image.transforms.bright_sampler module": [[46, "module-cyto_dl.image.transforms.bright_sampler"]], "cyto_dl.image.transforms.clip module": [[47, "module-cyto_dl.image.transforms.clip"]], "cyto_dl.image.transforms.contrastadjust module": [[48, "module-cyto_dl.image.transforms.contrastadjust"]], "cyto_dl.image.transforms.generate_jepa_masks module": [[49, "module-cyto_dl.image.transforms.generate_jepa_masks"]], "cyto_dl.image.transforms.merge module": [[50, "module-cyto_dl.image.transforms.merge"]], "cyto_dl.image.transforms.multiscale_cropper module": [[51, "module-cyto_dl.image.transforms.multiscale_cropper"]], "cyto_dl.image.transforms.pad module": [[52, "module-cyto_dl.image.transforms.pad"]], "cyto_dl.image.transforms.project module": [[53, "module-cyto_dl.image.transforms.project"]], "cyto_dl.image.transforms.rotation_mask_transform module": [[54, "cyto-dl-image-transforms-rotation-mask-transform-module"]], "cyto_dl.image.transforms.save module": [[55, "module-cyto_dl.image.transforms.save"]], "cyto_dl.image.transforms.so2_random_rotation module": [[56, "cyto-dl-image-transforms-so2-random-rotation-module"]], "cyto_dl.image.transforms.track_transforms module": [[57, "module-cyto_dl.image.transforms.track_transforms"]], "cyto_dl.loggers package": [[58, "module-cyto_dl.loggers"]], "cyto_dl.loggers.mlflow module": [[59, "module-cyto_dl.loggers.mlflow"]], "cyto_dl.models package": [[60, "module-cyto_dl.models"]], "cyto_dl.models.base_model module": [[61, "module-cyto_dl.models.base_model"]], "cyto_dl.models.basic_model module": [[62, "module-cyto_dl.models.basic_model"]], "cyto_dl.models.classification package": [[63, "module-cyto_dl.models.classification"]], "cyto_dl.models.classification.classification module": [[64, "module-cyto_dl.models.classification.classification"]], "cyto_dl.models.classification.timepoint_classification module": [[65, "module-cyto_dl.models.classification.timepoint_classification"]], "cyto_dl.models.contrastive package": [[66, "module-cyto_dl.models.contrastive"]], "cyto_dl.models.contrastive.contrastive module": [[67, "module-cyto_dl.models.contrastive.contrastive"]], "cyto_dl.models.handlers package": [[68, "cyto-dl-models-handlers-package"]], "cyto_dl.models.handlers.base_handler module": [[69, "cyto-dl-models-handlers-base-handler-module"]], "cyto_dl.models.handlers.image_handler module": [[70, "cyto-dl-models-handlers-image-handler-module"]], "cyto_dl.models.handlers.load_image_patch module": [[71, "cyto-dl-models-handlers-load-image-patch-module"]], "cyto_dl.models.im2im package": [[72, "module-cyto_dl.models.im2im"]], "cyto_dl.models.im2im.gan module": [[73, "module-cyto_dl.models.im2im.gan"]], "cyto_dl.models.im2im.multi_task module": [[74, "module-cyto_dl.models.im2im.multi_task"]], "cyto_dl.models.im2im.utils package": [[75, "module-cyto_dl.models.im2im.utils"]], "cyto_dl.models.im2im.utils.instance_seg module": [[76, "module-cyto_dl.models.im2im.utils.instance_seg"]], "cyto_dl.models.im2im.utils.noise_annealer module": [[77, "module-cyto_dl.models.im2im.utils.noise_annealer"]], "cyto_dl.models.im2im.utils.postprocessing package": [[78, "module-cyto_dl.models.im2im.utils.postprocessing"]], "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label module": [[79, "module-cyto_dl.models.im2im.utils.postprocessing.act_thresh_label"]], "cyto_dl.models.im2im.utils.postprocessing.auto_thresh module": [[80, "module-cyto_dl.models.im2im.utils.postprocessing.auto_thresh"]], "cyto_dl.models.im2im.utils.postprocessing.dict_to_im module": [[81, "module-cyto_dl.models.im2im.utils.postprocessing.dict_to_im"]], "cyto_dl.models.jepa package": [[82, "module-cyto_dl.models.jepa"]], "cyto_dl.models.jepa.ijepa module": [[83, "module-cyto_dl.models.jepa.ijepa"]], "cyto_dl.models.jepa.iwm module": [[84, "module-cyto_dl.models.jepa.iwm"]], "cyto_dl.models.jepa.jepa_base module": [[85, "module-cyto_dl.models.jepa.jepa_base"]], "cyto_dl.models.utils package": [[86, "module-cyto_dl.models.utils"]], "cyto_dl.models.utils.mlflow module": [[87, "module-cyto_dl.models.utils.mlflow"]], "cyto_dl.models.utils.utils module": [[88, "module-cyto_dl.models.utils.utils"]], "cyto_dl.models.vae package": [[89, "cyto-dl-models-vae-package"]], "cyto_dl.models.vae.base_vae module": [[90, "module-cyto_dl.models.vae.base_vae"]], "cyto_dl.models.vae.image_canon_vae module": [[91, "cyto-dl-models-vae-image-canon-vae-module"]], "cyto_dl.models.vae.image_encoder module": [[92, "cyto-dl-models-vae-image-encoder-module"]], "cyto_dl.models.vae.image_vae module": [[93, "cyto-dl-models-vae-image-vae-module"]], "cyto_dl.models.vae.latent_loss_vae module": [[94, "cyto-dl-models-vae-latent-loss-vae-module"]], "cyto_dl.models.vae.o2_spharm_vae package": [[95, "cyto-dl-models-vae-o2-spharm-vae-package"]], "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_encoder module": [[96, "cyto-dl-models-vae-o2-spharm-vae-o2-spharm-encoder-module"]], "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_vae module": [[97, "cyto-dl-models-vae-o2-spharm-vae-o2-spharm-vae-module"]], "cyto_dl.models.vae.point_cloud_vae module": [[98, "cyto-dl-models-vae-point-cloud-vae-module"]], "cyto_dl.models.vae.priors package": [[99, "module-cyto_dl.models.vae.priors"]], "cyto_dl.models.vae.priors.abstract_prior module": [[100, "module-cyto_dl.models.vae.priors.abstract_prior"]], "cyto_dl.models.vae.priors.gaussian module": [[101, "module-cyto_dl.models.vae.priors.gaussian"]], "cyto_dl.models.vae.priors.identity_prior module": [[102, "module-cyto_dl.models.vae.priors.identity_prior"]], "cyto_dl.models.vae.priors.joint_prior module": [[103, "module-cyto_dl.models.vae.priors.joint_prior"]], "cyto_dl.models.vae.tabular_vae module": [[104, "cyto-dl-models-vae-tabular-vae-module"]], "cyto_dl.nn package": [[105, "module-cyto_dl.nn"]], "cyto_dl.nn.discriminators package": [[106, "module-cyto_dl.nn.discriminators"]], "cyto_dl.nn.discriminators.multi_scale_discriminator module": [[107, "module-cyto_dl.nn.discriminators.multi_scale_discriminator"]], "cyto_dl.nn.discriminators.n_layer_discriminator module": [[108, "module-cyto_dl.nn.discriminators.n_layer_discriminator"]], "cyto_dl.nn.head package": [[109, "module-cyto_dl.nn.head"]], "cyto_dl.nn.head.base_head module": [[110, "module-cyto_dl.nn.head.base_head"]], "cyto_dl.nn.head.gan_head module": [[111, "module-cyto_dl.nn.head.gan_head"]], "cyto_dl.nn.head.gan_head_superres module": [[112, "module-cyto_dl.nn.head.gan_head_superres"]], "cyto_dl.nn.head.mae_head module": [[113, "module-cyto_dl.nn.head.mae_head"]], "cyto_dl.nn.head.mask_head module": [[114, "module-cyto_dl.nn.head.mask_head"]], "cyto_dl.nn.head.res_blocks_head module": [[115, "module-cyto_dl.nn.head.res_blocks_head"]], "cyto_dl.nn.head.vic_reg module": [[116, "module-cyto_dl.nn.head.vic_reg"]], "cyto_dl.nn.hr_skip module": [[117, "module-cyto_dl.nn.hr_skip"]], "cyto_dl.nn.losses package": [[118, "module-cyto_dl.nn.losses"]], "cyto_dl.nn.losses.adversarial_loss module": [[119, "module-cyto_dl.nn.losses.adversarial_loss"]], "cyto_dl.nn.losses.chamfer_loss module": [[120, "module-cyto_dl.nn.losses.chamfer_loss"]], "cyto_dl.nn.losses.continuous_bernoulli module": [[121, "module-cyto_dl.nn.losses.continuous_bernoulli"]], "cyto_dl.nn.losses.cosine_loss module": [[122, "module-cyto_dl.nn.losses.cosine_loss"]], "cyto_dl.nn.losses.gan_loss module": [[123, "module-cyto_dl.nn.losses.gan_loss"]], "cyto_dl.nn.losses.gaussian_nll_loss module": [[124, "module-cyto_dl.nn.losses.gaussian_nll_loss"]], "cyto_dl.nn.losses.geomloss module": [[125, "cyto-dl-nn-losses-geomloss-module"]], "cyto_dl.nn.losses.loss_wrapper module": [[126, "module-cyto_dl.nn.losses.loss_wrapper"]], "cyto_dl.nn.losses.spharm_loss module": [[127, "cyto-dl-nn-losses-spharm-loss-module"]], "cyto_dl.nn.losses.threshold_loss module": [[128, "module-cyto_dl.nn.losses.threshold_loss"]], "cyto_dl.nn.losses.vic_reg module": [[129, "module-cyto_dl.nn.losses.vic_reg"]], "cyto_dl.nn.losses.weibull module": [[130, "module-cyto_dl.nn.losses.weibull"]], "cyto_dl.nn.losses.weighted_mse_loss module": [[131, "module-cyto_dl.nn.losses.weighted_mse_loss"]], "cyto_dl.nn.mlp module": [[132, "module-cyto_dl.nn.mlp"]], "cyto_dl.nn.point_cloud package": [[133, "cyto-dl-nn-point-cloud-package"]], "cyto_dl.nn.point_cloud.dgcnn module": [[134, "cyto-dl-nn-point-cloud-dgcnn-module"]], "cyto_dl.nn.point_cloud.folding_net module": [[135, "cyto-dl-nn-point-cloud-folding-net-module"]], "cyto_dl.nn.point_cloud.graph_functions module": [[136, "cyto-dl-nn-point-cloud-graph-functions-module"]], "cyto_dl.nn.point_cloud.vnn module": [[137, "cyto-dl-nn-point-cloud-vnn-module"]], "cyto_dl.nn.res_unit module": [[138, "module-cyto_dl.nn.res_unit"]], "cyto_dl.nn.spatial_transformer module": [[139, "module-cyto_dl.nn.spatial_transformer"]], "cyto_dl.nn.track_sequence_predictor module": [[140, "module-cyto_dl.nn.track_sequence_predictor"]], "cyto_dl.nn.vits package": [[141, "module-cyto_dl.nn.vits"]], "cyto_dl.nn.vits.blocks package": [[142, "module-cyto_dl.nn.vits.blocks"]], "cyto_dl.nn.vits.blocks.cross_attention module": [[143, "module-cyto_dl.nn.vits.blocks.cross_attention"]], "cyto_dl.nn.vits.blocks.intermediate_weigher module": [[144, "module-cyto_dl.nn.vits.blocks.intermediate_weigher"]], "cyto_dl.nn.vits.blocks.masked_unit_attention module": [[145, "module-cyto_dl.nn.vits.blocks.masked_unit_attention"]], "cyto_dl.nn.vits.blocks.patchify package": [[146, "module-cyto_dl.nn.vits.blocks.patchify"]], "cyto_dl.nn.vits.blocks.patchify.patchify module": [[147, "module-cyto_dl.nn.vits.blocks.patchify.patchify"]], "cyto_dl.nn.vits.blocks.patchify.patchify_base module": [[148, "module-cyto_dl.nn.vits.blocks.patchify.patchify_base"]], "cyto_dl.nn.vits.blocks.patchify.patchify_hiera module": [[149, "module-cyto_dl.nn.vits.blocks.patchify.patchify_hiera"]], "cyto_dl.nn.vits.decoder module": [[150, "module-cyto_dl.nn.vits.decoder"]], "cyto_dl.nn.vits.encoder module": [[151, "module-cyto_dl.nn.vits.encoder"]], "cyto_dl.nn.vits.mae module": [[152, "module-cyto_dl.nn.vits.mae"]], "cyto_dl.nn.vits.predictor module": [[153, "module-cyto_dl.nn.vits.predictor"]], "cyto_dl.nn.vits.seg module": [[154, "module-cyto_dl.nn.vits.seg"]], "cyto_dl.nn.vits.utils module": [[155, "module-cyto_dl.nn.vits.utils"]], "cyto_dl.point_cloud package": [[156, "module-cyto_dl.point_cloud"]], "cyto_dl.point_cloud.io package": [[157, "cyto-dl-point-cloud-io-package"]], "cyto_dl.point_cloud.io.read_pcloud module": [[158, "cyto-dl-point-cloud-io-read-pcloud-module"]], "cyto_dl.train module": [[159, "module-cyto_dl.train"]], "cyto_dl.utils package": [[160, "module-cyto_dl.utils"]], "cyto_dl.utils.array module": [[161, "module-cyto_dl.utils.array"]], "cyto_dl.utils.checkpoint module": [[162, "module-cyto_dl.utils.checkpoint"]], "cyto_dl.utils.config module": [[163, "module-cyto_dl.utils.config"]], "cyto_dl.utils.download_test_data module": [[164, "module-cyto_dl.utils.download_test_data"]], "cyto_dl.utils.dummy_dataset module": [[165, "module-cyto_dl.utils.dummy_dataset"]], "cyto_dl.utils.pylogger module": [[166, "module-cyto_dl.utils.pylogger"]], "cyto_dl.utils.rich_utils module": [[167, "module-cyto_dl.utils.rich_utils"]], "cyto_dl.utils.rotation module": [[168, "module-cyto_dl.utils.rotation"]], "cyto_dl.utils.spharm package": [[169, "cyto-dl-utils-spharm-package"]], "cyto_dl.utils.spharm.reconstruction module": [[170, "cyto-dl-utils-spharm-reconstruction-module"]], "cyto_dl.utils.spharm.rotation module": [[171, "cyto-dl-utils-spharm-rotation-module"]], "cyto_dl.utils.template_utils module": [[172, "module-cyto_dl.utils.template_utils"]], "Overview": [[173, "overview"], [176, "overview"]], "data": [[173, "data"], [176, "data"]], "experiment": [[173, "experiment"], [176, "experiment"]], "model": [[173, "model"], [176, "model"]], "Installation": [[174, "installation"]], "cyto_dl": [[175, "cyto-dl"]], "Modifying the Example Configs": [[177, "modifying-the-example-configs"]], "Training": [[177, "training"]], "Testing/Prediction": [[177, "testing-prediction"]]}, "indexentries": {"cyto_dl": [[1, "module-cyto_dl"]], "module": [[1, "module-cyto_dl"], [2, "module-cyto_dl.api"], [3, "module-cyto_dl.api.cyto_dl_model"], [4, "module-cyto_dl.api.cyto_dl_model.cyto_dl_base_model"], [5, "module-cyto_dl.api.cyto_dl_model.segmentation_plugin_model"], [6, "module-cyto_dl.api.data"], [7, "module-cyto_dl.api.model"], [8, "module-cyto_dl.callbacks"], [10, "module-cyto_dl.callbacks.csv_saver"], [11, "module-cyto_dl.callbacks.image_saver"], [13, "module-cyto_dl.callbacks.layer_freeze"], [14, "module-cyto_dl.callbacks.model_utils"], [15, "module-cyto_dl.callbacks.outlier_detection"], [17, "module-cyto_dl.dataframe"], [18, "module-cyto_dl.dataframe.readers"], [19, "module-cyto_dl.dataframe.transforms"], [20, "module-cyto_dl.dataframe.transforms.filter"], [21, "module-cyto_dl.dataframe.transforms.group_cols"], [22, "module-cyto_dl.dataframe.transforms.misc"], [23, "module-cyto_dl.dataframe.transforms.split"], [24, "module-cyto_dl.datamodules"], [25, "module-cyto_dl.datamodules.array"], [26, "module-cyto_dl.datamodules.data_dict"], [27, "module-cyto_dl.datamodules.dataframe"], [28, "module-cyto_dl.datamodules.dataframe.dataframe_datamodule"], [29, "module-cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule"], [30, "module-cyto_dl.datamodules.dataframe.utils"], [31, "module-cyto_dl.datamodules.folder"], [32, "module-cyto_dl.datamodules.multidim_image"], [33, "module-cyto_dl.datamodules.smartcache"], [34, "module-cyto_dl.datamodules.torchvision"], [35, "module-cyto_dl.eval"], [36, "module-cyto_dl.image"], [37, "module-cyto_dl.image.io"], [38, "module-cyto_dl.image.io.aicsimage_loader"], [39, "module-cyto_dl.image.io.monai_bio_reader"], [40, "module-cyto_dl.image.io.numpy_reader"], [41, "module-cyto_dl.image.io.ome_zarr_reader"], [42, "module-cyto_dl.image.io.polygon_loader"], [43, "module-cyto_dl.image.io.skimage_reader"], [44, "module-cyto_dl.image.transforms"], [45, "module-cyto_dl.image.transforms.add_meta"], [46, "module-cyto_dl.image.transforms.bright_sampler"], [47, "module-cyto_dl.image.transforms.clip"], [48, "module-cyto_dl.image.transforms.contrastadjust"], [49, "module-cyto_dl.image.transforms.generate_jepa_masks"], [50, "module-cyto_dl.image.transforms.merge"], [51, "module-cyto_dl.image.transforms.multiscale_cropper"], [52, "module-cyto_dl.image.transforms.pad"], [53, "module-cyto_dl.image.transforms.project"], [55, "module-cyto_dl.image.transforms.save"], [57, "module-cyto_dl.image.transforms.track_transforms"], [58, "module-cyto_dl.loggers"], [59, "module-cyto_dl.loggers.mlflow"], [60, "module-cyto_dl.models"], [61, "module-cyto_dl.models.base_model"], [62, "module-cyto_dl.models.basic_model"], [63, "module-cyto_dl.models.classification"], [64, "module-cyto_dl.models.classification.classification"], [65, "module-cyto_dl.models.classification.timepoint_classification"], [66, "module-cyto_dl.models.contrastive"], [67, "module-cyto_dl.models.contrastive.contrastive"], [72, "module-cyto_dl.models.im2im"], [73, "module-cyto_dl.models.im2im.gan"], [74, "module-cyto_dl.models.im2im.multi_task"], [75, "module-cyto_dl.models.im2im.utils"], [76, "module-cyto_dl.models.im2im.utils.instance_seg"], [77, "module-cyto_dl.models.im2im.utils.noise_annealer"], [78, "module-cyto_dl.models.im2im.utils.postprocessing"], [79, "module-cyto_dl.models.im2im.utils.postprocessing.act_thresh_label"], [80, "module-cyto_dl.models.im2im.utils.postprocessing.auto_thresh"], [81, "module-cyto_dl.models.im2im.utils.postprocessing.dict_to_im"], [82, "module-cyto_dl.models.jepa"], [83, "module-cyto_dl.models.jepa.ijepa"], [84, "module-cyto_dl.models.jepa.iwm"], [85, "module-cyto_dl.models.jepa.jepa_base"], [86, "module-cyto_dl.models.utils"], [87, "module-cyto_dl.models.utils.mlflow"], [88, "module-cyto_dl.models.utils.utils"], [90, "module-cyto_dl.models.vae.base_vae"], [99, "module-cyto_dl.models.vae.priors"], [100, "module-cyto_dl.models.vae.priors.abstract_prior"], [101, "module-cyto_dl.models.vae.priors.gaussian"], [102, "module-cyto_dl.models.vae.priors.identity_prior"], [103, "module-cyto_dl.models.vae.priors.joint_prior"], [105, "module-cyto_dl.nn"], [106, "module-cyto_dl.nn.discriminators"], [107, "module-cyto_dl.nn.discriminators.multi_scale_discriminator"], [108, "module-cyto_dl.nn.discriminators.n_layer_discriminator"], [109, "module-cyto_dl.nn.head"], [110, "module-cyto_dl.nn.head.base_head"], [111, "module-cyto_dl.nn.head.gan_head"], [112, "module-cyto_dl.nn.head.gan_head_superres"], [113, "module-cyto_dl.nn.head.mae_head"], [114, "module-cyto_dl.nn.head.mask_head"], [115, "module-cyto_dl.nn.head.res_blocks_head"], [116, "module-cyto_dl.nn.head.vic_reg"], [117, "module-cyto_dl.nn.hr_skip"], [118, "module-cyto_dl.nn.losses"], [119, "module-cyto_dl.nn.losses.adversarial_loss"], [120, "module-cyto_dl.nn.losses.chamfer_loss"], [121, "module-cyto_dl.nn.losses.continuous_bernoulli"], [122, "module-cyto_dl.nn.losses.cosine_loss"], [123, "module-cyto_dl.nn.losses.gan_loss"], [124, "module-cyto_dl.nn.losses.gaussian_nll_loss"], [126, "module-cyto_dl.nn.losses.loss_wrapper"], [128, "module-cyto_dl.nn.losses.threshold_loss"], [129, "module-cyto_dl.nn.losses.vic_reg"], [130, "module-cyto_dl.nn.losses.weibull"], [131, "module-cyto_dl.nn.losses.weighted_mse_loss"], [132, "module-cyto_dl.nn.mlp"], [138, "module-cyto_dl.nn.res_unit"], [139, "module-cyto_dl.nn.spatial_transformer"], [140, "module-cyto_dl.nn.track_sequence_predictor"], [141, "module-cyto_dl.nn.vits"], [142, "module-cyto_dl.nn.vits.blocks"], [143, "module-cyto_dl.nn.vits.blocks.cross_attention"], [144, "module-cyto_dl.nn.vits.blocks.intermediate_weigher"], [145, "module-cyto_dl.nn.vits.blocks.masked_unit_attention"], [146, "module-cyto_dl.nn.vits.blocks.patchify"], [147, "module-cyto_dl.nn.vits.blocks.patchify.patchify"], [148, "module-cyto_dl.nn.vits.blocks.patchify.patchify_base"], [149, "module-cyto_dl.nn.vits.blocks.patchify.patchify_hiera"], [150, "module-cyto_dl.nn.vits.decoder"], [151, "module-cyto_dl.nn.vits.encoder"], [152, "module-cyto_dl.nn.vits.mae"], [153, "module-cyto_dl.nn.vits.predictor"], [154, "module-cyto_dl.nn.vits.seg"], [155, "module-cyto_dl.nn.vits.utils"], [156, "module-cyto_dl.point_cloud"], [159, "module-cyto_dl.train"], [160, "module-cyto_dl.utils"], [161, "module-cyto_dl.utils.array"], [162, "module-cyto_dl.utils.checkpoint"], [163, "module-cyto_dl.utils.config"], [164, "module-cyto_dl.utils.download_test_data"], [165, "module-cyto_dl.utils.dummy_dataset"], [166, "module-cyto_dl.utils.pylogger"], [167, "module-cyto_dl.utils.rich_utils"], [168, "module-cyto_dl.utils.rotation"], [172, "module-cyto_dl.utils.template_utils"]], "cyto_dl.api": [[2, "module-cyto_dl.api"]], "cyto_dl.api.cyto_dl_model": [[3, "module-cyto_dl.api.cyto_dl_model"]], "cytodlbasemodel (class in cyto_dl.api.cyto_dl_model.cyto_dl_base_model)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel"]], "cyto_dl.api.cyto_dl_model.cyto_dl_base_model": [[4, "module-cyto_dl.api.cyto_dl_model.cyto_dl_base_model"]], "from_default_config() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel class method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.from_default_config"]], "from_existing_config() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel class method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.from_existing_config"]], "get_config() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.get_config"]], "get_experiment_name() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.get_experiment_name"]], "predict() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.predict"]], "save_config() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.save_config"]], "set_experiment_name() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.set_experiment_name"]], "train() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.train"]], "segmentationpluginmodel (class in cyto_dl.api.cyto_dl_model.segmentation_plugin_model)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel"]], "cyto_dl.api.cyto_dl_model.segmentation_plugin_model": [[5, "module-cyto_dl.api.cyto_dl_model.segmentation_plugin_model"]], "get_hardware_type() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_hardware_type"]], "get_input_channel() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_input_channel"]], "get_manifest_column_names() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_manifest_column_names"]], "get_patch_size() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_patch_size"]], "get_raw_image_channels() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_raw_image_channels"]], "get_split_column() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_split_column"]], "remove_split_column() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.remove_split_column"]], "set_hardware_type() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_hardware_type"]], "set_input_channel() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_input_channel"]], "set_manifest_column_names() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_manifest_column_names"]], "set_patch_size() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_patch_size"]], "set_raw_image_channels() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_raw_image_channels"]], "set_split_column() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_split_column"]], "autothresholdmethod (class in cyto_dl.api.data)": [[6, "cyto_dl.api.data.AutoThresholdMethod"]], "cpu (cyto_dl.api.data.hardwaretype attribute)": [[6, "cyto_dl.api.data.HardwareType.CPU"]], "experimenttype (class in cyto_dl.api.data)": [[6, "cyto_dl.api.data.ExperimentType"]], "gan (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.GAN"]], "gpu (cyto_dl.api.data.hardwaretype attribute)": [[6, "cyto_dl.api.data.HardwareType.GPU"]], "hardwaretype (class in cyto_dl.api.data)": [[6, "cyto_dl.api.data.HardwareType"]], "instance_seg (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.INSTANCE_SEG"]], "isodata (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.ISODATA"]], "label_free (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.LABEL_FREE"]], "large (cyto_dl.api.data.patchsize attribute)": [[6, "cyto_dl.api.data.PatchSize.LARGE"]], "li (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.LI"]], "local (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.LOCAL"]], "mean (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.MEAN"]], "medium (cyto_dl.api.data.patchsize attribute)": [[6, "cyto_dl.api.data.PatchSize.MEDIUM"]], "minimum (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.MINIMUM"]], "multiotsu (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.MULTIOTSU"]], "niblack (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.NIBLACK"]], "otsu (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.OTSU"]], "patchsize (class in cyto_dl.api.data)": [[6, "cyto_dl.api.data.PatchSize"]], "sauvola (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.SAUVOLA"]], "segmentation (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.SEGMENTATION"]], "segmentation_array (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.SEGMENTATION_ARRAY"]], "segmentation_plugin (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.SEGMENTATION_PLUGIN"]], "small (cyto_dl.api.data.patchsize attribute)": [[6, "cyto_dl.api.data.PatchSize.SMALL"]], "triangle (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.TRIANGLE"]], "yen (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.YEN"]], "cyto_dl.api.data": [[6, "module-cyto_dl.api.data"]], "cytodlmodel (class in cyto_dl.api.model)": [[7, "cyto_dl.api.model.CytoDLModel"]], "cyto_dl.api.model": [[7, "module-cyto_dl.api.model"]], "download_example_data() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.download_example_data"]], "load_config_from_dict() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.load_config_from_dict"]], "load_config_from_file() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.load_config_from_file"]], "load_default_experiment() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.load_default_experiment"]], "override_config() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.override_config"]], "predict() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.predict"]], "print_config() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.print_config"]], "save_config() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.save_config"]], "train() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.train"]], "cyto_dl.callbacks": [[8, "module-cyto_dl.callbacks"]], "csvsaver (class in cyto_dl.callbacks.csv_saver)": [[10, "cyto_dl.callbacks.csv_saver.CSVSaver"]], "jepasaver (class in cyto_dl.callbacks.csv_saver)": [[10, "cyto_dl.callbacks.csv_saver.JEPASaver"]], "cyto_dl.callbacks.csv_saver": [[10, "module-cyto_dl.callbacks.csv_saver"]], "on_predict_epoch_end() (cyto_dl.callbacks.csv_saver.csvsaver method)": [[10, "cyto_dl.callbacks.csv_saver.CSVSaver.on_predict_epoch_end"]], "pred_to_csv() (cyto_dl.callbacks.csv_saver.csvsaver method)": [[10, "cyto_dl.callbacks.csv_saver.CSVSaver.pred_to_csv"]], "pred_to_csv() (cyto_dl.callbacks.csv_saver.jepasaver method)": [[10, "cyto_dl.callbacks.csv_saver.JEPASaver.pred_to_csv"]], "save_feats() (cyto_dl.callbacks.csv_saver.csvsaver method)": [[10, "cyto_dl.callbacks.csv_saver.CSVSaver.save_feats"]], "imagesaver (class in cyto_dl.callbacks.image_saver)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver"]], "cyto_dl.callbacks.image_saver": [[11, "module-cyto_dl.callbacks.image_saver"]], "on_predict_batch_end() (cyto_dl.callbacks.image_saver.imagesaver method)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver.on_predict_batch_end"]], "on_test_batch_end() (cyto_dl.callbacks.image_saver.imagesaver method)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver.on_test_batch_end"]], "on_train_batch_end() (cyto_dl.callbacks.image_saver.imagesaver method)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver.on_train_batch_end"]], "on_validation_batch_end() (cyto_dl.callbacks.image_saver.imagesaver method)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver.on_validation_batch_end"]], "save() (cyto_dl.callbacks.image_saver.imagesaver method)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver.save"]], "layerfreeze (class in cyto_dl.callbacks.layer_freeze)": [[13, "cyto_dl.callbacks.layer_freeze.LayerFreeze"]], "cyto_dl.callbacks.layer_freeze": [[13, "module-cyto_dl.callbacks.layer_freeze"]], "setup() (cyto_dl.callbacks.layer_freeze.layerfreeze method)": [[13, "cyto_dl.callbacks.layer_freeze.LayerFreeze.setup"]], "getembeddings (class in cyto_dl.callbacks.model_utils)": [[14, "cyto_dl.callbacks.model_utils.GetEmbeddings"]], "cyto_dl.callbacks.model_utils": [[14, "module-cyto_dl.callbacks.model_utils"]], "get_all_embeddings() (in module cyto_dl.callbacks.model_utils)": [[14, "cyto_dl.callbacks.model_utils.get_all_embeddings"]], "on_test_epoch_end() (cyto_dl.callbacks.model_utils.getembeddings method)": [[14, "cyto_dl.callbacks.model_utils.GetEmbeddings.on_test_epoch_end"]], "save_predictions_classifier() (in module cyto_dl.callbacks.model_utils)": [[14, "cyto_dl.callbacks.model_utils.save_predictions_classifier"]], "outlierdetection (class in cyto_dl.callbacks.outlier_detection)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection"]], "calculate_mahalanobis_hook() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.calculate_mahalanobis_hook"]], "cyto_dl.callbacks.outlier_detection": [[15, "module-cyto_dl.callbacks.outlier_detection"]], "flatten_activations() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.flatten_activations"]], "on_load_checkpoint() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_load_checkpoint"]], "on_predict_batch_end() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_predict_batch_end"]], "on_predict_epoch_start() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_predict_epoch_start"]], "on_save_checkpoint() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_save_checkpoint"]], "on_test_batch_end() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_test_batch_end"]], "on_test_epoch_start() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_test_epoch_start"]], "on_train_epoch_start() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_train_epoch_start"]], "update_covariance_hook() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.update_covariance_hook"]], "cyto_dl.dataframe": [[17, "module-cyto_dl.dataframe"]], "cyto_dl.dataframe.readers": [[18, "module-cyto_dl.dataframe.readers"]], "filter_columns() (in module cyto_dl.dataframe.readers)": [[18, "cyto_dl.dataframe.readers.filter_columns"]], "read_csv() (in module cyto_dl.dataframe.readers)": [[18, "cyto_dl.dataframe.readers.read_csv"]], "read_dataframe() (in module cyto_dl.dataframe.readers)": [[18, "cyto_dl.dataframe.readers.read_dataframe"]], "read_h5ad() (in module cyto_dl.dataframe.readers)": [[18, "cyto_dl.dataframe.readers.read_h5ad"]], "read_parquet() (in module cyto_dl.dataframe.readers)": [[18, "cyto_dl.dataframe.readers.read_parquet"]], "cyto_dl.dataframe.transforms": [[19, "module-cyto_dl.dataframe.transforms"]], "cyto_dl.dataframe.transforms.filter": [[20, "module-cyto_dl.dataframe.transforms.filter"]], "filter_columns() (in module cyto_dl.dataframe.transforms.filter)": [[20, "cyto_dl.dataframe.transforms.filter.filter_columns"]], "filter_rows() (in module cyto_dl.dataframe.transforms.filter)": [[20, "cyto_dl.dataframe.transforms.filter.filter_rows"]], "groupcols (class in cyto_dl.dataframe.transforms.group_cols)": [[21, "cyto_dl.dataframe.transforms.group_cols.GroupCols"]], "cyto_dl.dataframe.transforms.group_cols": [[21, "module-cyto_dl.dataframe.transforms.group_cols"]], "append_class_weights() (in module cyto_dl.dataframe.transforms.misc)": [[22, "cyto_dl.dataframe.transforms.misc.append_class_weights"]], "append_labels_to_integers() (in module cyto_dl.dataframe.transforms.misc)": [[22, "cyto_dl.dataframe.transforms.misc.append_labels_to_integers"]], "append_one_hot() (in module cyto_dl.dataframe.transforms.misc)": [[22, "cyto_dl.dataframe.transforms.misc.append_one_hot"]], "cyto_dl.dataframe.transforms.misc": [[22, "module-cyto_dl.dataframe.transforms.misc"]], "make_random_df() (in module cyto_dl.dataframe.transforms.misc)": [[22, "cyto_dl.dataframe.transforms.misc.make_random_df"]], "cyto_dl.dataframe.transforms.split": [[23, "module-cyto_dl.dataframe.transforms.split"]], "sample_n_each() (in module cyto_dl.dataframe.transforms.split)": [[23, "cyto_dl.dataframe.transforms.split.sample_n_each"]], "split_dataframe() (in module cyto_dl.dataframe.transforms.split)": [[23, "cyto_dl.dataframe.transforms.split.split_dataframe"]], "cyto_dl.datamodules": [[24, "module-cyto_dl.datamodules"]], "cyto_dl.datamodules.array": [[25, "module-cyto_dl.datamodules.array"]], "make_array_dataloader() (in module cyto_dl.datamodules.array)": [[25, "cyto_dl.datamodules.array.make_array_dataloader"]], "cyto_dl.datamodules.data_dict": [[26, "module-cyto_dl.datamodules.data_dict"]], "make_data_dict_dataloader() (in module cyto_dl.datamodules.data_dict)": [[26, "cyto_dl.datamodules.data_dict.make_data_dict_dataloader"]], "cyto_dl.datamodules.dataframe": [[27, "module-cyto_dl.datamodules.dataframe"]], "dataframedatamodule (class in cyto_dl.datamodules.dataframe.dataframe_datamodule)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule"]], "cyto_dl.datamodules.dataframe.dataframe_datamodule": [[28, "module-cyto_dl.datamodules.dataframe.dataframe_datamodule"]], "get_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.get_dataloader"]], "get_dataset() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.get_dataset"]], "make_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.make_dataloader"]], "predict_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.predict_dataloader"]], "test_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.test_dataloader"]], "train_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.train_dataloader"]], "val_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.val_dataloader"]], "groupeddataframedatamodule (class in cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule)": [[29, "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule.GroupedDataframeDatamodule"]], "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule": [[29, "module-cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule"]], "make_dataloader() (cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule.groupeddataframedatamodule method)": [[29, "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule.GroupedDataframeDatamodule.make_dataloader"]], "alternatingbatchsampler (class in cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.AlternatingBatchSampler"]], "removenankeysd (class in cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.RemoveNaNKeysd"]], "cyto_dl.datamodules.dataframe.utils": [[30, "module-cyto_dl.datamodules.dataframe.utils"]], "get_canonical_split_name() (in module cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.get_canonical_split_name"]], "get_dataset() (in module cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.get_dataset"]], "make_multiple_dataframe_splits() (in module cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.make_multiple_dataframe_splits"]], "make_single_dataframe_splits() (in module cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.make_single_dataframe_splits"]], "parse_transforms() (in module cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.parse_transforms"]], "cyto_dl.datamodules.folder": [[31, "module-cyto_dl.datamodules.folder"]], "make_folder_dataloader() (in module cyto_dl.datamodules.folder)": [[31, "cyto_dl.datamodules.folder.make_folder_dataloader"]], "multidimimagedataset (class in cyto_dl.datamodules.multidim_image)": [[32, "cyto_dl.datamodules.multidim_image.MultiDimImageDataset"]], "create_metatensor() (cyto_dl.datamodules.multidim_image.multidimimagedataset method)": [[32, "cyto_dl.datamodules.multidim_image.MultiDimImageDataset.create_metatensor"]], "cyto_dl.datamodules.multidim_image": [[32, "module-cyto_dl.datamodules.multidim_image"]], "get_per_file_args() (cyto_dl.datamodules.multidim_image.multidimimagedataset method)": [[32, "cyto_dl.datamodules.multidim_image.MultiDimImageDataset.get_per_file_args"]], "is_batch() (cyto_dl.datamodules.multidim_image.multidimimagedataset method)": [[32, "cyto_dl.datamodules.multidim_image.MultiDimImageDataset.is_batch"]], "make_multidim_image_dataloader() (in module cyto_dl.datamodules.multidim_image)": [[32, "cyto_dl.datamodules.multidim_image.make_multidim_image_dataloader"]], "smartcachedatamodule (class in cyto_dl.datamodules.smartcache)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule"]], "cyto_dl.datamodules.smartcache": [[33, "module-cyto_dl.datamodules.smartcache"]], "get_per_file_args() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.get_per_file_args"]], "make_dataloader() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.make_dataloader"]], "predict_dataloader() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.predict_dataloader"]], "prepare_data() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.prepare_data"]], "setup() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.setup"]], "test_dataloader() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.test_dataloader"]], "train_dataloader() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.train_dataloader"]], "val_dataloader() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.val_dataloader"]], "cifar10 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CIFAR10"]], "cifar100 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CIFAR100"]], "clevrclassification (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CLEVRClassification"]], "caltech101 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Caltech101"]], "caltech256 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Caltech256"]], "celeba (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CelebA"]], "cityscapes (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Cityscapes"]], "cococaptions (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CocoCaptions"]], "cocodetection (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CocoDetection"]], "country211 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Country211"]], "dtd (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.DTD"]], "datasetfolder (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.DatasetFolder"]], "emnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.EMNIST"]], "eurosat (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.EuroSAT"]], "fer2013 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.FER2013"]], "fgvcaircraft (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.FGVCAircraft"]], "fakedata (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.FakeData"]], "fashionmnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.FashionMNIST"]], "flickr30k (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Flickr30k"]], "flickr8k (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Flickr8k"]], "flowers102 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Flowers102"]], "food101 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Food101"]], "gtsrb (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.GTSRB"]], "hmdb51 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.HMDB51"]], "inaturalist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.INaturalist"]], "imagefolder (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.ImageFolder"]], "imagenet (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.ImageNet"]], "kmnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.KMNIST"]], "kinetics (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Kinetics"]], "kitti (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Kitti"]], "lfwpairs (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.LFWPairs"]], "lfwpeople (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.LFWPeople"]], "lsun (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.LSUN"]], "lsunclass (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.LSUNClass"]], "mnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.MNIST"]], "movingmnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.MovingMNIST"]], "omniglot (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Omniglot"]], "oxfordiiitpet (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.OxfordIIITPet"]], "pcam (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.PCAM"]], "phototour (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.PhotoTour"]], "places365 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Places365"]], "qmnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.QMNIST"]], "renderedsst2 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.RenderedSST2"]], "sbdataset (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.SBDataset"]], "sbu (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.SBU"]], "semeion (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.SEMEION"]], "stl10 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.STL10"]], "sun397 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.SUN397"]], "svhn (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.SVHN"]], "stanfordcars (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.StanfordCars"]], "ucf101 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.UCF101"]], "usps (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.USPS"]], "vocdetection (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.VOCDetection"]], "vocsegmentation (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.VOCSegmentation"]], "visiondataset (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.VisionDataset"]], "widerface (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.WIDERFace"]], "cyto_dl.datamodules.torchvision": [[34, "module-cyto_dl.datamodules.torchvision"]], "cyto_dl.eval": [[35, "module-cyto_dl.eval"]], "main() (in module cyto_dl.eval)": [[35, "cyto_dl.eval.main"]], "cyto_dl.image": [[36, "module-cyto_dl.image"]], "cyto_dl.image.io": [[37, "module-cyto_dl.image.io"]], "aicsimageloaderd (class in cyto_dl.image.io.aicsimage_loader)": [[38, "cyto_dl.image.io.aicsimage_loader.AICSImageLoaderd"]], "cyto_dl.image.io.aicsimage_loader": [[38, "module-cyto_dl.image.io.aicsimage_loader"]], "split_args() (cyto_dl.image.io.aicsimage_loader.aicsimageloaderd method)": [[38, "cyto_dl.image.io.aicsimage_loader.AICSImageLoaderd.split_args"]], "monaibioreader (class in cyto_dl.image.io.monai_bio_reader)": [[39, "cyto_dl.image.io.monai_bio_reader.MonaiBioReader"]], "cyto_dl.image.io.monai_bio_reader": [[39, "module-cyto_dl.image.io.monai_bio_reader"]], "get_data() (cyto_dl.image.io.monai_bio_reader.monaibioreader method)": [[39, "cyto_dl.image.io.monai_bio_reader.MonaiBioReader.get_data"]], "read() (cyto_dl.image.io.monai_bio_reader.monaibioreader method)": [[39, "cyto_dl.image.io.monai_bio_reader.MonaiBioReader.read"]], "verify_suffix() (cyto_dl.image.io.monai_bio_reader.monaibioreader method)": [[39, "cyto_dl.image.io.monai_bio_reader.MonaiBioReader.verify_suffix"]], "readnumpyfile (class in cyto_dl.image.io.numpy_reader)": [[40, "cyto_dl.image.io.numpy_reader.ReadNumpyFile"]], "cyto_dl.image.io.numpy_reader": [[40, "module-cyto_dl.image.io.numpy_reader"]], "omezarrreader (class in cyto_dl.image.io.ome_zarr_reader)": [[41, "cyto_dl.image.io.ome_zarr_reader.OmeZarrReader"]], "cyto_dl.image.io.ome_zarr_reader": [[41, "module-cyto_dl.image.io.ome_zarr_reader"]], "get_data() (cyto_dl.image.io.ome_zarr_reader.omezarrreader method)": [[41, "cyto_dl.image.io.ome_zarr_reader.OmeZarrReader.get_data"]], "read() (cyto_dl.image.io.ome_zarr_reader.omezarrreader method)": [[41, "cyto_dl.image.io.ome_zarr_reader.OmeZarrReader.read"]], "verify_suffix() (cyto_dl.image.io.ome_zarr_reader.omezarrreader method)": [[41, "cyto_dl.image.io.ome_zarr_reader.OmeZarrReader.verify_suffix"]], "polygonloaderd (class in cyto_dl.image.io.polygon_loader)": [[42, "cyto_dl.image.io.polygon_loader.PolygonLoaderd"]], "cyto_dl.image.io.polygon_loader": [[42, "module-cyto_dl.image.io.polygon_loader"]], "skimagereader (class in cyto_dl.image.io.skimage_reader)": [[43, "cyto_dl.image.io.skimage_reader.SkimageReader"]], "cyto_dl.image.io.skimage_reader": [[43, "module-cyto_dl.image.io.skimage_reader"]], "get_data() (cyto_dl.image.io.skimage_reader.skimagereader method)": [[43, "cyto_dl.image.io.skimage_reader.SkimageReader.get_data"]], "read() (cyto_dl.image.io.skimage_reader.skimagereader method)": [[43, "cyto_dl.image.io.skimage_reader.SkimageReader.read"]], "verify_suffix() (cyto_dl.image.io.skimage_reader.skimagereader method)": [[43, "cyto_dl.image.io.skimage_reader.SkimageReader.verify_suffix"]], "cyto_dl.image.transforms": [[44, "module-cyto_dl.image.transforms"]], "addmeta (class in cyto_dl.image.transforms.add_meta)": [[45, "cyto_dl.image.transforms.add_meta.AddMeta"]], "cyto_dl.image.transforms.add_meta": [[45, "module-cyto_dl.image.transforms.add_meta"]], "brightsampler (class in cyto_dl.image.transforms.bright_sampler)": [[46, "cyto_dl.image.transforms.bright_sampler.BrightSampler"]], "cyto_dl.image.transforms.bright_sampler": [[46, "module-cyto_dl.image.transforms.bright_sampler"]], "clip (class in cyto_dl.image.transforms.clip)": [[47, "cyto_dl.image.transforms.clip.Clip"]], "clipd (class in cyto_dl.image.transforms.clip)": [[47, "cyto_dl.image.transforms.clip.Clipd"]], "cyto_dl.image.transforms.clip": [[47, "module-cyto_dl.image.transforms.clip"]], "contrastadjust (class in cyto_dl.image.transforms.contrastadjust)": [[48, "cyto_dl.image.transforms.contrastadjust.ContrastAdjust"]], "contrastadjustd (class in cyto_dl.image.transforms.contrastadjust)": [[48, "cyto_dl.image.transforms.contrastadjust.ContrastAdjustd"]], "cyto_dl.image.transforms.contrastadjust": [[48, "module-cyto_dl.image.transforms.contrastadjust"]], "jepamaskgenerator (class in cyto_dl.image.transforms.generate_jepa_masks)": [[49, "cyto_dl.image.transforms.generate_jepa_masks.JEPAMaskGenerator"]], "cyto_dl.image.transforms.generate_jepa_masks": [[49, "module-cyto_dl.image.transforms.generate_jepa_masks"]], "remove_excess_pixels() (cyto_dl.image.transforms.generate_jepa_masks.jepamaskgenerator method)": [[49, "cyto_dl.image.transforms.generate_jepa_masks.JEPAMaskGenerator.remove_excess_pixels"]], "merged (class in cyto_dl.image.transforms.merge)": [[50, "cyto_dl.image.transforms.merge.Merged"]], "cyto_dl.image.transforms.merge": [[50, "module-cyto_dl.image.transforms.merge"]], "randommultiscalecropd (class in cyto_dl.image.transforms.multiscale_cropper)": [[51, "cyto_dl.image.transforms.multiscale_cropper.RandomMultiScaleCropd"]], "cyto_dl.image.transforms.multiscale_cropper": [[51, "module-cyto_dl.image.transforms.multiscale_cropper"]], "generate_slices() (cyto_dl.image.transforms.multiscale_cropper.randommultiscalecropd method)": [[51, "cyto_dl.image.transforms.multiscale_cropper.RandomMultiScaleCropd.generate_slices"]], "padzd (class in cyto_dl.image.transforms.pad)": [[52, "cyto_dl.image.transforms.pad.PadZd"]], "cyto_dl.image.transforms.pad": [[52, "module-cyto_dl.image.transforms.pad"]], "maxprojectd (class in cyto_dl.image.transforms.project)": [[53, "cyto_dl.image.transforms.project.MaxProjectd"]], "cyto_dl.image.transforms.project": [[53, "module-cyto_dl.image.transforms.project"]], "save (class in cyto_dl.image.transforms.save)": [[55, "cyto_dl.image.transforms.save.Save"]], "saved (class in cyto_dl.image.transforms.save)": [[55, "cyto_dl.image.transforms.save.Saved"]], "cyto_dl.image.transforms.save": [[55, "module-cyto_dl.image.transforms.save"]], "cropresize (class in cyto_dl.image.transforms.track_transforms)": [[57, "cyto_dl.image.transforms.track_transforms.CropResize"]], "generatetracklabels (class in cyto_dl.image.transforms.track_transforms)": [[57, "cyto_dl.image.transforms.track_transforms.GenerateTrackLabels"]], "perchannel (class in cyto_dl.image.transforms.track_transforms)": [[57, "cyto_dl.image.transforms.track_transforms.PerChannel"]], "splittrackd (class in cyto_dl.image.transforms.track_transforms)": [[57, "cyto_dl.image.transforms.track_transforms.SplitTrackd"]], "cyto_dl.image.transforms.track_transforms": [[57, "module-cyto_dl.image.transforms.track_transforms"]], "cyto_dl.loggers": [[58, "module-cyto_dl.loggers"]], "mlflowlogger (class in cyto_dl.loggers.mlflow)": [[59, "cyto_dl.loggers.mlflow.MLFlowLogger"]], "after_save_checkpoint() (cyto_dl.loggers.mlflow.mlflowlogger method)": [[59, "cyto_dl.loggers.mlflow.MLFlowLogger.after_save_checkpoint"]], "cyto_dl.loggers.mlflow": [[59, "module-cyto_dl.loggers.mlflow"]], "log_hyperparams() (cyto_dl.loggers.mlflow.mlflowlogger method)": [[59, "cyto_dl.loggers.mlflow.MLFlowLogger.log_hyperparams"]], "log_metrics() (cyto_dl.loggers.mlflow.mlflowlogger method)": [[59, "cyto_dl.loggers.mlflow.MLFlowLogger.log_metrics"]], "cyto_dl.models": [[60, "module-cyto_dl.models"]], "basemodel (class in cyto_dl.models.base_model)": [[61, "cyto_dl.models.base_model.BaseModel"]], "basemodelmeta (class in cyto_dl.models.base_model)": [[61, "cyto_dl.models.base_model.BaseModelMeta"]], "compute_metrics() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.compute_metrics"]], "configure_optimizers() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.configure_optimizers"]], "cyto_dl.models.base_model": [[61, "module-cyto_dl.models.base_model"]], "forward() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.forward"]], "model_step() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.model_step"]], "on_train_start() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.on_train_start"]], "parse_batch() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.parse_batch"]], "predict_step() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.predict_step"]], "test_step() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.test_step"]], "training_step() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.training_step"]], "validation_step() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.validation_step"]], "basicmodel (class in cyto_dl.models.basic_model)": [[62, "cyto_dl.models.basic_model.BasicModel"]], "cyto_dl.models.basic_model": [[62, "module-cyto_dl.models.basic_model"]], "forward() (cyto_dl.models.basic_model.basicmodel method)": [[62, "cyto_dl.models.basic_model.BasicModel.forward"]], "model_step() (cyto_dl.models.basic_model.basicmodel method)": [[62, "cyto_dl.models.basic_model.BasicModel.model_step"]], "cyto_dl.models.classification": [[63, "module-cyto_dl.models.classification"]], "classification (class in cyto_dl.models.classification.classification)": [[64, "cyto_dl.models.classification.classification.Classification"]], "cyto_dl.models.classification.classification": [[64, "module-cyto_dl.models.classification.classification"]], "forward() (cyto_dl.models.classification.classification.classification method)": [[64, "cyto_dl.models.classification.classification.Classification.forward"]], "model_step() (cyto_dl.models.classification.classification.classification method)": [[64, "cyto_dl.models.classification.classification.Classification.model_step"]], "predict_step() (cyto_dl.models.classification.classification.classification method)": [[64, "cyto_dl.models.classification.classification.Classification.predict_step"]], "save_images() (cyto_dl.models.classification.classification.classification method)": [[64, "cyto_dl.models.classification.classification.Classification.save_images"]], "should_save_image() (cyto_dl.models.classification.classification.classification method)": [[64, "cyto_dl.models.classification.classification.Classification.should_save_image"]], "timepointclassification (class in cyto_dl.models.classification.timepoint_classification)": [[65, "cyto_dl.models.classification.timepoint_classification.TimepointClassification"]], "cyto_dl.models.classification.timepoint_classification": [[65, "module-cyto_dl.models.classification.timepoint_classification"]], "predict_step() (cyto_dl.models.classification.timepoint_classification.timepointclassification method)": [[65, "cyto_dl.models.classification.timepoint_classification.TimepointClassification.predict_step"]], "cyto_dl.models.contrastive": [[66, "module-cyto_dl.models.contrastive"]], "contrastive (class in cyto_dl.models.contrastive.contrastive)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive"]], "cyto_dl.models.contrastive.contrastive": [[67, "module-cyto_dl.models.contrastive.contrastive"]], "forward() (cyto_dl.models.contrastive.contrastive.contrastive method)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive.forward"]], "model_step() (cyto_dl.models.contrastive.contrastive.contrastive method)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive.model_step"]], "plot_classes() (cyto_dl.models.contrastive.contrastive.contrastive method)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive.plot_classes"]], "plot_neighbors() (cyto_dl.models.contrastive.contrastive.contrastive method)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive.plot_neighbors"]], "predict_step() (cyto_dl.models.contrastive.contrastive.contrastive method)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive.predict_step"]], "cyto_dl.models.im2im": [[72, "module-cyto_dl.models.im2im"]], "gan (class in cyto_dl.models.im2im.gan)": [[73, "cyto_dl.models.im2im.gan.GAN"]], "configure_optimizers() (cyto_dl.models.im2im.gan.gan method)": [[73, "cyto_dl.models.im2im.gan.GAN.configure_optimizers"]], "cyto_dl.models.im2im.gan": [[73, "module-cyto_dl.models.im2im.gan"]], "model_step() (cyto_dl.models.im2im.gan.gan method)": [[73, "cyto_dl.models.im2im.gan.GAN.model_step"]], "predict_step() (cyto_dl.models.im2im.gan.gan method)": [[73, "cyto_dl.models.im2im.gan.GAN.predict_step"]], "multitaskim2im (class in cyto_dl.models.im2im.multi_task)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im"]], "configure_optimizers() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.configure_optimizers"]], "cyto_dl.models.im2im.multi_task": [[74, "module-cyto_dl.models.im2im.multi_task"]], "forward() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.forward"]], "get_n_postprocess_image() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.get_n_postprocess_image"]], "get_per_head() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.get_per_head"]], "model_step() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.model_step"]], "predict_step() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.predict_step"]], "run_forward() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.run_forward"]], "test_step() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.test_step"]], "training_step() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.training_step"]], "validation_step() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.validation_step"]], "cyto_dl.models.im2im.utils": [[75, "module-cyto_dl.models.im2im.utils"]], "instancesegcluster (class in cyto_dl.models.im2im.utils.instance_seg)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegCluster"]], "instancesegloss (class in cyto_dl.models.im2im.utils.instance_seg)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegLoss"]], "instancesegpreprocessd (class in cyto_dl.models.im2im.utils.instance_seg)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd"]], "instancesegrandflipd (class in cyto_dl.models.im2im.utils.instance_seg)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegRandFlipd"]], "cluster_object() (cyto_dl.models.im2im.utils.instance_seg.instancesegcluster method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegCluster.cluster_object"]], "cyto_dl.models.im2im.utils.instance_seg": [[76, "module-cyto_dl.models.im2im.utils.instance_seg"]], "embed_from_skel() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.embed_from_skel"]], "kd_clustering() (cyto_dl.models.im2im.utils.instance_seg.instancesegcluster method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegCluster.kd_clustering"]], "keep_largest_cc() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.keep_largest_cc"]], "label_2d() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.label_2d"]], "pad_slice() (in module cyto_dl.models.im2im.utils.instance_seg)": [[76, "cyto_dl.models.im2im.utils.instance_seg.pad_slice"]], "remove_small_skeletons() (cyto_dl.models.im2im.utils.instance_seg.instancesegcluster method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegCluster.remove_small_skeletons"]], "shrink() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.shrink"]], "skeleton_tall() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.skeleton_tall"]], "smooth_embedding() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.smooth_embedding"]], "topology_preserving_thinning() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.topology_preserving_thinning"]], "noiseannealer (class in cyto_dl.models.im2im.utils.noise_annealer)": [[77, "cyto_dl.models.im2im.utils.noise_annealer.NoiseAnnealer"]], "cyto_dl.models.im2im.utils.noise_annealer": [[77, "module-cyto_dl.models.im2im.utils.noise_annealer"]], "update_noise() (cyto_dl.models.im2im.utils.noise_annealer.noiseannealer method)": [[77, "cyto_dl.models.im2im.utils.noise_annealer.NoiseAnnealer.update_noise"]], "cyto_dl.models.im2im.utils.postprocessing": [[78, "module-cyto_dl.models.im2im.utils.postprocessing"]], "actthreshlabel (class in cyto_dl.models.im2im.utils.postprocessing.act_thresh_label)": [[79, "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label.ActThreshLabel"]], "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label": [[79, "module-cyto_dl.models.im2im.utils.postprocessing.act_thresh_label"]], "autothreshold (class in cyto_dl.models.im2im.utils.postprocessing.auto_thresh)": [[80, "cyto_dl.models.im2im.utils.postprocessing.auto_thresh.AutoThreshold"]], "cyto_dl.models.im2im.utils.postprocessing.auto_thresh": [[80, "module-cyto_dl.models.im2im.utils.postprocessing.auto_thresh"]], "dicttoim (class in cyto_dl.models.im2im.utils.postprocessing.dict_to_im)": [[81, "cyto_dl.models.im2im.utils.postprocessing.dict_to_im.DictToIm"]], "cyto_dl.models.im2im.utils.postprocessing.dict_to_im": [[81, "module-cyto_dl.models.im2im.utils.postprocessing.dict_to_im"]], "detach() (in module cyto_dl.models.im2im.utils.postprocessing.dict_to_im)": [[81, "cyto_dl.models.im2im.utils.postprocessing.dict_to_im.detach"]], "cyto_dl.models.jepa": [[82, "module-cyto_dl.models.jepa"]], "ijepa (class in cyto_dl.models.jepa.ijepa)": [[83, "cyto_dl.models.jepa.ijepa.IJEPA"]], "cyto_dl.models.jepa.ijepa": [[83, "module-cyto_dl.models.jepa.ijepa"]], "model_step() (cyto_dl.models.jepa.ijepa.ijepa method)": [[83, "cyto_dl.models.jepa.ijepa.IJEPA.model_step"]], "iwm (class in cyto_dl.models.jepa.iwm)": [[84, "cyto_dl.models.jepa.iwm.IWM"]], "cyto_dl.models.jepa.iwm": [[84, "module-cyto_dl.models.jepa.iwm"]], "extract_embeddings() (cyto_dl.models.jepa.iwm.iwm method)": [[84, "cyto_dl.models.jepa.iwm.IWM.extract_embeddings"]], "get_predict_masks() (cyto_dl.models.jepa.iwm.iwm method)": [[84, "cyto_dl.models.jepa.iwm.IWM.get_predict_masks"]], "model_step() (cyto_dl.models.jepa.iwm.iwm method)": [[84, "cyto_dl.models.jepa.iwm.IWM.model_step"]], "predict_step() (cyto_dl.models.jepa.iwm.iwm method)": [[84, "cyto_dl.models.jepa.iwm.IWM.predict_step"]], "test_step() (cyto_dl.models.jepa.iwm.iwm method)": [[84, "cyto_dl.models.jepa.iwm.IWM.test_step"]], "jepabase (class in cyto_dl.models.jepa.jepa_base)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase"]], "configure_optimizers() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.configure_optimizers"]], "cyto_dl.models.jepa.jepa_base": [[85, "module-cyto_dl.models.jepa.jepa_base"]], "forward() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.forward"]], "get_context_embeddings() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.get_context_embeddings"]], "get_mask() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.get_mask"]], "get_target_embeddings() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.get_target_embeddings"]], "model_step() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.model_step"]], "predict_step() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.predict_step"]], "remove_first_dim() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.remove_first_dim"]], "update_teacher() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.update_teacher"]], "cyto_dl.models.utils": [[86, "module-cyto_dl.models.utils"]], "cyto_dl.models.utils.mlflow": [[87, "module-cyto_dl.models.utils.mlflow"]], "get_config() (in module cyto_dl.models.utils.mlflow)": [[87, "cyto_dl.models.utils.mlflow.get_config"]], "load_model_from_checkpoint() (in module cyto_dl.models.utils.mlflow)": [[87, "cyto_dl.models.utils.mlflow.load_model_from_checkpoint"]], "cyto_dl.models.utils.utils": [[88, "module-cyto_dl.models.utils.utils"]], "find_indices() (in module cyto_dl.models.utils.utils)": [[88, "cyto_dl.models.utils.utils.find_indices"]], "basevae (class in cyto_dl.models.vae.base_vae)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE"]], "calculate_elbo() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.calculate_elbo"]], "calculate_rcl() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.calculate_rcl"]], "calculate_rcl_dict() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.calculate_rcl_dict"]], "cyto_dl.models.vae.base_vae": [[90, "module-cyto_dl.models.vae.base_vae"]], "decode() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.decode"]], "encode() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.encode"]], "forward() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.forward"]], "model_step() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.model_step"]], "sample_z() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.sample_z"]], "cyto_dl.models.vae.priors": [[99, "module-cyto_dl.models.vae.priors"]], "prior (class in cyto_dl.models.vae.priors.abstract_prior)": [[100, "cyto_dl.models.vae.priors.abstract_prior.Prior"]], "cyto_dl.models.vae.priors.abstract_prior": [[100, "module-cyto_dl.models.vae.priors.abstract_prior"]], "forward() (cyto_dl.models.vae.priors.abstract_prior.prior method)": [[100, "cyto_dl.models.vae.priors.abstract_prior.Prior.forward"]], "kl_divergence() (cyto_dl.models.vae.priors.abstract_prior.prior method)": [[100, "cyto_dl.models.vae.priors.abstract_prior.Prior.kl_divergence"]], "param_size (cyto_dl.models.vae.priors.abstract_prior.prior property)": [[100, "cyto_dl.models.vae.priors.abstract_prior.Prior.param_size"]], "sample() (cyto_dl.models.vae.priors.abstract_prior.prior method)": [[100, "cyto_dl.models.vae.priors.abstract_prior.Prior.sample"]], "diagonalgaussianprior (class in cyto_dl.models.vae.priors.gaussian)": [[101, "cyto_dl.models.vae.priors.gaussian.DiagonalGaussianPrior"]], "isotropicgaussianprior (class in cyto_dl.models.vae.priors.gaussian)": [[101, "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior"]], "compute_tc_penalty() (in module cyto_dl.models.vae.priors.gaussian)": [[101, "cyto_dl.models.vae.priors.gaussian.compute_tc_penalty"]], "cyto_dl.models.vae.priors.gaussian": [[101, "module-cyto_dl.models.vae.priors.gaussian"]], "forward() (cyto_dl.models.vae.priors.gaussian.diagonalgaussianprior method)": [[101, "cyto_dl.models.vae.priors.gaussian.DiagonalGaussianPrior.forward"]], "forward() (cyto_dl.models.vae.priors.gaussian.isotropicgaussianprior method)": [[101, "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior.forward"]], "kl_divergence() (cyto_dl.models.vae.priors.gaussian.diagonalgaussianprior class method)": [[101, "cyto_dl.models.vae.priors.gaussian.DiagonalGaussianPrior.kl_divergence"]], "kl_divergence() (cyto_dl.models.vae.priors.gaussian.isotropicgaussianprior class method)": [[101, "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior.kl_divergence"]], "param_size (cyto_dl.models.vae.priors.gaussian.diagonalgaussianprior property)": [[101, "cyto_dl.models.vae.priors.gaussian.DiagonalGaussianPrior.param_size"]], "param_size (cyto_dl.models.vae.priors.gaussian.isotropicgaussianprior property)": [[101, "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior.param_size"]], "sample() (cyto_dl.models.vae.priors.gaussian.isotropicgaussianprior class method)": [[101, "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior.sample"]], "identityprior (class in cyto_dl.models.vae.priors.identity_prior)": [[102, "cyto_dl.models.vae.priors.identity_prior.IdentityPrior"]], "cyto_dl.models.vae.priors.identity_prior": [[102, "module-cyto_dl.models.vae.priors.identity_prior"]], "forward() (cyto_dl.models.vae.priors.identity_prior.identityprior method)": [[102, "cyto_dl.models.vae.priors.identity_prior.IdentityPrior.forward"]], "param_size (cyto_dl.models.vae.priors.identity_prior.identityprior property)": [[102, "cyto_dl.models.vae.priors.identity_prior.IdentityPrior.param_size"]], "jointprior (class in cyto_dl.models.vae.priors.joint_prior)": [[103, "cyto_dl.models.vae.priors.joint_prior.JointPrior"]], "cyto_dl.models.vae.priors.joint_prior": [[103, "module-cyto_dl.models.vae.priors.joint_prior"]], "forward() (cyto_dl.models.vae.priors.joint_prior.jointprior method)": [[103, "cyto_dl.models.vae.priors.joint_prior.JointPrior.forward"]], "kl_divergence() (cyto_dl.models.vae.priors.joint_prior.jointprior method)": [[103, "cyto_dl.models.vae.priors.joint_prior.JointPrior.kl_divergence"]], "param_size (cyto_dl.models.vae.priors.joint_prior.jointprior property)": [[103, "cyto_dl.models.vae.priors.joint_prior.JointPrior.param_size"]], "sample() (cyto_dl.models.vae.priors.joint_prior.jointprior method)": [[103, "cyto_dl.models.vae.priors.joint_prior.JointPrior.sample"]], "cyto_dl.nn": [[105, "module-cyto_dl.nn"]], "cyto_dl.nn.discriminators": [[106, "module-cyto_dl.nn.discriminators"]], "multiscalediscriminator (class in cyto_dl.nn.discriminators.multi_scale_discriminator)": [[107, "cyto_dl.nn.discriminators.multi_scale_discriminator.MultiScaleDiscriminator"]], "cyto_dl.nn.discriminators.multi_scale_discriminator": [[107, "module-cyto_dl.nn.discriminators.multi_scale_discriminator"]], "forward() (cyto_dl.nn.discriminators.multi_scale_discriminator.multiscalediscriminator method)": [[107, "cyto_dl.nn.discriminators.multi_scale_discriminator.MultiScaleDiscriminator.forward"]], "nlayerdiscriminator (class in cyto_dl.nn.discriminators.n_layer_discriminator)": [[108, "cyto_dl.nn.discriminators.n_layer_discriminator.NLayerDiscriminator"]], "cyto_dl.nn.discriminators.n_layer_discriminator": [[108, "module-cyto_dl.nn.discriminators.n_layer_discriminator"]], "forward() (cyto_dl.nn.discriminators.n_layer_discriminator.nlayerdiscriminator method)": [[108, "cyto_dl.nn.discriminators.n_layer_discriminator.NLayerDiscriminator.forward"]], "cyto_dl.nn.head": [[109, "module-cyto_dl.nn.head"]], "basehead (class in cyto_dl.nn.head.base_head)": [[110, "cyto_dl.nn.head.base_head.BaseHead"]], "cyto_dl.nn.head.base_head": [[110, "module-cyto_dl.nn.head.base_head"]], "forward() (cyto_dl.nn.head.base_head.basehead method)": [[110, "cyto_dl.nn.head.base_head.BaseHead.forward"]], "generate_io_map() (cyto_dl.nn.head.base_head.basehead method)": [[110, "cyto_dl.nn.head.base_head.BaseHead.generate_io_map"]], "run_head() (cyto_dl.nn.head.base_head.basehead method)": [[110, "cyto_dl.nn.head.base_head.BaseHead.run_head"]], "update_params() (cyto_dl.nn.head.base_head.basehead method)": [[110, "cyto_dl.nn.head.base_head.BaseHead.update_params"]], "ganhead (class in cyto_dl.nn.head.gan_head)": [[111, "cyto_dl.nn.head.gan_head.GANHead"]], "cyto_dl.nn.head.gan_head": [[111, "module-cyto_dl.nn.head.gan_head"]], "forward() (cyto_dl.nn.head.gan_head.ganhead method)": [[111, "cyto_dl.nn.head.gan_head.GANHead.forward"]], "run_head() (cyto_dl.nn.head.gan_head.ganhead method)": [[111, "cyto_dl.nn.head.gan_head.GANHead.run_head"]], "ganhead_resize (class in cyto_dl.nn.head.gan_head_superres)": [[112, "cyto_dl.nn.head.gan_head_superres.GANHead_resize"]], "cyto_dl.nn.head.gan_head_superres": [[112, "module-cyto_dl.nn.head.gan_head_superres"]], "forward() (cyto_dl.nn.head.gan_head_superres.ganhead_resize method)": [[112, "cyto_dl.nn.head.gan_head_superres.GANHead_resize.forward"]], "maehead (class in cyto_dl.nn.head.mae_head)": [[113, "cyto_dl.nn.head.mae_head.MAEHead"]], "cyto_dl.nn.head.mae_head": [[113, "module-cyto_dl.nn.head.mae_head"]], "run_head() (cyto_dl.nn.head.mae_head.maehead method)": [[113, "cyto_dl.nn.head.mae_head.MAEHead.run_head"]], "maskhead (class in cyto_dl.nn.head.mask_head)": [[114, "cyto_dl.nn.head.mask_head.MaskHead"]], "cyto_dl.nn.head.mask_head": [[114, "module-cyto_dl.nn.head.mask_head"]], "run_head() (cyto_dl.nn.head.mask_head.maskhead method)": [[114, "cyto_dl.nn.head.mask_head.MaskHead.run_head"]], "resblockshead (class in cyto_dl.nn.head.res_blocks_head)": [[115, "cyto_dl.nn.head.res_blocks_head.ResBlocksHead"]], "cyto_dl.nn.head.res_blocks_head": [[115, "module-cyto_dl.nn.head.res_blocks_head"]], "forward() (cyto_dl.nn.head.res_blocks_head.resblockshead method)": [[115, "cyto_dl.nn.head.res_blocks_head.ResBlocksHead.forward"]], "vicreghead (class in cyto_dl.nn.head.vic_reg)": [[116, "cyto_dl.nn.head.vic_reg.VICRegHead"]], "cyto_dl.nn.head.vic_reg": [[116, "module-cyto_dl.nn.head.vic_reg"]], "run_head() (cyto_dl.nn.head.vic_reg.vicreghead method)": [[116, "cyto_dl.nn.head.vic_reg.VICRegHead.run_head"]], "hrskip (class in cyto_dl.nn.hr_skip)": [[117, "cyto_dl.nn.hr_skip.HRSkip"]], "cyto_dl.nn.hr_skip": [[117, "module-cyto_dl.nn.hr_skip"]], "forward() (cyto_dl.nn.hr_skip.hrskip method)": [[117, "cyto_dl.nn.hr_skip.HRSkip.forward"]], "cyto_dl.nn.losses": [[118, "module-cyto_dl.nn.losses"]], "adversarialloss (class in cyto_dl.nn.losses.adversarial_loss)": [[119, "cyto_dl.nn.losses.adversarial_loss.AdversarialLoss"]], "cyto_dl.nn.losses.adversarial_loss": [[119, "module-cyto_dl.nn.losses.adversarial_loss"]], "forward() (cyto_dl.nn.losses.adversarial_loss.adversarialloss method)": [[119, "cyto_dl.nn.losses.adversarial_loss.AdversarialLoss.forward"]], "chamferloss (class in cyto_dl.nn.losses.chamfer_loss)": [[120, "cyto_dl.nn.losses.chamfer_loss.ChamferLoss"]], "batch_pairwise_dist() (cyto_dl.nn.losses.chamfer_loss.chamferloss method)": [[120, "cyto_dl.nn.losses.chamfer_loss.ChamferLoss.batch_pairwise_dist"]], "batch_pairwise_dist2() (cyto_dl.nn.losses.chamfer_loss.chamferloss method)": [[120, "cyto_dl.nn.losses.chamfer_loss.ChamferLoss.batch_pairwise_dist2"]], "cyto_dl.nn.losses.chamfer_loss": [[120, "module-cyto_dl.nn.losses.chamfer_loss"]], "forward() (cyto_dl.nn.losses.chamfer_loss.chamferloss method)": [[120, "cyto_dl.nn.losses.chamfer_loss.ChamferLoss.forward"]], "cblogloss (class in cyto_dl.nn.losses.continuous_bernoulli)": [[121, "cyto_dl.nn.losses.continuous_bernoulli.CBLogLoss"]], "cyto_dl.nn.losses.continuous_bernoulli": [[121, "module-cyto_dl.nn.losses.continuous_bernoulli"]], "forward() (cyto_dl.nn.losses.continuous_bernoulli.cblogloss method)": [[121, "cyto_dl.nn.losses.continuous_bernoulli.CBLogLoss.forward"]], "cosineloss (class in cyto_dl.nn.losses.cosine_loss)": [[122, "cyto_dl.nn.losses.cosine_loss.CosineLoss"]], "cyto_dl.nn.losses.cosine_loss": [[122, "module-cyto_dl.nn.losses.cosine_loss"]], "forward() (cyto_dl.nn.losses.cosine_loss.cosineloss method)": [[122, "cyto_dl.nn.losses.cosine_loss.CosineLoss.forward"]], "ganloss (class in cyto_dl.nn.losses.gan_loss)": [[123, "cyto_dl.nn.losses.gan_loss.GANLoss"]], "pix2pixhd (class in cyto_dl.nn.losses.gan_loss)": [[123, "cyto_dl.nn.losses.gan_loss.Pix2PixHD"]], "cyto_dl.nn.losses.gan_loss": [[123, "module-cyto_dl.nn.losses.gan_loss"]], "get_feature_matching_loss() (cyto_dl.nn.losses.gan_loss.pix2pixhd method)": [[123, "cyto_dl.nn.losses.gan_loss.Pix2PixHD.get_feature_matching_loss"]], "get_gan_loss() (cyto_dl.nn.losses.gan_loss.pix2pixhd method)": [[123, "cyto_dl.nn.losses.gan_loss.Pix2PixHD.get_gan_loss"]], "get_target_tensor() (cyto_dl.nn.losses.gan_loss.ganloss method)": [[123, "cyto_dl.nn.losses.gan_loss.GANLoss.get_target_tensor"]], "gaussiannllloss (class in cyto_dl.nn.losses.gaussian_nll_loss)": [[124, "cyto_dl.nn.losses.gaussian_nll_loss.GaussianNLLLoss"]], "cyto_dl.nn.losses.gaussian_nll_loss": [[124, "module-cyto_dl.nn.losses.gaussian_nll_loss"]], "forward() (cyto_dl.nn.losses.gaussian_nll_loss.gaussiannllloss method)": [[124, "cyto_dl.nn.losses.gaussian_nll_loss.GaussianNLLLoss.forward"]], "cmap_loss (class in cyto_dl.nn.losses.loss_wrapper)": [[126, "cyto_dl.nn.losses.loss_wrapper.CMAP_loss"]], "losswrapper (class in cyto_dl.nn.losses.loss_wrapper)": [[126, "cyto_dl.nn.losses.loss_wrapper.LossWrapper"]], "cyto_dl.nn.losses.loss_wrapper": [[126, "module-cyto_dl.nn.losses.loss_wrapper"]], "thresholdloss (class in cyto_dl.nn.losses.threshold_loss)": [[128, "cyto_dl.nn.losses.threshold_loss.ThresholdLoss"]], "cyto_dl.nn.losses.threshold_loss": [[128, "module-cyto_dl.nn.losses.threshold_loss"]], "vicregloss (class in cyto_dl.nn.losses.vic_reg)": [[129, "cyto_dl.nn.losses.vic_reg.VICRegLoss"]], "cyto_dl.nn.losses.vic_reg": [[129, "module-cyto_dl.nn.losses.vic_reg"]], "forward() (cyto_dl.nn.losses.vic_reg.vicregloss method)": [[129, "cyto_dl.nn.losses.vic_reg.VICRegLoss.forward"]], "off_diagonal() (in module cyto_dl.nn.losses.vic_reg)": [[129, "cyto_dl.nn.losses.vic_reg.off_diagonal"]], "weibulllogloss (class in cyto_dl.nn.losses.weibull)": [[130, "cyto_dl.nn.losses.weibull.WeibullLogLoss"]], "cyto_dl.nn.losses.weibull": [[130, "module-cyto_dl.nn.losses.weibull"]], "forward() (cyto_dl.nn.losses.weibull.weibulllogloss method)": [[130, "cyto_dl.nn.losses.weibull.WeibullLogLoss.forward"]], "weibull_log_probs() (in module cyto_dl.nn.losses.weibull)": [[130, "cyto_dl.nn.losses.weibull.weibull_log_probs"]], "weightedmseloss (class in cyto_dl.nn.losses.weighted_mse_loss)": [[131, "cyto_dl.nn.losses.weighted_mse_loss.WeightedMSELoss"]], "cyto_dl.nn.losses.weighted_mse_loss": [[131, "module-cyto_dl.nn.losses.weighted_mse_loss"]], "forward() (cyto_dl.nn.losses.weighted_mse_loss.weightedmseloss method)": [[131, "cyto_dl.nn.losses.weighted_mse_loss.WeightedMSELoss.forward"]], "mlp (class in cyto_dl.nn.mlp)": [[132, "cyto_dl.nn.mlp.MLP"]], "cyto_dl.nn.mlp": [[132, "module-cyto_dl.nn.mlp"]], "forward() (cyto_dl.nn.mlp.mlp method)": [[132, "cyto_dl.nn.mlp.MLP.forward"]], "residualunit (class in cyto_dl.nn.res_unit)": [[138, "cyto_dl.nn.res_unit.ResidualUnit"]], "cyto_dl.nn.res_unit": [[138, "module-cyto_dl.nn.res_unit"]], "forward() (cyto_dl.nn.res_unit.residualunit method)": [[138, "cyto_dl.nn.res_unit.ResidualUnit.forward"]], "convpoolrelu (class in cyto_dl.nn.spatial_transformer)": [[139, "cyto_dl.nn.spatial_transformer.ConvPoolReLU"]], "stn (class in cyto_dl.nn.spatial_transformer)": [[139, "cyto_dl.nn.spatial_transformer.STN"]], "cyto_dl.nn.spatial_transformer": [[139, "module-cyto_dl.nn.spatial_transformer"]], "forward() (cyto_dl.nn.spatial_transformer.convpoolrelu method)": [[139, "cyto_dl.nn.spatial_transformer.ConvPoolReLU.forward"]], "forward() (cyto_dl.nn.spatial_transformer.stn method)": [[139, "cyto_dl.nn.spatial_transformer.STN.forward"]], "toggle() (cyto_dl.nn.spatial_transformer.stn method)": [[139, "cyto_dl.nn.spatial_transformer.STN.toggle"]], "trackclassifier (class in cyto_dl.nn.track_sequence_predictor)": [[140, "cyto_dl.nn.track_sequence_predictor.TrackClassifier"]], "cyto_dl.nn.track_sequence_predictor": [[140, "module-cyto_dl.nn.track_sequence_predictor"]], "forward() (cyto_dl.nn.track_sequence_predictor.trackclassifier method)": [[140, "cyto_dl.nn.track_sequence_predictor.TrackClassifier.forward"]], "positionalencoding1d() (in module cyto_dl.nn.track_sequence_predictor)": [[140, "cyto_dl.nn.track_sequence_predictor.positionalencoding1d"]], "cyto_dl.nn.vits": [[141, "module-cyto_dl.nn.vits"]], "cyto_dl.nn.vits.blocks": [[142, "module-cyto_dl.nn.vits.blocks"]], "crossattention (class in cyto_dl.nn.vits.blocks.cross_attention)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossAttention"]], "crossattentionblock (class in cyto_dl.nn.vits.blocks.cross_attention)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossAttentionBlock"]], "crossselfblock (class in cyto_dl.nn.vits.blocks.cross_attention)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossSelfBlock"]], "mlp (class in cyto_dl.nn.vits.blocks.cross_attention)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.Mlp"]], "cyto_dl.nn.vits.blocks.cross_attention": [[143, "module-cyto_dl.nn.vits.blocks.cross_attention"]], "forward() (cyto_dl.nn.vits.blocks.cross_attention.crossattention method)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossAttention.forward"]], "forward() (cyto_dl.nn.vits.blocks.cross_attention.crossattentionblock method)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossAttentionBlock.forward"]], "forward() (cyto_dl.nn.vits.blocks.cross_attention.crossselfblock method)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossSelfBlock.forward"]], "forward() (cyto_dl.nn.vits.blocks.cross_attention.mlp method)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.Mlp.forward"]], "intermediateweigher (class in cyto_dl.nn.vits.blocks.intermediate_weigher)": [[144, "cyto_dl.nn.vits.blocks.intermediate_weigher.IntermediateWeigher"]], "cyto_dl.nn.vits.blocks.intermediate_weigher": [[144, "module-cyto_dl.nn.vits.blocks.intermediate_weigher"]], "forward() (cyto_dl.nn.vits.blocks.intermediate_weigher.intermediateweigher method)": [[144, "cyto_dl.nn.vits.blocks.intermediate_weigher.IntermediateWeigher.forward"]], "hierablock (class in cyto_dl.nn.vits.blocks.masked_unit_attention)": [[145, "cyto_dl.nn.vits.blocks.masked_unit_attention.HieraBlock"]], "maskunitattention (class in cyto_dl.nn.vits.blocks.masked_unit_attention)": [[145, "cyto_dl.nn.vits.blocks.masked_unit_attention.MaskUnitAttention"]], "cyto_dl.nn.vits.blocks.masked_unit_attention": [[145, "module-cyto_dl.nn.vits.blocks.masked_unit_attention"]], "forward() (cyto_dl.nn.vits.blocks.masked_unit_attention.hierablock method)": [[145, "cyto_dl.nn.vits.blocks.masked_unit_attention.HieraBlock.forward"]], "forward() (cyto_dl.nn.vits.blocks.masked_unit_attention.maskunitattention method)": [[145, "cyto_dl.nn.vits.blocks.masked_unit_attention.MaskUnitAttention.forward"]], "cyto_dl.nn.vits.blocks.patchify": [[146, "module-cyto_dl.nn.vits.blocks.patchify"]], "patchify (class in cyto_dl.nn.vits.blocks.patchify.patchify)": [[147, "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify"]], "create_img2token() (cyto_dl.nn.vits.blocks.patchify.patchify.patchify method)": [[147, "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify.create_img2token"]], "cyto_dl.nn.vits.blocks.patchify.patchify": [[147, "module-cyto_dl.nn.vits.blocks.patchify.patchify"]], "extract_visible_tokens() (cyto_dl.nn.vits.blocks.patchify.patchify.patchify method)": [[147, "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify.extract_visible_tokens"]], "get_mask_args() (cyto_dl.nn.vits.blocks.patchify.patchify.patchify method)": [[147, "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify.get_mask_args"]], "img2token (cyto_dl.nn.vits.blocks.patchify.patchify.patchify property)": [[147, "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify.img2token"]], "patchifybase (class in cyto_dl.nn.vits.blocks.patchify.patchify_base)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase"]], "create_conv() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.create_conv"]], "create_patch2img() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.create_patch2img"]], "cyto_dl.nn.vits.blocks.patchify.patchify_base": [[148, "module-cyto_dl.nn.vits.blocks.patchify.patchify_base"]], "extract_visible_tokens() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.extract_visible_tokens"]], "forward() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.forward"]], "get_mask() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.get_mask"]], "get_mask_args() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.get_mask_args"]], "img2token (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase property)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.img2token"]], "patchifyhiera (class in cyto_dl.nn.vits.blocks.patchify.patchify_hiera)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera"]], "create_img2token() (cyto_dl.nn.vits.blocks.patchify.patchify_hiera.patchifyhiera method)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera.create_img2token"]], "cyto_dl.nn.vits.blocks.patchify.patchify_hiera": [[149, "module-cyto_dl.nn.vits.blocks.patchify.patchify_hiera"]], "extract_visible_tokens() (cyto_dl.nn.vits.blocks.patchify.patchify_hiera.patchifyhiera method)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera.extract_visible_tokens"]], "get_mask_args() (cyto_dl.nn.vits.blocks.patchify.patchify_hiera.patchifyhiera method)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera.get_mask_args"]], "img2token (cyto_dl.nn.vits.blocks.patchify.patchify_hiera.patchifyhiera property)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera.img2token"]], "take_indexes_mask() (in module cyto_dl.nn.vits.blocks.patchify.patchify_hiera)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.take_indexes_mask"]], "crossmae_decoder (class in cyto_dl.nn.vits.decoder)": [[150, "cyto_dl.nn.vits.decoder.CrossMAE_Decoder"]], "mae_decoder (class in cyto_dl.nn.vits.decoder)": [[150, "cyto_dl.nn.vits.decoder.MAE_Decoder"]], "add_mask_tokens() (cyto_dl.nn.vits.decoder.mae_decoder method)": [[150, "cyto_dl.nn.vits.decoder.MAE_Decoder.add_mask_tokens"]], "adjust_indices_for_cls() (cyto_dl.nn.vits.decoder.mae_decoder method)": [[150, "cyto_dl.nn.vits.decoder.MAE_Decoder.adjust_indices_for_cls"]], "cyto_dl.nn.vits.decoder": [[150, "module-cyto_dl.nn.vits.decoder"]], "forward() (cyto_dl.nn.vits.decoder.crossmae_decoder method)": [[150, "cyto_dl.nn.vits.decoder.CrossMAE_Decoder.forward"]], "forward() (cyto_dl.nn.vits.decoder.mae_decoder method)": [[150, "cyto_dl.nn.vits.decoder.MAE_Decoder.forward"]], "init_weight() (cyto_dl.nn.vits.decoder.mae_decoder method)": [[150, "cyto_dl.nn.vits.decoder.MAE_Decoder.init_weight"]], "hieraencoder (class in cyto_dl.nn.vits.encoder)": [[151, "cyto_dl.nn.vits.encoder.HieraEncoder"]], "jepaencoder (class in cyto_dl.nn.vits.encoder)": [[151, "cyto_dl.nn.vits.encoder.JEPAEncoder"]], "mae_encoder (class in cyto_dl.nn.vits.encoder)": [[151, "cyto_dl.nn.vits.encoder.MAE_Encoder"]], "spatialmerger (class in cyto_dl.nn.vits.encoder)": [[151, "cyto_dl.nn.vits.encoder.SpatialMerger"]], "cyto_dl.nn.vits.encoder": [[151, "module-cyto_dl.nn.vits.encoder"]], "forward() (cyto_dl.nn.vits.encoder.hieraencoder method)": [[151, "cyto_dl.nn.vits.encoder.HieraEncoder.forward"]], "forward() (cyto_dl.nn.vits.encoder.jepaencoder method)": [[151, "cyto_dl.nn.vits.encoder.JEPAEncoder.forward"]], "forward() (cyto_dl.nn.vits.encoder.mae_encoder method)": [[151, "cyto_dl.nn.vits.encoder.MAE_Encoder.forward"]], "forward() (cyto_dl.nn.vits.encoder.spatialmerger method)": [[151, "cyto_dl.nn.vits.encoder.SpatialMerger.forward"]], "init_weight() (cyto_dl.nn.vits.encoder.mae_encoder method)": [[151, "cyto_dl.nn.vits.encoder.MAE_Encoder.init_weight"]], "hieramae (class in cyto_dl.nn.vits.mae)": [[152, "cyto_dl.nn.vits.mae.HieraMAE"]], "mae (class in cyto_dl.nn.vits.mae)": [[152, "cyto_dl.nn.vits.mae.MAE"]], "mae_base (class in cyto_dl.nn.vits.mae)": [[152, "cyto_dl.nn.vits.mae.MAE_Base"]], "cyto_dl.nn.vits.mae": [[152, "module-cyto_dl.nn.vits.mae"]], "decoder (cyto_dl.nn.vits.mae.hieramae property)": [[152, "cyto_dl.nn.vits.mae.HieraMAE.decoder"]], "decoder (cyto_dl.nn.vits.mae.mae property)": [[152, "cyto_dl.nn.vits.mae.MAE.decoder"]], "decoder (cyto_dl.nn.vits.mae.mae_base property)": [[152, "cyto_dl.nn.vits.mae.MAE_Base.decoder"]], "encoder (cyto_dl.nn.vits.mae.hieramae property)": [[152, "cyto_dl.nn.vits.mae.HieraMAE.encoder"]], "encoder (cyto_dl.nn.vits.mae.mae property)": [[152, "cyto_dl.nn.vits.mae.MAE.encoder"]], "encoder (cyto_dl.nn.vits.mae.mae_base property)": [[152, "cyto_dl.nn.vits.mae.MAE_Base.encoder"]], "forward() (cyto_dl.nn.vits.mae.mae_base method)": [[152, "cyto_dl.nn.vits.mae.MAE_Base.forward"]], "init_decoder() (cyto_dl.nn.vits.mae.mae_base method)": [[152, "cyto_dl.nn.vits.mae.MAE_Base.init_decoder"]], "init_encoder() (cyto_dl.nn.vits.mae.mae_base method)": [[152, "cyto_dl.nn.vits.mae.MAE_Base.init_encoder"]], "iwmpredictor (class in cyto_dl.nn.vits.predictor)": [[153, "cyto_dl.nn.vits.predictor.IWMPredictor"]], "jepapredictor (class in cyto_dl.nn.vits.predictor)": [[153, "cyto_dl.nn.vits.predictor.JEPAPredictor"]], "cyto_dl.nn.vits.predictor": [[153, "module-cyto_dl.nn.vits.predictor"]], "forward() (cyto_dl.nn.vits.predictor.iwmpredictor method)": [[153, "cyto_dl.nn.vits.predictor.IWMPredictor.forward"]], "forward() (cyto_dl.nn.vits.predictor.jepapredictor method)": [[153, "cyto_dl.nn.vits.predictor.JEPAPredictor.forward"]], "init_weight() (cyto_dl.nn.vits.predictor.jepapredictor method)": [[153, "cyto_dl.nn.vits.predictor.JEPAPredictor.init_weight"]], "predict_target_features() (cyto_dl.nn.vits.predictor.jepapredictor method)": [[153, "cyto_dl.nn.vits.predictor.JEPAPredictor.predict_target_features"]], "encodedskip (class in cyto_dl.nn.vits.seg)": [[154, "cyto_dl.nn.vits.seg.EncodedSkip"]], "seg_vit (class in cyto_dl.nn.vits.seg)": [[154, "cyto_dl.nn.vits.seg.Seg_ViT"]], "superresdecoder (class in cyto_dl.nn.vits.seg)": [[154, "cyto_dl.nn.vits.seg.SuperresDecoder"]], "cyto_dl.nn.vits.seg": [[154, "module-cyto_dl.nn.vits.seg"]], "forward() (cyto_dl.nn.vits.seg.encodedskip method)": [[154, "cyto_dl.nn.vits.seg.EncodedSkip.forward"]], "forward() (cyto_dl.nn.vits.seg.seg_vit method)": [[154, "cyto_dl.nn.vits.seg.Seg_ViT.forward"]], "forward() (cyto_dl.nn.vits.seg.superresdecoder method)": [[154, "cyto_dl.nn.vits.seg.SuperresDecoder.forward"]], "cyto_dl.nn.vits.utils": [[155, "module-cyto_dl.nn.vits.utils"]], "get_positional_embedding() (in module cyto_dl.nn.vits.utils)": [[155, "cyto_dl.nn.vits.utils.get_positional_embedding"]], "match_tuple_dimensions() (in module cyto_dl.nn.vits.utils)": [[155, "cyto_dl.nn.vits.utils.match_tuple_dimensions"]], "random_indexes() (in module cyto_dl.nn.vits.utils)": [[155, "cyto_dl.nn.vits.utils.random_indexes"]], "take_indexes() (in module cyto_dl.nn.vits.utils)": [[155, "cyto_dl.nn.vits.utils.take_indexes"]], "cyto_dl.point_cloud": [[156, "module-cyto_dl.point_cloud"]], "cyto_dl.train": [[159, "module-cyto_dl.train"]], "main() (in module cyto_dl.train)": [[159, "cyto_dl.train.main"]], "cyto_dl.utils": [[160, "module-cyto_dl.utils"]], "create_dataloader() (in module cyto_dl.utils.array)": [[161, "cyto_dl.utils.array.create_dataloader"]], "cyto_dl.utils.array": [[161, "module-cyto_dl.utils.array"]], "extract_array_predictions() (in module cyto_dl.utils.array)": [[161, "cyto_dl.utils.array.extract_array_predictions"]], "cyto_dl.utils.checkpoint": [[162, "module-cyto_dl.utils.checkpoint"]], "load_checkpoint() (in module cyto_dl.utils.checkpoint)": [[162, "cyto_dl.utils.checkpoint.load_checkpoint"]], "cyto_dl.utils.config": [[163, "module-cyto_dl.utils.config"]], "is_config() (in module cyto_dl.utils.config)": [[163, "cyto_dl.utils.config.is_config"]], "kv_to_dict() (in module cyto_dl.utils.config)": [[163, "cyto_dl.utils.config.kv_to_dict"]], "remove_aux_key() (in module cyto_dl.utils.config)": [[163, "cyto_dl.utils.config.remove_aux_key"]], "cyto_dl.utils.download_test_data": [[164, "module-cyto_dl.utils.download_test_data"]], "delete_test_data() (in module cyto_dl.utils.download_test_data)": [[164, "cyto_dl.utils.download_test_data.delete_test_data"]], "download_test_data() (in module cyto_dl.utils.download_test_data)": [[164, "cyto_dl.utils.download_test_data.download_test_data"]], "parse_s3_path() (in module cyto_dl.utils.download_test_data)": [[164, "cyto_dl.utils.download_test_data.parse_s3_path"]], "setup_paths() (in module cyto_dl.utils.download_test_data)": [[164, "cyto_dl.utils.download_test_data.setup_paths"]], "dummydatamodule (class in cyto_dl.utils.dummy_dataset)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule"]], "dummydataset (class in cyto_dl.utils.dummy_dataset)": [[165, "cyto_dl.utils.dummy_dataset.DummyDataset"]], "cyto_dl.utils.dummy_dataset": [[165, "module-cyto_dl.utils.dummy_dataset"]], "generate_img() (cyto_dl.utils.dummy_dataset.dummydataset method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDataset.generate_img"]], "get_dataloader() (cyto_dl.utils.dummy_dataset.dummydatamodule method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule.get_dataloader"]], "predict_dataloader() (cyto_dl.utils.dummy_dataset.dummydatamodule method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule.predict_dataloader"]], "test_dataloader() (cyto_dl.utils.dummy_dataset.dummydatamodule method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule.test_dataloader"]], "train_dataloader() (cyto_dl.utils.dummy_dataset.dummydatamodule method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule.train_dataloader"]], "val_dataloader() (cyto_dl.utils.dummy_dataset.dummydatamodule method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule.val_dataloader"]], "cyto_dl.utils.pylogger": [[166, "module-cyto_dl.utils.pylogger"]], "get_pylogger() (in module cyto_dl.utils.pylogger)": [[166, "cyto_dl.utils.pylogger.get_pylogger"]], "cyto_dl.utils.rich_utils": [[167, "module-cyto_dl.utils.rich_utils"]], "enforce_tags() (in module cyto_dl.utils.rich_utils)": [[167, "cyto_dl.utils.rich_utils.enforce_tags"]], "print_config_tree() (in module cyto_dl.utils.rich_utils)": [[167, "cyto_dl.utils.rich_utils.print_config_tree"]], "rotationmodule (class in cyto_dl.utils.rotation)": [[168, "cyto_dl.utils.rotation.RotationModule"]], "compute_rotation_matrix() (cyto_dl.utils.rotation.rotationmodule method)": [[168, "cyto_dl.utils.rotation.RotationModule.compute_rotation_matrix"]], "cyto_dl.utils.rotation": [[168, "module-cyto_dl.utils.rotation"]], "get_rotation_matrix_so2() (in module cyto_dl.utils.rotation)": [[168, "cyto_dl.utils.rotation.get_rotation_matrix_so2"]], "get_rotation_matrix_so3() (in module cyto_dl.utils.rotation)": [[168, "cyto_dl.utils.rotation.get_rotation_matrix_so3"]], "close_loggers() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.close_loggers"]], "cyto_dl.utils.template_utils": [[172, "module-cyto_dl.utils.template_utils"]], "extras() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.extras"]], "get_metric_value() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.get_metric_value"]], "instantiate_callbacks() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.instantiate_callbacks"]], "instantiate_loggers() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.instantiate_loggers"]], "log_hyperparameters() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.log_hyperparameters"]], "save_file() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.save_file"]], "task_wrapper() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.task_wrapper"]]}})
\ No newline at end of file
+Search.setIndex({"docnames": ["api", "cyto_dl", "cyto_dl.api", "cyto_dl.api.cyto_dl_model", "cyto_dl.api.cyto_dl_model.cyto_dl_base_model", "cyto_dl.api.cyto_dl_model.segmentation_plugin_model", "cyto_dl.api.data", "cyto_dl.api.model", "cyto_dl.callbacks", "cyto_dl.callbacks.callback_utils", "cyto_dl.callbacks.csv_saver", "cyto_dl.callbacks.image_saver", "cyto_dl.callbacks.latent_walk", "cyto_dl.callbacks.layer_freeze", "cyto_dl.callbacks.model_utils", "cyto_dl.callbacks.outlier_detection", "cyto_dl.compile", "cyto_dl.dataframe", "cyto_dl.dataframe.readers", "cyto_dl.dataframe.transforms", "cyto_dl.dataframe.transforms.filter", "cyto_dl.dataframe.transforms.group_cols", "cyto_dl.dataframe.transforms.misc", "cyto_dl.dataframe.transforms.split", "cyto_dl.datamodules", "cyto_dl.datamodules.array", "cyto_dl.datamodules.data_dict", "cyto_dl.datamodules.dataframe", "cyto_dl.datamodules.dataframe.dataframe_datamodule", "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule", "cyto_dl.datamodules.dataframe.utils", "cyto_dl.datamodules.folder", "cyto_dl.datamodules.multidim_image", "cyto_dl.datamodules.smartcache", "cyto_dl.datamodules.torchvision", "cyto_dl.eval", "cyto_dl.image", "cyto_dl.image.io", "cyto_dl.image.io.aicsimage_loader", "cyto_dl.image.io.monai_bio_reader", "cyto_dl.image.io.numpy_reader", "cyto_dl.image.io.ome_zarr_reader", "cyto_dl.image.io.polygon_loader", "cyto_dl.image.io.skimage_reader", "cyto_dl.image.transforms", "cyto_dl.image.transforms.add_meta", "cyto_dl.image.transforms.bright_sampler", "cyto_dl.image.transforms.clip", "cyto_dl.image.transforms.contrastadjust", "cyto_dl.image.transforms.generate_jepa_masks", "cyto_dl.image.transforms.merge", "cyto_dl.image.transforms.multiscale_cropper", "cyto_dl.image.transforms.pad", "cyto_dl.image.transforms.project", "cyto_dl.image.transforms.rotation_mask_transform", "cyto_dl.image.transforms.save", "cyto_dl.image.transforms.so2_random_rotation", "cyto_dl.image.transforms.track_transforms", "cyto_dl.loggers", "cyto_dl.loggers.mlflow", "cyto_dl.models", "cyto_dl.models.base_model", "cyto_dl.models.basic_model", "cyto_dl.models.classification", "cyto_dl.models.classification.classification", "cyto_dl.models.classification.timepoint_classification", "cyto_dl.models.contrastive", "cyto_dl.models.contrastive.contrastive", "cyto_dl.models.handlers", "cyto_dl.models.handlers.base_handler", "cyto_dl.models.handlers.image_handler", "cyto_dl.models.handlers.load_image_patch", "cyto_dl.models.im2im", "cyto_dl.models.im2im.gan", "cyto_dl.models.im2im.multi_task", "cyto_dl.models.im2im.utils", "cyto_dl.models.im2im.utils.instance_seg", "cyto_dl.models.im2im.utils.noise_annealer", "cyto_dl.models.im2im.utils.postprocessing", "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label", "cyto_dl.models.im2im.utils.postprocessing.auto_thresh", "cyto_dl.models.im2im.utils.postprocessing.dict_to_im", "cyto_dl.models.jepa", "cyto_dl.models.jepa.ijepa", "cyto_dl.models.jepa.iwm", "cyto_dl.models.jepa.jepa_base", "cyto_dl.models.utils", "cyto_dl.models.utils.mlflow", "cyto_dl.models.utils.utils", "cyto_dl.models.vae", "cyto_dl.models.vae.base_vae", "cyto_dl.models.vae.image_canon_vae", "cyto_dl.models.vae.image_encoder", "cyto_dl.models.vae.image_vae", "cyto_dl.models.vae.latent_loss_vae", "cyto_dl.models.vae.o2_spharm_vae", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_encoder", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_vae", "cyto_dl.models.vae.point_cloud_vae", "cyto_dl.models.vae.priors", "cyto_dl.models.vae.priors.abstract_prior", "cyto_dl.models.vae.priors.gaussian", "cyto_dl.models.vae.priors.identity_prior", "cyto_dl.models.vae.priors.joint_prior", "cyto_dl.models.vae.tabular_vae", "cyto_dl.nn", "cyto_dl.nn.discriminators", "cyto_dl.nn.discriminators.multi_scale_discriminator", "cyto_dl.nn.discriminators.n_layer_discriminator", "cyto_dl.nn.head", "cyto_dl.nn.head.base_head", "cyto_dl.nn.head.gan_head", "cyto_dl.nn.head.gan_head_superres", "cyto_dl.nn.head.mae_head", "cyto_dl.nn.head.mask_head", "cyto_dl.nn.head.res_blocks_head", "cyto_dl.nn.head.vic_reg", "cyto_dl.nn.hr_skip", "cyto_dl.nn.losses", "cyto_dl.nn.losses.adversarial_loss", "cyto_dl.nn.losses.chamfer_loss", "cyto_dl.nn.losses.continuous_bernoulli", "cyto_dl.nn.losses.cosine_loss", "cyto_dl.nn.losses.gan_loss", "cyto_dl.nn.losses.gaussian_nll_loss", "cyto_dl.nn.losses.geomloss", "cyto_dl.nn.losses.loss_wrapper", "cyto_dl.nn.losses.spharm_loss", "cyto_dl.nn.losses.threshold_loss", "cyto_dl.nn.losses.vic_reg", "cyto_dl.nn.losses.weibull", "cyto_dl.nn.losses.weighted_mse_loss", "cyto_dl.nn.mlp", "cyto_dl.nn.point_cloud", "cyto_dl.nn.point_cloud.dgcnn", "cyto_dl.nn.point_cloud.folding_net", "cyto_dl.nn.point_cloud.graph_functions", "cyto_dl.nn.point_cloud.vnn", "cyto_dl.nn.res_unit", "cyto_dl.nn.spatial_transformer", "cyto_dl.nn.track_sequence_predictor", "cyto_dl.nn.vits", "cyto_dl.nn.vits.blocks", "cyto_dl.nn.vits.blocks.cross_attention", "cyto_dl.nn.vits.blocks.intermediate_weigher", "cyto_dl.nn.vits.blocks.masked_unit_attention", "cyto_dl.nn.vits.blocks.patchify", "cyto_dl.nn.vits.blocks.patchify.patchify", "cyto_dl.nn.vits.blocks.patchify.patchify_base", "cyto_dl.nn.vits.blocks.patchify.patchify_hiera", "cyto_dl.nn.vits.decoder", "cyto_dl.nn.vits.encoder", "cyto_dl.nn.vits.mae", "cyto_dl.nn.vits.predictor", "cyto_dl.nn.vits.seg", "cyto_dl.nn.vits.utils", "cyto_dl.point_cloud", "cyto_dl.point_cloud.io", "cyto_dl.point_cloud.io.read_pcloud", "cyto_dl.train", "cyto_dl.utils", "cyto_dl.utils.array", "cyto_dl.utils.checkpoint", "cyto_dl.utils.config", "cyto_dl.utils.download_test_data", "cyto_dl.utils.dummy_dataset", "cyto_dl.utils.pylogger", "cyto_dl.utils.rich_utils", "cyto_dl.utils.rotation", "cyto_dl.utils.spharm", "cyto_dl.utils.spharm.reconstruction", "cyto_dl.utils.spharm.rotation", "cyto_dl.utils.template_utils", "index", "installation", "modules", "overview", "using_examples"], "filenames": ["api.rst", "cyto_dl.rst", "cyto_dl.api.rst", "cyto_dl.api.cyto_dl_model.rst", "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.rst", "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.rst", "cyto_dl.api.data.rst", "cyto_dl.api.model.rst", "cyto_dl.callbacks.rst", "cyto_dl.callbacks.callback_utils.rst", "cyto_dl.callbacks.csv_saver.rst", "cyto_dl.callbacks.image_saver.rst", "cyto_dl.callbacks.latent_walk.rst", "cyto_dl.callbacks.layer_freeze.rst", "cyto_dl.callbacks.model_utils.rst", "cyto_dl.callbacks.outlier_detection.rst", "cyto_dl.compile.rst", "cyto_dl.dataframe.rst", "cyto_dl.dataframe.readers.rst", "cyto_dl.dataframe.transforms.rst", "cyto_dl.dataframe.transforms.filter.rst", "cyto_dl.dataframe.transforms.group_cols.rst", "cyto_dl.dataframe.transforms.misc.rst", "cyto_dl.dataframe.transforms.split.rst", "cyto_dl.datamodules.rst", "cyto_dl.datamodules.array.rst", "cyto_dl.datamodules.data_dict.rst", "cyto_dl.datamodules.dataframe.rst", "cyto_dl.datamodules.dataframe.dataframe_datamodule.rst", "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule.rst", "cyto_dl.datamodules.dataframe.utils.rst", "cyto_dl.datamodules.folder.rst", "cyto_dl.datamodules.multidim_image.rst", "cyto_dl.datamodules.smartcache.rst", "cyto_dl.datamodules.torchvision.rst", "cyto_dl.eval.rst", "cyto_dl.image.rst", "cyto_dl.image.io.rst", "cyto_dl.image.io.aicsimage_loader.rst", "cyto_dl.image.io.monai_bio_reader.rst", "cyto_dl.image.io.numpy_reader.rst", "cyto_dl.image.io.ome_zarr_reader.rst", "cyto_dl.image.io.polygon_loader.rst", "cyto_dl.image.io.skimage_reader.rst", "cyto_dl.image.transforms.rst", "cyto_dl.image.transforms.add_meta.rst", "cyto_dl.image.transforms.bright_sampler.rst", "cyto_dl.image.transforms.clip.rst", "cyto_dl.image.transforms.contrastadjust.rst", "cyto_dl.image.transforms.generate_jepa_masks.rst", "cyto_dl.image.transforms.merge.rst", "cyto_dl.image.transforms.multiscale_cropper.rst", "cyto_dl.image.transforms.pad.rst", "cyto_dl.image.transforms.project.rst", "cyto_dl.image.transforms.rotation_mask_transform.rst", "cyto_dl.image.transforms.save.rst", "cyto_dl.image.transforms.so2_random_rotation.rst", "cyto_dl.image.transforms.track_transforms.rst", "cyto_dl.loggers.rst", "cyto_dl.loggers.mlflow.rst", "cyto_dl.models.rst", "cyto_dl.models.base_model.rst", "cyto_dl.models.basic_model.rst", "cyto_dl.models.classification.rst", "cyto_dl.models.classification.classification.rst", "cyto_dl.models.classification.timepoint_classification.rst", "cyto_dl.models.contrastive.rst", "cyto_dl.models.contrastive.contrastive.rst", "cyto_dl.models.handlers.rst", "cyto_dl.models.handlers.base_handler.rst", "cyto_dl.models.handlers.image_handler.rst", "cyto_dl.models.handlers.load_image_patch.rst", "cyto_dl.models.im2im.rst", "cyto_dl.models.im2im.gan.rst", "cyto_dl.models.im2im.multi_task.rst", "cyto_dl.models.im2im.utils.rst", "cyto_dl.models.im2im.utils.instance_seg.rst", "cyto_dl.models.im2im.utils.noise_annealer.rst", "cyto_dl.models.im2im.utils.postprocessing.rst", "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label.rst", "cyto_dl.models.im2im.utils.postprocessing.auto_thresh.rst", "cyto_dl.models.im2im.utils.postprocessing.dict_to_im.rst", "cyto_dl.models.jepa.rst", "cyto_dl.models.jepa.ijepa.rst", "cyto_dl.models.jepa.iwm.rst", "cyto_dl.models.jepa.jepa_base.rst", "cyto_dl.models.utils.rst", "cyto_dl.models.utils.mlflow.rst", "cyto_dl.models.utils.utils.rst", "cyto_dl.models.vae.rst", "cyto_dl.models.vae.base_vae.rst", "cyto_dl.models.vae.image_canon_vae.rst", "cyto_dl.models.vae.image_encoder.rst", "cyto_dl.models.vae.image_vae.rst", "cyto_dl.models.vae.latent_loss_vae.rst", "cyto_dl.models.vae.o2_spharm_vae.rst", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_encoder.rst", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_vae.rst", "cyto_dl.models.vae.point_cloud_vae.rst", "cyto_dl.models.vae.priors.rst", "cyto_dl.models.vae.priors.abstract_prior.rst", "cyto_dl.models.vae.priors.gaussian.rst", "cyto_dl.models.vae.priors.identity_prior.rst", "cyto_dl.models.vae.priors.joint_prior.rst", "cyto_dl.models.vae.tabular_vae.rst", "cyto_dl.nn.rst", "cyto_dl.nn.discriminators.rst", "cyto_dl.nn.discriminators.multi_scale_discriminator.rst", "cyto_dl.nn.discriminators.n_layer_discriminator.rst", "cyto_dl.nn.head.rst", "cyto_dl.nn.head.base_head.rst", "cyto_dl.nn.head.gan_head.rst", "cyto_dl.nn.head.gan_head_superres.rst", "cyto_dl.nn.head.mae_head.rst", "cyto_dl.nn.head.mask_head.rst", "cyto_dl.nn.head.res_blocks_head.rst", "cyto_dl.nn.head.vic_reg.rst", "cyto_dl.nn.hr_skip.rst", "cyto_dl.nn.losses.rst", "cyto_dl.nn.losses.adversarial_loss.rst", "cyto_dl.nn.losses.chamfer_loss.rst", "cyto_dl.nn.losses.continuous_bernoulli.rst", "cyto_dl.nn.losses.cosine_loss.rst", "cyto_dl.nn.losses.gan_loss.rst", "cyto_dl.nn.losses.gaussian_nll_loss.rst", "cyto_dl.nn.losses.geomloss.rst", "cyto_dl.nn.losses.loss_wrapper.rst", "cyto_dl.nn.losses.spharm_loss.rst", "cyto_dl.nn.losses.threshold_loss.rst", "cyto_dl.nn.losses.vic_reg.rst", "cyto_dl.nn.losses.weibull.rst", "cyto_dl.nn.losses.weighted_mse_loss.rst", "cyto_dl.nn.mlp.rst", "cyto_dl.nn.point_cloud.rst", "cyto_dl.nn.point_cloud.dgcnn.rst", "cyto_dl.nn.point_cloud.folding_net.rst", "cyto_dl.nn.point_cloud.graph_functions.rst", "cyto_dl.nn.point_cloud.vnn.rst", "cyto_dl.nn.res_unit.rst", "cyto_dl.nn.spatial_transformer.rst", "cyto_dl.nn.track_sequence_predictor.rst", "cyto_dl.nn.vits.rst", "cyto_dl.nn.vits.blocks.rst", "cyto_dl.nn.vits.blocks.cross_attention.rst", "cyto_dl.nn.vits.blocks.intermediate_weigher.rst", "cyto_dl.nn.vits.blocks.masked_unit_attention.rst", "cyto_dl.nn.vits.blocks.patchify.rst", "cyto_dl.nn.vits.blocks.patchify.patchify.rst", "cyto_dl.nn.vits.blocks.patchify.patchify_base.rst", "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.rst", "cyto_dl.nn.vits.decoder.rst", "cyto_dl.nn.vits.encoder.rst", "cyto_dl.nn.vits.mae.rst", "cyto_dl.nn.vits.predictor.rst", "cyto_dl.nn.vits.seg.rst", "cyto_dl.nn.vits.utils.rst", "cyto_dl.point_cloud.rst", "cyto_dl.point_cloud.io.rst", "cyto_dl.point_cloud.io.read_pcloud.rst", "cyto_dl.train.rst", "cyto_dl.utils.rst", "cyto_dl.utils.array.rst", "cyto_dl.utils.checkpoint.rst", "cyto_dl.utils.config.rst", "cyto_dl.utils.download_test_data.rst", "cyto_dl.utils.dummy_dataset.rst", "cyto_dl.utils.pylogger.rst", "cyto_dl.utils.rich_utils.rst", "cyto_dl.utils.rotation.rst", "cyto_dl.utils.spharm.rst", "cyto_dl.utils.spharm.reconstruction.rst", "cyto_dl.utils.spharm.rotation.rst", "cyto_dl.utils.template_utils.rst", "index.rst", "installation.rst", "modules.rst", "overview.md", "using_examples.rst"], "titles": ["Using the API", "cyto_dl package", "cyto_dl.api package", "cyto_dl.api.cyto_dl_model package", "cyto_dl.api.cyto_dl_model.cyto_dl_base_model module", "cyto_dl.api.cyto_dl_model.segmentation_plugin_model module", "cyto_dl.api.data module", "cyto_dl.api.model module", "cyto_dl.callbacks package", "cyto_dl.callbacks.callback_utils module", "cyto_dl.callbacks.csv_saver module", "cyto_dl.callbacks.image_saver module", "cyto_dl.callbacks.latent_walk module", "cyto_dl.callbacks.layer_freeze module", "cyto_dl.callbacks.model_utils module", "cyto_dl.callbacks.outlier_detection module", "cyto_dl.compile module", "cyto_dl.dataframe package", "cyto_dl.dataframe.readers module", "cyto_dl.dataframe.transforms package", "cyto_dl.dataframe.transforms.filter module", "cyto_dl.dataframe.transforms.group_cols module", "cyto_dl.dataframe.transforms.misc module", "cyto_dl.dataframe.transforms.split module", "cyto_dl.datamodules package", "cyto_dl.datamodules.array module", "cyto_dl.datamodules.data_dict module", "cyto_dl.datamodules.dataframe package", "cyto_dl.datamodules.dataframe.dataframe_datamodule module", "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule module", "cyto_dl.datamodules.dataframe.utils module", "cyto_dl.datamodules.folder module", "cyto_dl.datamodules.multidim_image module", "cyto_dl.datamodules.smartcache module", "cyto_dl.datamodules.torchvision module", "cyto_dl.eval module", "cyto_dl.image package", "cyto_dl.image.io package", "cyto_dl.image.io.aicsimage_loader module", "cyto_dl.image.io.monai_bio_reader module", "cyto_dl.image.io.numpy_reader module", "cyto_dl.image.io.ome_zarr_reader module", "cyto_dl.image.io.polygon_loader module", "cyto_dl.image.io.skimage_reader module", "cyto_dl.image.transforms package", "cyto_dl.image.transforms.add_meta module", "cyto_dl.image.transforms.bright_sampler module", "cyto_dl.image.transforms.clip module", "cyto_dl.image.transforms.contrastadjust module", "cyto_dl.image.transforms.generate_jepa_masks module", "cyto_dl.image.transforms.merge module", "cyto_dl.image.transforms.multiscale_cropper module", "cyto_dl.image.transforms.pad module", "cyto_dl.image.transforms.project module", "cyto_dl.image.transforms.rotation_mask_transform module", "cyto_dl.image.transforms.save module", "cyto_dl.image.transforms.so2_random_rotation module", "cyto_dl.image.transforms.track_transforms module", "cyto_dl.loggers package", "cyto_dl.loggers.mlflow module", "cyto_dl.models package", "cyto_dl.models.base_model module", "cyto_dl.models.basic_model module", "cyto_dl.models.classification package", "cyto_dl.models.classification.classification module", "cyto_dl.models.classification.timepoint_classification module", "cyto_dl.models.contrastive package", "cyto_dl.models.contrastive.contrastive module", "cyto_dl.models.handlers package", "cyto_dl.models.handlers.base_handler module", "cyto_dl.models.handlers.image_handler module", "cyto_dl.models.handlers.load_image_patch module", "cyto_dl.models.im2im package", "cyto_dl.models.im2im.gan module", "cyto_dl.models.im2im.multi_task module", "cyto_dl.models.im2im.utils package", "cyto_dl.models.im2im.utils.instance_seg module", "cyto_dl.models.im2im.utils.noise_annealer module", "cyto_dl.models.im2im.utils.postprocessing package", "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label module", "cyto_dl.models.im2im.utils.postprocessing.auto_thresh module", "cyto_dl.models.im2im.utils.postprocessing.dict_to_im module", "cyto_dl.models.jepa package", "cyto_dl.models.jepa.ijepa module", "cyto_dl.models.jepa.iwm module", "cyto_dl.models.jepa.jepa_base module", "cyto_dl.models.utils package", "cyto_dl.models.utils.mlflow module", "cyto_dl.models.utils.utils module", "cyto_dl.models.vae package", "cyto_dl.models.vae.base_vae module", "cyto_dl.models.vae.image_canon_vae module", "cyto_dl.models.vae.image_encoder module", "cyto_dl.models.vae.image_vae module", "cyto_dl.models.vae.latent_loss_vae module", "cyto_dl.models.vae.o2_spharm_vae package", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_encoder module", "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_vae module", "cyto_dl.models.vae.point_cloud_vae module", "cyto_dl.models.vae.priors package", "cyto_dl.models.vae.priors.abstract_prior module", "cyto_dl.models.vae.priors.gaussian module", "cyto_dl.models.vae.priors.identity_prior module", "cyto_dl.models.vae.priors.joint_prior module", "cyto_dl.models.vae.tabular_vae module", "cyto_dl.nn package", "cyto_dl.nn.discriminators package", "cyto_dl.nn.discriminators.multi_scale_discriminator module", "cyto_dl.nn.discriminators.n_layer_discriminator module", "cyto_dl.nn.head package", "cyto_dl.nn.head.base_head module", "cyto_dl.nn.head.gan_head module", "cyto_dl.nn.head.gan_head_superres module", "cyto_dl.nn.head.mae_head module", "cyto_dl.nn.head.mask_head module", "cyto_dl.nn.head.res_blocks_head module", "cyto_dl.nn.head.vic_reg module", "cyto_dl.nn.hr_skip module", "cyto_dl.nn.losses package", "cyto_dl.nn.losses.adversarial_loss module", "cyto_dl.nn.losses.chamfer_loss module", "cyto_dl.nn.losses.continuous_bernoulli module", "cyto_dl.nn.losses.cosine_loss module", "cyto_dl.nn.losses.gan_loss module", "cyto_dl.nn.losses.gaussian_nll_loss module", "cyto_dl.nn.losses.geomloss module", "cyto_dl.nn.losses.loss_wrapper module", "cyto_dl.nn.losses.spharm_loss module", "cyto_dl.nn.losses.threshold_loss module", "cyto_dl.nn.losses.vic_reg module", "cyto_dl.nn.losses.weibull module", "cyto_dl.nn.losses.weighted_mse_loss module", "cyto_dl.nn.mlp module", "cyto_dl.nn.point_cloud package", "cyto_dl.nn.point_cloud.dgcnn module", "cyto_dl.nn.point_cloud.folding_net module", "cyto_dl.nn.point_cloud.graph_functions module", "cyto_dl.nn.point_cloud.vnn module", "cyto_dl.nn.res_unit module", "cyto_dl.nn.spatial_transformer module", "cyto_dl.nn.track_sequence_predictor module", "cyto_dl.nn.vits package", "cyto_dl.nn.vits.blocks package", "cyto_dl.nn.vits.blocks.cross_attention module", "cyto_dl.nn.vits.blocks.intermediate_weigher module", "cyto_dl.nn.vits.blocks.masked_unit_attention module", "cyto_dl.nn.vits.blocks.patchify package", "cyto_dl.nn.vits.blocks.patchify.patchify module", "cyto_dl.nn.vits.blocks.patchify.patchify_base module", "cyto_dl.nn.vits.blocks.patchify.patchify_hiera module", "cyto_dl.nn.vits.decoder module", "cyto_dl.nn.vits.encoder module", "cyto_dl.nn.vits.mae module", "cyto_dl.nn.vits.predictor module", "cyto_dl.nn.vits.seg module", "cyto_dl.nn.vits.utils module", "cyto_dl.point_cloud package", "cyto_dl.point_cloud.io package", "cyto_dl.point_cloud.io.read_pcloud module", "cyto_dl.train module", "cyto_dl.utils package", "cyto_dl.utils.array module", "cyto_dl.utils.checkpoint module", "cyto_dl.utils.config module", "cyto_dl.utils.download_test_data module", "cyto_dl.utils.dummy_dataset module", "cyto_dl.utils.pylogger module", "cyto_dl.utils.rich_utils module", "cyto_dl.utils.rotation module", "cyto_dl.utils.spharm package", "cyto_dl.utils.spharm.reconstruction module", "cyto_dl.utils.spharm.rotation module", "cyto_dl.utils.template_utils module", "Overview", "Installation", "cyto_dl", "Overview", "Modifying the Example Configs"], "terms": {"cytodl": [0, 5, 173, 176], "can": [0, 18, 20, 21, 28, 29, 31, 38, 40, 51, 52, 77, 90, 153, 155, 173, 176, 177], "import": [0, 138, 173, 176, 177], "python": [0, 21, 166, 172, 174], "packag": [0, 175], "programmat": 0, "cyto_dl": [0, 173, 176, 177], "cytodlmodel": [0, 1, 2, 7], "class": [0, 4, 5, 6, 7, 10, 11, 13, 14, 15, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 34, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 49, 50, 51, 52, 53, 55, 57, 59, 61, 62, 64, 65, 67, 73, 74, 76, 77, 79, 80, 81, 83, 84, 85, 90, 100, 101, 102, 103, 107, 108, 110, 111, 112, 113, 114, 115, 116, 117, 119, 120, 121, 122, 123, 124, 126, 128, 129, 130, 131, 132, 138, 139, 140, 143, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 165, 168], "addit": [0, 25, 26, 28, 29, 31, 33, 39, 61, 64, 67, 73, 74, 83, 84, 85, 90, 173, 176, 177], "command": [0, 166, 167, 172], "line": [0, 166, 167, 172, 177], "interfac": [0, 90], "we": [0, 173, 176, 177], "provid": [0, 7, 22, 32, 47, 48, 76, 126, 151, 167, 173, 176, 177], "configur": [0, 4, 5, 7, 167, 177], "file": [0, 4, 7, 18, 28, 29, 31, 32, 33, 40, 110, 172, 173, 176, 177], "model": [0, 1, 2, 4, 51, 115, 140, 161, 162, 167, 172, 175, 177], "make": [0, 14, 77, 172, 177], "from": [0, 4, 5, 7, 14, 18, 25, 26, 28, 29, 30, 31, 32, 38, 39, 43, 45, 46, 49, 51, 57, 62, 77, 81, 112, 115, 120, 123, 138, 143, 153, 161, 167, 168, 172, 177], "dictionari": [0, 7, 21, 25, 26, 28, 29, 30, 31, 32, 38, 40, 45, 47, 48, 51, 61, 67, 76, 81, 90, 151, 152, 161, 177], "cfg": [0, 4, 5, 35, 159, 163, 167, 172], "data": [0, 1, 2, 7, 22, 23, 25, 26, 28, 29, 30, 31, 33, 38, 39, 41, 43, 45, 79, 83, 84, 85, 115, 161, 167, 177], "load_config_from_dict": [0, 2, 7], "yaml": [0, 4], "load_config_from_yaml": 0, "path": [0, 4, 7, 11, 18, 26, 28, 29, 31, 32, 33, 38, 40, 42, 55, 62, 87, 145, 154, 167, 172, 173, 176, 177], "name": [0, 4, 13, 18, 21, 28, 29, 30, 31, 46, 47, 48, 50, 51, 52, 64, 153, 166, 173, 176, 177], "one": [0, 22, 28, 29, 30, 61, 172, 177], "our": [0, 177], "default": [0, 4, 5, 23, 28, 29, 41, 74, 79, 83, 84, 85, 138, 145, 154, 173, 176, 177], "avail": [0, 29, 30], "option": [0, 13, 18, 20, 21, 23, 26, 28, 29, 31, 32, 42, 49, 61, 62, 67, 74, 76, 83, 84, 85, 90, 115, 145, 154, 161, 167, 172, 174, 177], "ar": [0, 18, 20, 21, 22, 23, 25, 26, 28, 29, 31, 32, 38, 39, 42, 47, 48, 49, 61, 64, 73, 74, 76, 108, 115, 123, 148, 150, 151, 152, 153, 167, 172, 173, 176, 177], "gan": [0, 2, 6, 60, 72, 111, 112, 114, 123, 173, 176, 177], "instance_seg": [0, 2, 6, 72, 75], "labelfre": [0, 6, 173, 176, 177], "segmen": 0, "when": [0, 25, 26, 28, 29, 30, 31, 61, 76, 138, 173, 176], "experi": [0, 5, 177], "overrid": [0, 7, 173, 176], "list": [0, 7, 11, 13, 18, 20, 22, 25, 28, 29, 31, 32, 38, 40, 42, 43, 45, 50, 51, 53, 57, 62, 67, 74, 76, 79, 81, 90, 116, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 163, 172], "string": [0, 13, 18, 20, 21, 31, 61, 138], "format": [0, 38, 57, 177], "same": [0, 21, 57, 62, 90, 115, 123, 148, 150, 151, 152, 153, 155, 177], "wai": [0, 173, 176], "cli": 0, "load_default_experi": [0, 2, 7], "batch_siz": [0, 25, 26, 28, 29, 30, 31, 84, 165, 173, 176, 177], "16": [0, 6, 152, 154], "onc": [0, 177], "i": [0, 4, 18, 20, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 42, 49, 51, 52, 53, 61, 62, 76, 79, 81, 83, 84, 85, 90, 115, 123, 126, 138, 144, 150, 151, 152, 153, 154, 155, 163, 165, 168, 172, 173, 176, 177], "attribut": 0, "overridden": [0, 177], "override_config": [0, 2, 7], "which": [0, 7, 20, 21, 25, 26, 28, 29, 31, 32, 40, 61, 77, 107, 168, 172, 173, 176, 177], "take": [0, 18, 51, 53], "valu": [0, 6, 20, 21, 22, 23, 25, 28, 32, 38, 47, 48, 49, 79, 81, 83, 84, 85, 128, 138, 143, 145, 163, 165, 172, 173, 176, 177], "made": [0, 28, 61, 177], "async": 0, "pass": [0, 25, 26, 28, 29, 31, 32, 33, 38, 39, 51, 61, 64, 67, 73, 74, 77, 83, 84, 85, 90, 107, 173, 176, 177], "run_async": [0, 7], "true": [0, 7, 23, 28, 30, 32, 38, 39, 42, 45, 47, 51, 59, 76, 87, 90, 110, 111, 113, 114, 116, 128, 138, 145, 147, 148, 150, 151, 152, 153, 154, 155, 173, 176], "await": 0, "The": [0, 20, 23, 28, 29, 32, 49, 61, 62, 83, 84, 85, 123, 138, 168, 173, 176, 177], "To": [0, 173, 176], "examin": [0, 46], "print_config": [0, 2, 7], "download": 0, "exampl": [0, 32, 51, 61, 77, 115, 138, 151, 173, 176], "download_example_data": [0, 2, 7], "thi": [0, 4, 14, 23, 28, 29, 30, 32, 49, 51, 61, 62, 77, 165, 173, 176, 177], "api": [1, 175], "cyto_dl_model": [1, 2], "modul": [1, 2, 3, 8, 17, 19, 24, 27, 36, 37, 44, 58, 60, 63, 66, 68, 72, 75, 78, 82, 86, 89, 95, 99, 105, 106, 109, 118, 133, 141, 142, 146, 156, 157, 160, 169, 175], "autothresholdmethod": [1, 2, 6], "experimenttyp": [1, 2, 6], "hardwaretyp": [1, 2, 5, 6], "patchsiz": [1, 2, 5, 6], "callback": [1, 167, 172, 173, 175, 176], "callback_util": [1, 8], "csv_saver": [1, 8], "csvsaver": [1, 8, 10], "jepasav": [1, 8, 10], "image_sav": [1, 8], "imagesav": [1, 8, 11], "latent_walk": [1, 8], "layer_freez": [1, 8], "layerfreez": [1, 8, 13], "model_util": [1, 8], "getembed": [1, 8, 14], "get_all_embed": [1, 8, 14], "save_predictions_classifi": [1, 8, 14], "outlier_detect": [1, 8], "outlierdetect": [1, 8, 15], "datafram": [1, 24, 33, 173, 175, 176], "transform": [1, 17, 25, 26, 28, 29, 30, 31, 32, 33, 36, 38, 42, 43, 76, 79, 115, 140, 150, 151, 152, 153, 173, 176, 177], "reader": [1, 17], "filter_column": [1, 17, 18, 19, 20], "read_csv": [1, 17, 18], "read_datafram": [1, 17, 18], "read_h5ad": [1, 17, 18], "read_parquet": [1, 17, 18], "datamodul": [1, 14, 173, 175, 176], "arrai": [1, 24, 32, 42, 81, 115, 126, 148, 160, 173, 176], "make_array_dataload": [1, 24, 25], "data_dict": [1, 24], "make_data_dict_dataload": [1, 24, 26], "folder": [1, 24, 167, 173, 176], "make_folder_dataload": [1, 24, 31], "multidim_imag": [1, 24], "multidimimagedataset": [1, 24, 32], "make_multidim_image_dataload": [1, 24, 32], "smartcach": [1, 24], "smartcachedatamodul": [1, 24, 33], "torchvis": [1, 24], "cifar10": [1, 24, 34], "cifar100": [1, 24, 34], "clevrclassif": [1, 24, 34], "caltech101": [1, 24, 34], "caltech256": [1, 24, 34], "celeba": [1, 24, 34], "cityscap": [1, 24, 34], "cococapt": [1, 24, 34], "cocodetect": [1, 24, 34], "country211": [1, 24, 34], "dtd": [1, 24, 34], "datasetfold": [1, 24, 34], "emnist": [1, 24, 34], "eurosat": [1, 24, 34], "fer2013": [1, 24, 34], "fgvcaircraft": [1, 24, 34], "fakedata": [1, 24, 34], "fashionmnist": [1, 24, 34], "flickr30k": [1, 24, 34], "flickr8k": [1, 24, 34], "flowers102": [1, 24, 34], "food101": [1, 24, 34], "gtsrb": [1, 24, 34], "hmdb51": [1, 24, 34], "inaturalist": [1, 24, 34], "imagefold": [1, 24, 34], "imagenet": [1, 24, 34], "kmnist": [1, 24, 34], "kinet": [1, 24, 34], "kitti": [1, 24, 34], "lfwpair": [1, 24, 34], "lfwpeopl": [1, 24, 34], "lsun": [1, 24, 34], "lsunclass": [1, 24, 34], "mnist": [1, 24, 34], "movingmnist": [1, 24, 34], "omniglot": [1, 24, 34], "oxfordiiitpet": [1, 24, 34], "pcam": [1, 24, 34], "phototour": [1, 24, 34], "places365": [1, 24, 34], "qmnist": [1, 24, 34], "renderedsst2": [1, 24, 34], "sbdataset": [1, 24, 34], "sbu": [1, 24, 34], "semeion": [1, 24, 34], "stl10": [1, 24, 34], "sun397": [1, 24, 34], "svhn": [1, 24, 34], "stanfordcar": [1, 24, 34], "ucf101": [1, 24, 34], "usp": [1, 24, 34], "vocdetect": [1, 24, 34], "vocsegment": [1, 24, 34], "visiondataset": [1, 24, 34], "widerfac": [1, 24, 34], "imag": [1, 11, 26, 29, 30, 32, 33, 64, 67, 73, 74, 76, 79, 81, 83, 84, 108, 110, 111, 112, 113, 114, 115, 116, 123, 126, 140, 147, 148, 149, 150, 151, 152, 153, 173, 175, 176, 177], "io": [1, 18, 36, 73, 74, 115, 150, 156, 173, 176], "logger": [1, 166, 167, 172, 173, 175, 176], "mlflow": [1, 58, 60, 86, 173, 176], "mlflowlogg": [1, 58, 59], "classif": [1, 60], "contrast": [1, 48, 60], "handler": [1, 60], "im2im": [1, 60], "jepa": [1, 49, 60, 153], "util": [1, 24, 25, 26, 27, 28, 29, 31, 60, 72, 105, 141, 173, 175, 176, 177], "vae": [1, 60], "base_model": [1, 60], "basemodel": [1, 60, 61, 62, 64, 67, 73, 74, 83, 84, 85, 90], "basemodelmeta": [1, 60, 61], "basic_model": [1, 60], "basicmodel": [1, 60, 62], "nn": [1, 62, 64, 67, 73, 74, 79, 83, 84, 85, 90, 175], "discrimin": [1, 73, 77, 105, 111, 119, 123, 173, 176], "head": [1, 64, 67, 73, 74, 105, 145, 150, 151, 152, 153, 154, 161, 173, 176, 177], "loss": [1, 61, 62, 76, 90, 102, 105, 110, 111, 112, 113, 114, 115, 116, 173, 176, 177], "point_cloud": [1, 105, 175], "vit": [1, 105], "hr_skip": [1, 105], "hrskip": [1, 105, 117], "mlp": [1, 105, 116, 141, 142, 143, 145], "res_unit": [1, 105], "residualunit": [1, 105, 138], "spatial_transform": [1, 105], "convpoolrelu": [1, 105, 139], "stn": [1, 105, 139], "track_sequence_predictor": [1, 105], "trackclassifi": [1, 105, 140], "positionalencoding1d": [1, 105, 140], "spharm": [1, 160], "create_dataload": [1, 160, 161], "extract_array_predict": [1, 160, 161], "checkpoint": [1, 4, 15, 87, 154, 160], "load_checkpoint": [1, 160, 162], "config": [1, 4, 7, 160, 161, 167, 172, 173, 176], "is_config": [1, 160, 163], "kv_to_dict": [1, 160, 163], "remove_aux_kei": [1, 160, 163], "download_test_data": [1, 160], "delete_test_data": [1, 160, 164], "parse_s3_path": [1, 160, 164], "setup_path": [1, 160, 164], "dummy_dataset": [1, 160], "dummydatamodul": [1, 160, 165], "dummydataset": [1, 160, 165], "pylogg": [1, 160], "get_pylogg": [1, 160, 166], "rich_util": [1, 160], "enforce_tag": [1, 160, 167], "print_config_tre": [1, 160, 167], "rotat": [1, 160, 169, 177], "rotationmodul": [1, 160, 168], "get_rotation_matrix_so2": [1, 160, 168], "get_rotation_matrix_so3": [1, 160, 168], "template_util": [1, 160], "close_logg": [1, 160, 172], "extra": [1, 148, 149, 151, 152, 160, 167, 172, 174], "get_metric_valu": [1, 160, 172], "instantiate_callback": [1, 160, 172], "instantiate_logg": [1, 160, 172], "log_hyperparamet": [1, 160, 172], "save_fil": [1, 160, 172], "task_wrapp": [1, 160, 172], "compil": [1, 64, 73, 74, 173, 175, 176], "eval": [1, 175], "main": [1, 35, 120, 159, 175], "train": [1, 2, 3, 4, 5, 6, 7, 11, 23, 25, 26, 28, 29, 30, 31, 32, 49, 51, 59, 61, 62, 64, 73, 74, 76, 83, 84, 85, 87, 114, 154, 173, 175, 176], "cyto_dl_base_model": [2, 3], "cytodlbasemodel": [2, 3, 4, 5], "segmentation_plugin_model": [2, 3], "segmentationpluginmodel": [2, 3, 5], "isodata": [2, 6], "li": [2, 6], "local": [2, 6], "mean": [2, 6, 46, 76, 101, 119, 121, 122], "minimum": [2, 6, 33], "multiotsu": [2, 6], "niblack": [2, 6], "otsu": [2, 6], "sauvola": [2, 6], "triangl": [2, 6], "yen": [2, 6], "label_fre": [2, 6], "segment": [2, 6, 52, 76, 77, 154, 173, 176, 177], "segmentation_arrai": [2, 6], "segmentation_plugin": [2, 5, 6], "cpu": [2, 6, 61, 81], "gpu": [2, 6, 166, 172, 177], "larg": [2, 6, 33, 177], "medium": [2, 6], "small": [2, 6, 76], "load_config_from_fil": [2, 7], "predict": [2, 3, 4, 5, 7, 10, 25, 28, 29, 32, 61, 62, 64, 67, 83, 84, 85, 110, 111, 112, 113, 114, 115, 123, 126, 153, 161, 173, 176], "save_config": [2, 3, 4, 7], "from_default_config": [3, 4], "from_existing_config": [3, 4], "get_config": [3, 4, 60, 86, 87], "get_experiment_nam": [3, 4], "set_experiment_nam": [3, 4], "get_hardware_typ": [3, 5], "get_input_channel": [3, 5], "get_manifest_column_nam": [3, 5], "get_patch_s": [3, 5], "get_raw_image_channel": [3, 5], "get_split_column": [3, 5], "remove_split_column": [3, 5], "set_hardware_typ": [3, 5], "set_input_channel": [3, 5], "set_manifest_column_nam": [3, 5], "set_patch_s": [3, 5], "set_raw_image_channel": [3, 5], "set_split_column": [3, 5], "dictconfig": [4, 5, 26, 35, 159, 163, 167, 172], "sourc": [4, 5, 6, 7, 10, 11, 13, 14, 15, 18, 20, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 35, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 49, 50, 51, 52, 53, 55, 57, 59, 61, 62, 64, 65, 67, 73, 74, 76, 77, 79, 80, 81, 83, 84, 85, 87, 88, 90, 100, 101, 102, 103, 107, 108, 110, 111, 112, 113, 114, 115, 116, 117, 119, 120, 121, 122, 123, 124, 126, 128, 129, 130, 131, 132, 138, 139, 140, 143, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 159, 161, 162, 163, 164, 165, 166, 167, 168, 172], "base": [4, 5, 6, 7, 10, 11, 13, 14, 15, 21, 22, 26, 28, 29, 30, 31, 32, 33, 34, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 49, 50, 51, 52, 53, 55, 57, 59, 61, 62, 64, 65, 67, 73, 74, 76, 77, 79, 80, 81, 83, 84, 85, 90, 100, 101, 102, 103, 107, 108, 110, 111, 112, 113, 114, 115, 116, 117, 119, 120, 121, 122, 123, 124, 126, 128, 129, 130, 131, 132, 138, 139, 140, 143, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 165, 168, 173, 176, 177], "abc": [4, 79, 110, 148, 152], "A": [4, 5, 18, 20, 22, 26, 28, 29, 31, 62, 123, 138, 168], "us": [4, 5, 14, 18, 20, 21, 22, 23, 25, 26, 28, 29, 31, 32, 33, 38, 39, 47, 48, 49, 50, 51, 52, 61, 62, 64, 67, 73, 74, 76, 77, 83, 84, 85, 90, 110, 112, 114, 115, 123, 138, 148, 150, 151, 152, 153, 165, 167, 173, 174, 176, 177], "run": [4, 32, 74, 110, 113, 114, 116, 173, 176, 177], "cyto": [4, 174], "dl": [4, 174], "Not": [4, 114], "intend": 4, "direct": [4, 139], "client": 4, "pleas": 4, "see": [4, 18, 20, 25, 26, 28, 29, 31, 115, 138, 173, 176], "classmethod": [4, 101], "factori": 4, "method": [4, 61, 80, 115], "instead": [4, 38, 84, 152], "spatial_dim": [4, 32, 33, 49, 112, 115, 138, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 168, 177], "int": [4, 5, 7, 11, 22, 23, 28, 29, 30, 32, 33, 46, 48, 49, 51, 52, 53, 57, 67, 76, 77, 79, 83, 84, 85, 100, 102, 107, 108, 112, 115, 116, 126, 132, 138, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 165, 168], "return": [4, 18, 23, 32, 51, 61, 76, 107, 123, 140, 144, 163, 168], "paramet": [4, 7, 11, 13, 18, 20, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 38, 40, 42, 45, 46, 47, 48, 49, 50, 51, 52, 53, 55, 57, 62, 64, 67, 73, 74, 76, 77, 79, 81, 83, 84, 85, 90, 101, 107, 108, 110, 111, 112, 113, 114, 115, 116, 123, 126, 128, 140, 145, 148, 150, 151, 152, 153, 154, 163, 168, 172, 173, 176, 177], "dimens": [4, 32, 33, 49, 76, 115, 116, 138, 140, 144, 145, 148, 149, 150, 151, 152, 153, 154, 155], "e": [4, 18, 32, 90, 153, 173, 174, 176, 177], "g": [4, 32, 153, 173, 176, 177], "2": [4, 29, 32, 49, 51, 76, 107, 108, 117, 138, 139, 140, 145, 151, 154, 168], "config_filepath": 4, "an": [4, 6, 18, 25, 26, 29, 32, 49, 51, 53, 61, 79, 101, 126, 138, 148, 151, 173, 176, 177], "exist": [4, 23], "basi": 4, "must": [4, 13, 18, 20, 23, 25, 31, 32, 45, 51, 61, 173, 176], "gener": [4, 18, 22, 33, 49, 51, 57, 62, 76, 77, 79, 83, 84, 85, 108, 110, 111, 112, 155, 165, 173, 176], "subclass": [4, 30, 90], "want": [4, 173, 176, 177], "str": [4, 5, 7, 11, 13, 14, 15, 18, 20, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 50, 51, 52, 53, 55, 57, 59, 62, 64, 67, 73, 74, 76, 79, 80, 81, 83, 84, 85, 90, 114, 123, 126, 138, 147, 148, 149, 153, 154, 167, 172], "manifest_path": 4, "output_dir": [4, 7, 14], "none": [4, 5, 7, 11, 13, 14, 18, 20, 21, 22, 23, 26, 28, 29, 30, 31, 32, 33, 35, 41, 43, 51, 52, 59, 61, 62, 64, 74, 76, 79, 80, 90, 101, 108, 110, 111, 112, 113, 114, 115, 116, 117, 124, 131, 132, 138, 143, 147, 148, 149, 150, 151, 152, 153, 154, 159, 161, 165, 167, 172], "max_epoch": [4, 83, 84, 85], "handl": [5, 32, 42, 61, 123], "tupl": [5, 32, 39, 41, 43, 49, 61, 79, 138, 150, 155], "hardware_typ": 5, "input_channel": [5, 147, 148, 149, 151, 152], "target1": 5, "target2": 5, "merge_mask": 5, "exclude_mask": 5, "base_imag": 5, "patch_siz": [5, 140, 147, 148, 149, 150, 151, 152, 177], "channel": [5, 32, 33, 38, 41, 43, 46, 52, 57, 79, 108, 115, 126, 138, 148, 149, 151, 152, 154, 173, 176, 177], "split_column": [5, 28, 29, 30, 173, 176], "enum": 6, "enumer": [6, 33, 38], "threshold_isodata": 6, "threshold_li": 6, "threshold_loc": 6, "threshold_mean": 6, "threshold_minimum": 6, "threshold_multiotsu": 6, "threshold_niblack": 6, "threshold_otsu": 6, "threshold_sauvola": 6, "threshold_triangl": 6, "threshold_yen": 6, "patch": [6, 46, 49, 51, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 177], "size": [6, 30, 49, 51, 76, 123, 138, 148, 149, 150, 151, 152, 154, 155, 177], "respect": [6, 18, 90, 177], "shape": [6, 42, 148, 165], "128": [6, 140], "64": [6, 108, 139, 149, 151, 152], "32": [6, 139, 154], "object": [7, 18, 32, 46, 76, 77, 79, 80, 81, 108, 123, 154, 168], "dict": [7, 20, 21, 25, 26, 28, 29, 32, 39, 41, 43, 51, 52, 59, 62, 73, 74, 76, 83, 84, 85, 90, 151, 152, 165, 172], "load": [7, 18, 26, 28, 29, 31, 32, 33, 38, 39, 40, 62, 173, 176, 177], "config_path": 7, "experiment_typ": 7, "directori": [7, 11, 26, 28, 29, 31, 64, 67, 73, 74, 84, 177], "float": [7, 23, 33, 46, 47, 48, 49, 76, 77, 79, 80, 83, 84, 85, 90, 123, 126, 128, 138, 145, 152, 159, 168, 172], "bool": [7, 11, 20, 23, 28, 29, 30, 32, 38, 39, 40, 41, 42, 43, 45, 47, 48, 51, 52, 53, 55, 57, 76, 79, 81, 84, 112, 115, 123, 126, 128, 138, 145, 147, 148, 150, 151, 152, 153, 154, 155, 167], "fals": [7, 11, 20, 23, 28, 29, 30, 32, 38, 39, 40, 45, 47, 48, 51, 53, 55, 57, 64, 73, 74, 76, 79, 81, 84, 90, 100, 101, 103, 108, 112, 114, 115, 117, 119, 138, 143, 145, 148, 150, 151, 152, 153, 167], "save": [7, 8, 11, 32, 36, 44, 57, 62, 64, 67, 73, 74, 84, 110, 113, 114, 116, 151, 172, 173, 176, 177], "current": [7, 29, 32, 42], "on_predict_epoch_end": [8, 10], "pred_to_csv": [8, 10], "save_feat": [8, 10], "on_predict_batch_end": [8, 11, 15], "on_test_batch_end": [8, 11, 15], "on_train_batch_end": [8, 11], "on_validation_batch_end": [8, 11], "setup": [8, 13, 24, 33, 172], "on_test_epoch_end": [8, 14], "calculate_mahalanobis_hook": [8, 15], "flatten_activ": [8, 15], "on_load_checkpoint": [8, 15], "on_predict_epoch_start": [8, 15], "on_save_checkpoint": [8, 15], "on_test_epoch_start": [8, 15], "on_train_epoch_start": [8, 15], "update_covariance_hook": [8, 15], "save_dir": [10, 11, 15, 59, 64, 67, 73, 74, 84, 177], "meta_kei": [10, 45, 67], "trainer": [10, 11, 13, 14, 15, 167, 173, 176], "pl_modul": [10, 11, 13, 14, 15], "pred": [10, 14, 61, 107, 120], "stage": [10, 11, 13, 33, 61, 62, 64, 67, 73, 74, 83, 84, 85, 90, 110, 111, 113, 114, 116], "save_every_n_epoch": 11, "1": [11, 23, 29, 30, 33, 49, 51, 53, 64, 73, 74, 76, 79, 90, 101, 110, 111, 112, 113, 115, 116, 123, 126, 129, 131, 132, 138, 145, 147, 148, 149, 151, 152, 164, 177], "val": [11, 23, 28, 29, 61, 62, 88], "save_input": 11, "after": [11, 45, 76, 79, 115, 143, 172], "postprocess": [11, 72, 75, 110, 111, 112, 113, 114, 115, 116, 173, 176, 177], "ead": 11, "union": [11, 13, 18, 20, 21, 25, 26, 28, 29, 31, 32, 33, 40, 42, 50, 53, 62, 74, 76, 79, 81], "frequenc": [11, 64, 67, 73, 74], "whether": [11, 23, 28, 29, 30, 32, 38, 39, 40, 42, 46, 47, 48, 51, 52, 53, 57, 64, 73, 74, 76, 79, 81, 84, 114, 115, 128, 138, 148, 150, 151, 152, 165, 167, 168, 177], "input": [11, 20, 22, 23, 25, 26, 28, 29, 31, 40, 62, 64, 73, 74, 76, 83, 85, 108, 110, 111, 112, 113, 114, 115, 116, 119, 121, 122, 123, 124, 131, 138, 145, 148, 149, 151, 152, 153, 163, 167, 173, 176, 177], "output": [11, 15, 32, 38, 62, 79, 90, 110, 115, 116, 123, 138, 145, 151, 154, 161, 167, 172, 173, 176], "batch": [11, 15, 21, 26, 28, 29, 30, 31, 32, 45, 55, 61, 62, 64, 65, 67, 73, 74, 83, 84, 85, 90, 101, 110, 111, 113, 114, 116, 145, 149, 165, 168, 177], "batch_idx": [11, 15, 61, 62, 64, 65, 67, 73, 74, 83, 84, 85, 90], "dataloader_idx": [11, 15], "0": [11, 15, 33, 41, 46, 47, 48, 49, 51, 76, 77, 83, 84, 85, 90, 101, 112, 115, 117, 123, 126, 128, 138, 143, 145, 147, 148, 149, 151, 152, 154, 168], "step": [11, 32, 59, 61, 77, 177], "contain": [13, 18, 20, 21, 25, 26, 28, 31, 32, 33, 42, 177], "exclud": [13, 18, 20, 31, 76], "search": 13, "within": [13, 48, 151], "layer": [13, 108, 115, 116, 123, 138, 144, 145, 150, 151, 152, 153, 154], "freez": 13, "x_label": [14, 62, 90], "id_label": [14, 61, 90], "arg": [14, 25, 26, 28, 29, 31, 34, 38, 40, 61, 62, 64, 65, 67, 73, 74, 83, 84, 85, 90, 138, 165, 167], "id_field": 14, "lightningmodul": [14, 61, 172], "train_dataload": [14, 24, 27, 28, 33, 160, 165], "val_dataload": [14, 24, 27, 28, 33, 160, 165], "test_dataload": [14, 24, 27, 28, 33, 160, 165], "todo": [14, 23, 28, 29], "better": [14, 148, 150, 151, 152, 153, 177], "mayb": 14, "vol": 14, "predictor": [14, 83, 84, 85, 105, 141], "code": [14, 173, 176], "drop": [14, 30, 143], "unnecessari": 14, "index": [14, 53, 148, 149, 150, 155], "n_epoch": 15, "layer_nam": 15, "act": [15, 138], "set": [15, 28, 48, 172], "forward": [15, 60, 61, 62, 63, 64, 66, 67, 72, 74, 82, 85, 89, 90, 99, 100, 101, 102, 103, 105, 106, 107, 108, 109, 110, 111, 112, 115, 117, 118, 119, 120, 121, 122, 124, 129, 130, 131, 132, 138, 139, 140, 141, 142, 143, 144, 145, 146, 148, 150, 151, 152, 153, 154], "hook": 15, "callabl": [15, 25, 26, 31, 32, 51, 62, 79, 112, 115, 172], "filter": [17, 18, 19, 31, 108, 154], "filter_row": [17, 19, 20], "group_col": [17, 19], "groupcol": [17, 19, 21], "misc": [17, 19], "append_class_weight": [17, 19, 22], "append_labels_to_integ": [17, 19, 22], "append_one_hot": [17, 19, 22], "make_random_df": [17, 19, 22], "split": [17, 19, 28, 29, 30, 33, 61, 173, 176, 177], "sample_n_each": [17, 19, 23], "split_datafram": [17, 19, 23], "columns_to_filt": 18, "sequenc": [18, 20, 22, 25, 26, 28, 29, 30, 31, 39, 40, 41, 43, 45, 51, 55, 62, 76, 79, 90, 138, 140, 147, 148, 149, 155, 167, 173, 176], "regex": [18, 20, 31], "startswith": [18, 20, 31], "endswith": [18, 20, 31], "column": [18, 20, 21, 22, 23, 28, 29, 30, 31, 32, 33, 173, 176, 177], "combin": [18, 30, 173, 176, 177], "differ": [18, 20, 23, 38, 123, 126, 153, 154, 173, 176], "queri": [18, 143, 145, 151, 152], "pattern": 18, "If": [18, 20, 22, 25, 28, 29, 30, 31, 32, 38, 39, 45, 62, 115, 145, 148, 150, 151, 152, 153, 155, 173, 176, 177], "suppli": [18, 20], "preced": 18, "remain": [18, 20, 21], "argument": [18, 22, 25, 26, 28, 29, 31, 33, 39, 64, 67, 73, 74, 83, 84, 85, 90, 107, 138, 173, 176], "ignor": [18, 20, 28, 29, 42, 172], "otherwis": [18, 62, 177], "logic": [18, 61], "AND": 18, "appli": [18, 25, 26, 31, 33, 51, 52, 53, 57, 79, 115, 144, 172, 173, 176], "all": [18, 23, 28, 29, 32, 33, 51, 74, 79, 150, 153, 172, 173, 176, 177], "condit": 18, "regular": [18, 20, 31], "express": [18, 20, 31], "match": [18, 20, 31, 61, 155], "substr": [18, 20, 31], "start": [18, 20, 31, 32, 33, 51, 57, 172, 173, 176], "end": [18, 20, 31], "include_column": 18, "read": [18, 32, 33, 37, 38, 39, 41, 43], "store": [18, 26, 28, 29, 31, 62, 173, 176], "csv": [18, 29, 32, 33, 38, 67, 173, 176, 177], "includ": [18, 20, 23, 148, 149, 151, 152], "onli": [18, 20, 25, 26, 28, 29, 30, 31, 32, 42, 52, 62, 67, 76, 110, 138, 150, 152, 172, 177], "given": [18, 20, 22, 23, 28, 29, 46, 61, 102, 128, 155, 173, 176], "upath": [18, 26, 28, 29, 31], "desir": 18, "result": [18, 20, 61, 62, 177], "type": [18, 21, 29, 30, 32, 38, 61, 79, 112, 115, 123, 138, 145, 163, 168, 173, 176, 177], "pd": [18, 20, 22, 23], "required_column": 18, "parquet": [18, 28, 29], "assert": 18, "expect": [18, 26, 173, 176, 177], "requir": [18, 173, 174, 176, 177], "either": [18, 20, 21, 23, 28, 61, 155, 168], "support": [18, 29, 42, 77, 123], "aren": 18, "t": [18, 33, 38, 102, 174], "found": [18, 173, 176], "valueerror": 18, "thrown": 18, "those": [18, 173, 176], "get": [18, 33, 173, 176], "append": [18, 22, 23], "without": [18, 155], "duplic": 18, "back": 18, "anndata": 18, "h5ad": 18, "r": [18, 174], "": [18, 20, 28, 29, 30, 73, 74, 76, 77, 90, 111, 112, 168, 173, 176], "doc": [18, 25, 26, 28, 29, 31, 73, 74, 115, 173, 176], "detail": [18, 20], "http": [18, 25, 26, 28, 29, 31, 39, 43, 73, 74, 115, 120, 121, 150, 173, 174, 176], "readthedoc": 18, "en": [18, 73, 74, 115, 173, 176], "latest": 18, "html": [18, 25, 26, 28, 29, 31, 73, 74, 115, 173, 176], "select": [20, 23, 30, 46, 173, 176], "dataset": [20, 25, 28, 29, 30, 32, 33, 165, 173, 176], "more": [20, 25, 26, 28, 29, 31, 172, 173, 176], "oper": 20, "It": [20, 28, 101, 177], "panda": [20, 23], "case": [20, 21, 23, 28, 29, 61, 173, 176, 177], "explicit": [20, 130], "keep": [20, 76, 148, 173, 176], "row": [20, 22, 23], "where": [20, 21, 29, 32, 61, 79, 150, 154, 163, 165, 173, 176], "group": [21, 29, 149, 168], "dtype": [21, 38, 79, 126], "kei": [21, 26, 28, 29, 30, 31, 32, 38, 40, 42, 45, 46, 47, 48, 50, 51, 52, 53, 55, 57, 61, 62, 64, 67, 73, 74, 76, 81, 83, 84, 85, 90, 143, 145, 151, 152, 163, 165, 173, 176], "becom": [21, 165, 173, 176], "kwarg": [21, 33, 34, 38, 40, 61, 62, 64, 65, 67, 73, 74, 83, 84, 85, 90, 100, 101, 102, 103, 107, 120, 165], "col": 21, "filter_col": 21, "singl": [21, 25, 28, 32, 151, 155, 173, 176], "np": [21, 25, 79], "numpi": [21, 25, 32, 38, 42, 79, 81, 161], "recogn": 21, "turn": 21, "add": [22, 45, 61, 145], "weight": [22, 62, 76, 83, 84, 85, 90, 111, 112, 126, 131, 144, 151, 154], "modifi": [22, 23, 29, 107], "its": [22, 46, 167], "integ": [22, 32, 51], "encod": [22, 49, 83, 84, 85, 89, 90, 105, 140, 141, 143, 148, 149, 150, 152], "convert": [22, 32, 42, 76, 81, 147, 148, 149, 151, 161], "hot": 22, "call": [22, 28, 29, 61, 77, 172], "function": [22, 32, 51, 62, 76, 90, 110, 111, 112, 113, 114, 115, 116, 126, 128, 172, 173, 176, 177], "one_hot_encod": 22, "b": [22, 130], "c": [22, 38], "d": 22, "n_row": 22, "100": [22, 51, 76, 83, 84, 85, 111, 112], "random": [22, 23, 28, 29, 51], "test": [22, 23, 28, 29, 32, 61, 62, 173, 176], "wrangl": 22, "pipelin": 22, "creat": [22, 25, 26, 28, 29, 31, 32, 42, 64, 76, 123, 154, 161, 173, 174, 176, 177], "number": [22, 23, 28, 29, 33, 38, 49, 51, 52, 57, 77, 83, 84, 85, 107, 108, 115, 126, 138, 145, 148, 149, 150, 151, 152, 153, 154, 155, 172, 177], "forc": 23, "seed": [23, 28, 29], "42": [23, 28, 29], "have": [23, 45, 51, 61, 76, 138, 150, 151, 152], "equal": [23, 126], "per": [23, 28, 29, 51, 76, 126, 145, 161], "In": [23, 61, 168, 173, 176, 177], "ha": [23, 77, 123, 151, 155], "less": 23, "than": [23, 138], "correspond": [23, 29, 31, 51, 165], "sampl": [23, 25, 26, 28, 29, 30, 31, 46, 49, 51, 99, 100, 101, 103, 165], "replac": [23, 33], "uniqu": [23, 173, 176], "toggl": [23, 105, 139], "upsampl": [23, 115, 154, 173, 176], "smaller": 23, "train_frac": 23, "val_frac": 23, "return_split": 23, "perform": [23, 115, 151], "three": [23, 28], "identifi": 23, "each": [23, 25, 26, 28, 29, 30, 31, 33, 49, 57, 61, 64, 76, 90, 138, 140, 144, 148, 149, 150, 151, 152, 153, 154, 155, 165, 173, 176, 177], "belong": 23, "extend": 23, "enabl": 23, "balanc": 23, "stratifi": 23, "fraction": [23, 152], "valid": [23, 33, 61, 64, 73, 74, 173, 176], "By": [23, 177], "half": 23, "between": [23, 29, 30, 32, 64, 73, 74, 101, 110, 115, 126, 153], "separ": [23, 32, 38, 76], "reproduc": 23, "dataframe_datamodul": [24, 27], "dataframedatamodul": [24, 27, 28, 29], "grouped_dataframe_datamodul": [24, 27], "groupeddataframedatamodul": [24, 27, 29], "alternatingbatchsampl": [24, 27, 29, 30], "removenankeysd": [24, 27, 30], "get_canonical_split_nam": [24, 27, 30], "get_dataset": [24, 27, 28, 30], "make_multiple_dataframe_split": [24, 27, 30], "make_single_dataframe_split": [24, 27, 30], "parse_transform": [24, 27, 30], "create_metatensor": [24, 32], "get_per_file_arg": [24, 32, 33], "is_batch": [24, 32], "make_dataload": [24, 27, 28, 29, 33], "predict_dataload": [24, 27, 28, 33, 160, 165], "prepare_data": [24, 33], "ndarrai": [25, 39, 41, 43, 76, 81], "source_kei": [25, 84], "dataloader_kwarg": [25, 26, 28, 29, 31, 32], "dataload": [25, 26, 28, 29, 31, 32, 33, 62, 161, 173, 176], "element": [25, 61, 138, 155, 165], "keyword": [25, 26, 28, 29, 31, 39], "torch": [25, 26, 28, 29, 30, 31, 42, 62, 64, 73, 74, 79, 90, 101, 108, 111, 112, 115, 123, 143, 144, 145, 168, 177], "instanti": [25, 26, 28, 29, 31, 90, 172, 173, 176], "asid": [25, 26, 28, 29, 31], "shuffl": [25, 26, 28, 29, 30, 31], "among": [25, 26, 28, 29, 31], "num_work": [25, 26, 28, 29, 31, 33, 173, 176], "etc": [25, 26, 28, 29, 31, 173, 176, 177], "pytorch": [25, 26, 28, 29, 30, 31, 62, 90, 173, 176], "info": [25, 26, 28, 29, 31], "org": [25, 26, 28, 29, 31, 121], "stabl": [25, 26, 28, 29, 31, 73, 74, 115, 173, 176], "cache_dir": [26, 28, 29, 30, 31, 173, 176], "usual": 26, "cach": [26, 28, 29, 31, 33, 173, 176], "acceler": [26, 28, 29, 31], "get_dataload": [27, 28, 160, 165], "split_map": [28, 29, 30], "just_infer": [28, 29, 30], "subsampl": [28, 29], "refresh_subsampl": [28, 29], "smartcache_arg": [28, 29, 30], "lightningdatamodul": [28, 33, 165], "lightn": [28, 62, 172, 173, 176, 177], "fold": 28, "addition": [28, 172], "go": 28, "flag": 28, "so": [28, 29, 77, 168, 173, 176, 177], "whole": [28, 29], "predict_datamodul": 28, "simpli": [28, 61, 173, 176], "construct": [28, 173, 176], "regardless": 28, "specif": [28, 29, 61, 73, 74, 90, 115, 173, 176, 177], "everyth": [28, 29, 48], "document": [28, 29, 173, 176], "just": [28, 29, 84, 151], "infer": [28, 29, 61, 73, 74, 90, 100, 101, 103, 168], "epoch": [28, 29, 83, 84, 85], "refresh": [28, 29], "time": [28, 29, 30, 77, 151, 152, 172], "smartcachedataset": [28, 29], "target_column": [29, 30], "grouping_column": [29, 30], "should": [29, 32, 33, 40, 51, 61, 115, 126, 138, 151, 152, 154, 177], "some": [29, 32, 138, 173, 176, 177], "criterion": 29, "leverag": 29, "two": [29, 50, 101, 173, 176], "multitask": 29, "ground": [29, 30, 76, 115, 123, 173, 176, 177], "truth": [29, 30, 76, 115, 123, 173, 176, 177], "task": [29, 30, 64, 67, 73, 74, 77, 110, 111, 113, 114, 115, 116, 147, 148, 149, 172, 173, 176, 177], "characterist": 29, "altern": [29, 30, 177], "dure": [29, 32, 64, 67, 73, 74, 114, 154, 172, 173, 176], "factor": [29, 51, 154, 177], "homogen": 29, "across": [29, 53, 76, 173, 176], "subset": 30, "drop_last": 30, "sampler": 30, "subsetrandomsampl": 30, "batchsampl": 30, "mutual": 30, "exclus": 30, "monai": [30, 33, 51, 53, 73, 74, 115, 138, 173, 176, 177], "wrap": [30, 62, 172, 173, 176], "repres": [30, 138], "last": [30, 52, 87, 123, 138, 144, 151], "incomplet": 30, "randomli": [30, 49, 52, 138], "follow": [30, 151, 152, 177], "order": [30, 138, 148, 149, 152, 154, 167], "remov": [30, 49, 76, 152], "nan": 30, "ad": [30, 45, 49, 177], "allow_missing_kei": [30, 38, 47, 48, 51, 52, 53, 55, 57, 76, 81], "allow": [30, 38, 52, 55, 57, 177], "multi": [30, 32, 151, 166, 172, 173, 176, 177], "target": [30, 49, 61, 67, 77, 84, 119, 121, 122, 123, 124, 128, 130, 131, 153, 173, 176, 177], "split_path": 30, "dataframe_path": 30, "orig_fnam": 31, "origin": [31, 45], "filenam": [31, 39, 41, 43], "extens": 31, "out": [31, 64, 73, 74, 138, 152, 154, 173, 176], "rule": [31, 177], "csv_path": [32, 33], "img_path_column": [32, 33], "channel_column": [32, 33], "out_kei": [32, 38], "3": [32, 33, 49, 51, 76, 77, 107, 108, 112, 115, 117, 138, 145, 147, 148, 149, 150, 151, 152, 153, 154, 168, 174, 177], "scene_column": 32, "scene": [32, 33, 38, 39], "time_start_column": 32, "time_stop_column": 32, "stop": 32, "time_step_column": 32, "dict_meta": 32, "dask_load": [32, 38, 39], "dimension": [32, 100, 101, 102, 151], "timelaps": 32, "metadata": [32, 45, 165], "timepoint": [32, 33, 38, 39], "extract": [32, 38, 39, 83, 84, 85, 152], "spatial": [32, 33, 49, 76, 107, 115, 126, 138, 145, 148, 149, 151, 152, 153, 154, 177], "yx": [32, 49, 148, 149, 152, 177], "zyx": [32, 49, 148, 149, 152, 154, 177], "specifi": [32, 33, 38, 39, 48, 53, 81, 90, 149, 151, 152, 173, 176, 177], "multipl": [32, 51, 107, 108, 138, 153, 173, 176], "thei": [32, 45, 77], "comma": [32, 38], "scene1": 32, "scene2": 32, "ani": [32, 59, 79, 126, 173, 176, 177], "start_column": 32, "stop_column": 32, "step_column": 32, "For": [32, 51, 115, 138, 151, 173, 176, 177], "everi": [32, 62, 163], "other": [32, 173, 176, 177], "version": [32, 107, 177], "accept": [32, 126], "normal": [32, 108, 138, 143, 144, 145, 173, 176, 177], "could": 32, "here": [32, 33, 61, 121, 173, 176, 177], "dask": [32, 38, 39], "full": [32, 38, 39], "memori": [32, 33, 38, 39, 177], "befor": [32, 38, 39, 51, 57, 128, 172], "img": [32, 39, 41, 43, 57, 76, 81, 140, 148, 151, 152, 154], "meta": 32, "df": [32, 33], "x": [32, 61, 62, 64, 74, 85, 90, 107, 108, 110, 111, 112, 115, 117, 120, 129, 132, 138, 139, 143, 144, 145, 149, 151, 177], "listconfig": [32, 52, 57, 163], "multidimimag": 32, "cannot": 32, "dim": [32, 76, 107, 108, 116, 132, 145], "multiscen": 32, "alreadi": [32, 173, 176], "compos": [33, 167], "img_data": 33, "n_val": 33, "20": 33, "pct_val": 33, "raw": [33, 38, 51, 173, 176, 177], "ch": [33, 46, 79], "num_neighbor": 33, "4": [33, 140, 143, 145, 150, 151, 152, 154], "cache_r": 33, "5": [33, 49, 51, 76, 129], "replace_r": 33, "czi": 33, "don": 33, "fit": 33, "bioio": 33, "n_imag": 33, "percentag": 33, "neighbor": [33, 76], "worker": 33, "most": [33, 61], "schedul": 33, "parallel": 33, "submodul": [36, 156, 175], "aicsimage_load": [36, 37], "aicsimageloaderd": [36, 37, 38], "monai_bio_read": [36, 37, 38], "monaibioread": [36, 37, 39], "numpy_read": [36, 37], "readnumpyfil": [36, 37, 40], "ome_zarr_read": [36, 37], "omezarrread": [36, 37, 41], "polygon_load": [36, 37], "polygonloaderd": [36, 37, 42], "skimage_read": [36, 37], "skimageread": [36, 37, 43], "add_meta": [36, 44], "addmeta": [36, 44, 45], "bright_sampl": [36, 44], "brightsampl": [36, 44, 46], "clip": [36, 44, 48], "clipd": [36, 44, 47], "contrastadjust": [36, 44], "contrastadjustd": [36, 44, 48], "generate_jepa_mask": [36, 44], "jepamaskgener": [36, 44, 49], "merg": [36, 44], "multiscale_cropp": [36, 44], "randommultiscalecropd": [36, 44, 51], "pad": [36, 44, 76, 138], "padzd": [36, 44, 52], "project": [36, 44, 145, 173, 176], "maxprojectd": [36, 44, 53], "rotation_mask_transform": [36, 44], "so2_random_rot": [36, 44], "track_transform": [36, 44], "cropres": [36, 44, 57], "generatetracklabel": [36, 44, 57], "perchannel": [36, 44, 57], "splittrackd": [36, 44, 57], "split_arg": [37, 38], "get_data": [37, 39, 41, 43], "verify_suffix": [37, 39, 41, 43], "path_kei": 38, "scene_kei": 38, "kwargs_kei": 38, "dimension_order_out": [38, 177], "float16": 38, "channel_kei": 38, "timepoint_kei": 38, "fix": [38, 148, 150, 151, 152, 153, 155], "initi": [38, 77, 115, 123, 166], "bioimag": [38, 39], "get_image_dask_data": [38, 39], "miss": [38, 42, 47, 48, 52, 53, 55, 57, 76, 81], "adapt": [39, 43, 120], "github": [39, 43, 120, 150, 174], "com": [39, 43, 120, 174], "mmv": [39, 43], "lab": [39, 43], "mmv_im2im": [39, 43], "reader_kwarg": 39, "imageread": [39, 41, 43], "get_image_data": 39, "pathlik": [39, 41, 43], "maptransform": 40, "thereof": 40, "interpret": 40, "point": [40, 51, 76, 138, 177], "cloud": [40, 120], "remot": 40, "fsspec": 40, "locat": [40, 173, 176], "level": [41, 149, 173, 176, 177], "image_nam": 41, "shape_reference_kei": 42, "propagate_3d": 42, "missing_key_mod": 42, "rais": [42, 53, 76, 81], "polygon": 42, "vertic": 42, "tensor": [42, 81, 84, 85, 101, 123, 124, 131, 138, 145, 147, 154, 168, 177], "mask": [42, 49, 50, 85, 114, 145, 147, 148, 149, 150, 151, 152, 154], "propag": 42, "2d": [42, 49, 83, 115, 138, 148, 149, 168, 173, 176, 177], "3d": [42, 49, 76, 83, 115, 138, 148, 149, 168, 173, 176, 177], "how": [42, 115, 165, 173, 176], "keyerror": 42, "do": [42, 115, 123, 173, 176], "noth": 42, "new": [42, 45, 173, 176, 177], "blank": 42, "remove_excess_pixel": [44, 49], "generate_slic": [44, 51], "image_kei": [45, 50, 52, 76], "delet": 45, "updat": [45, 61], "metatensor": [45, 165], "been": [45, 151], "threshold": [46, 79, 128], "base_prob": 46, "decid": [46, 51], "intens": [46, 47, 48, 79, 177], "background": [46, 48], "rate": [46, 138, 145], "probabilii": 46, "check": [46, 52], "against": 46, "low": [47, 48, 49], "01": 47, "high": [47, 48, 49], "99": 47, "percentil": [47, 48], "absolut": [47, 48], "lower": [47, 48], "bound": [47, 48], "upper": [47, 48], "per_channel": 47, "resiz": [47, 48, 51, 57, 177], "fail": [47, 48], "adjust": 48, "rang": [48, 49, 79, 173, 176, 177], "outsid": 48, "everywher": 48, "mask_siz": 49, "12": [49, 145, 151, 152, 153, 154], "block_aspect_ratio": 49, "num_patch": [49, 148, 150, 151, 152, 153, 154, 155], "6": [49, 76, 154], "24": 49, "mask_ratio": [49, 147, 148, 149, 151, 152, 154], "9": [49, 174], "randomizabletransform": [49, 51, 57, 76], "block": [49, 105, 115, 138, 141, 151, 152], "contigu": 49, "work": [49, 101, 148, 150, 151, 152, 153, 174, 177], "until": [49, 177], "met": 49, "exceed": [49, 177], "exterior": 49, "exactli": 49, "determin": [49, 115, 138, 167, 173, 176, 177], "aspect": 49, "ratio": [49, 115, 138, 145, 152, 154], "proport": 49, "pixel": [49, 57, 76, 126, 148, 149, 151, 152, 177], "along": 49, "boundari": 49, "reach": 49, "mask_kei": [50, 114], "base_image_kei": 50, "output_nam": 50, "serv": 50, "patch_shap": [51, 139, 177], "scales_dict": 51, "patch_per_imag": 51, "selection_fn": 51, "max_attempt": 51, "style": [51, 53, 152, 177], "slice": [51, 52, 76, 177], "scale": [51, 107, 111, 112, 123, 126], "superresolut": 51, "x_kei": [51, 64, 73, 74, 83, 85], "network": [51, 62, 64, 73, 74, 90, 115, 138, 177], "Its": 51, "resolut": [51, 112, 115, 151], "len": 51, "map": [51, 110, 138, 177], "seg": [51, 76, 105, 141, 173, 176], "would": [51, 61], "kept": 51, "discard": 51, "max": [51, 53], "attempt": 51, "try": [51, 177], "quit": 51, "image_dict": 51, "segmentation_kei": 52, "pad_amount": 52, "pad_kei": 52, "segmentation_ch": 52, "top": [52, 154, 177], "bottom": 52, "crop": 52, "repeat": [52, 151, 152, 155], "first": [52, 101, 108, 151, 168, 173, 176, 177], "present": [52, 76], "offset": 52, "presenc": 52, "projection_dim": 53, "nczyx": 53, "comput": [53, 101, 128, 168], "error": [53, 76, 81], "save_path": 55, "disk": 55, "debug": 55, "max_shift": 57, "8": [57, 101, 140, 143, 145, 149, 151, 152, 154, 173, 176], "maximum": [57, 83, 84, 85], "shift": 57, "img_kei": 57, "formation_kei": 57, "breakdown_kei": 57, "breakdown": 57, "track_start_kei": 57, "track_start": 57, "label_kei": [57, 76], "label": [57, 64, 67, 76, 79, 123], "track": 57, "after_save_checkpoint": [58, 59], "log_hyperparam": [58, 59], "log_metr": [58, 59], "experiment_nam": [59, 173, 176], "lightning_log": 59, "run_nam": [59, 173, 176], "tracking_uri": [59, 87], "tag": [59, 153, 167, 172], "mlrun": 59, "prefix": 59, "artifact_loc": 59, "run_id": [59, 87], "fault_toler": 59, "ckpt_callback": 59, "param": [59, 110], "namespac": 59, "mode": [59, 87, 100, 101, 102, 103, 117, 121, 130, 172], "metric": [59, 61, 110, 113, 114, 116, 172], "timepoint_classif": [60, 63], "timepointclassif": [60, 63, 65], "base_handl": [60, 68], "image_handl": [60, 68], "load_image_patch": [60, 68], "multi_task": [60, 72], "multitaskim2im": [60, 72, 73, 74], "ijepa": [60, 82, 85], "iwm": [60, 82, 85], "jepa_bas": [60, 82], "jepabas": [60, 82, 83, 84, 85], "load_model_from_checkpoint": [60, 86, 87], "find_indic": [60, 86, 88], "o2_spharm_va": [60, 89], "prior": [60, 89, 90], "base_va": [60, 89], "baseva": [60, 89, 90], "image_canon_va": [60, 89], "image_encod": [60, 89], "image_va": [60, 89], "latent_loss_va": [60, 89], "point_cloud_va": [60, 89], "tabular_va": [60, 89], "compute_metr": [60, 61], "configure_optim": [60, 61, 72, 73, 74, 82, 85], "model_step": [60, 61, 62, 63, 64, 66, 67, 72, 73, 74, 82, 83, 84, 85, 89, 90], "on_train_start": [60, 61], "parse_batch": [60, 61], "predict_step": [60, 61, 63, 64, 65, 66, 67, 72, 73, 74, 82, 84, 85], "test_step": [60, 61, 72, 74, 82, 84], "training_step": [60, 61, 72, 74], "validation_step": [60, 61, 72, 74], "log": [61, 62, 101, 172, 173, 176, 177], "assumpt": 61, "_step": 61, "whose": 61, "mai": 61, "self": [61, 76, 83, 84, 151, 152], "structur": [61, 167, 177], "part": [61, 90, 172, 173, 176], "arbitrari": 61, "denot": 61, "you": [61, 173, 176, 177], "implement": [61, 173, 176, 177], "process": [61, 172, 177], "variabl": [61, 173, 176], "hparam": 61, "id": 61, "detach": [61, 75, 78, 81, 110, 111, 112, 113, 114, 115], "consist": [61, 154, 173, 176], "your": [61, 173, 176, 177], "might": 61, "wish": 61, "post": 61, "minim": 62, "wrapper": [62, 126, 128], "around": [62, 148, 149, 151, 152], "assum": [62, 168, 173, 176, 177], "gt": [62, 120], "optim": [62, 111, 112, 173, 176, 177], "retriev": [62, 172], "adam": 62, "save_predict": 62, "serotini": [62, 173, 176], "fields_to_log": 62, "field": [62, 167, 177], "pretrained_weight": 62, "pretrain": [62, 154], "via": [62, 151], "load_state_dict": 62, "save_imag": [63, 64], "should_save_imag": [63, 64], "share": [64, 73, 74, 177], "save_images_every_n_epoch": [64, 73, 74], "base_kwarg": [64, 67, 73, 74, 83, 84, 85, 90], "logit": [64, 115, 173, 176], "text": 64, "overlaid": 64, "plot_class": [66, 67], "plot_neighbor": [66, 67], "backbon": [67, 73, 74, 110, 113, 114, 115, 116, 154, 173, 176, 177], "task_head": [67, 73, 74, 161, 177], "anchor_kei": 67, "anchor": 67, "positive_kei": 67, "posit": [67, 140, 147, 148, 149, 150, 151, 152, 153, 155], "target_kei": [67, 84, 90], "visual": [67, 173, 176], "viz_freq": 67, "x1": 67, "x2": 67, "embedding1": 67, "embedding2": 67, "instancesegclust": [72, 75, 76], "instancesegloss": [72, 75, 76], "instancesegpreprocessd": [72, 75, 76], "instancesegrandflipd": [72, 75, 76], "pad_slic": [72, 75, 76], "noise_anneal": [72, 75, 108], "noiseanneal": [72, 75, 77], "get_n_postprocess_imag": [72, 74], "get_per_head": [72, 74], "run_forward": [72, 74, 110, 111, 113, 114, 116], "basic": [73, 90, 173, 176, 177], "inference_arg": [73, 74], "slide": [73, 74], "window": [73, 74], "inference_head": 74, "run_head": [74, 109, 110, 111, 112, 113, 114, 116], "n_postprocess": [74, 110, 111, 113, 114, 116], "act_thresh_label": [75, 78], "actthreshlabel": [75, 78, 79], "auto_thresh": [75, 78], "autothreshold": [75, 78, 80], "dict_to_im": [75, 78], "dicttoim": [75, 78, 81], "cluster_object": [75, 76], "kd_cluster": [75, 76], "remove_small_skeleton": [75, 76], "embed_from_skel": [75, 76], "keep_largest_cc": [75, 76], "label_2d": [75, 76], "shrink": [75, 76], "skeleton_tal": [75, 76], "smooth_embed": [75, 76], "topology_preserving_thin": [75, 76], "update_nois": [75, 77], "anisotropi": 76, "skel_threshold": 76, "semantic_threshold": 76, "min_siz": 76, "1000": 76, "distance_threshold": 76, "progress": 76, "cluster": 76, "instanceseg": 76, "find": 76, "skeleton": 76, "assign": 76, "semant": 76, "embed": [76, 84, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155], "nearest": [76, 117], "distanc": [76, 177], "skel": 76, "closest": 76, "below": [76, 128, 173, 176], "touch": 76, "edg": [76, 177], "compon": [76, 167], "kernel_s": [76, 138, 139], "thin": 76, "keep_largest": 76, "instanc": [76, 77, 138, 154, 177], "kernel": [76, 138], "gaussian": [76, 77, 89, 99], "smooth": 76, "flow": 76, "amount": [76, 115, 138], "psuedo": 76, "largest": 76, "connect": [76, 115, 138, 154], "iseg": 76, "vector": [76, 168, 177], "z": [76, 90, 100, 101, 102, 152, 168, 173, 176], "im": [76, 108], "topologi": 76, "preserv": 76, "binari": 76, "max_label": 76, "increas": [76, 77, 177], "thick": 76, "convolv": 76, "non": [76, 151], "bw": 76, "bridg": 76, "gap": 76, "eros": 76, "spatial_axi": 76, "prob": [76, 121], "flip": [76, 177], "augment": [76, 173, 176, 177], "sign": 76, "gradient": 76, "chang": [76, 173, 176, 177], "axi": [76, 168], "NOT": 76, "probabl": 76, "constraint": 76, "annealing_step": 77, "5000": 77, "init_vari": 77, "anneal": [77, 108], "varianc": [77, 101], "nois": [77, 108], "real": [77, 107, 123], "fake": [77, 123], "harder": 77, "distribut": [77, 102], "overlap": 77, "nice": 77, "theoret": 77, "implic": 77, "qualiti": 77, "also": [77, 101, 138, 173, 176, 177], "curriculum": 77, "learn": [77, 83, 84, 177], "techniqu": 77, "iter": 77, "unblur": 77, "over": 77, "linearli": 77, "activ": [79, 107, 115, 138, 143, 145, 174], "ident": [79, 112, 115, 138, 173, 176], "_type": 79, "_dtype_lik": 79, "_supportsdtyp": 79, "supportsindex": 79, "collect": 79, "_dtypedict": 79, "uint8": 79, "rescale_dtyp": 79, "purpos": 79, "rescal": [79, 173, 176], "dtypelik": 79, "multichannel": 81, "concaten": 81, "cuda": 81, "extract_embed": [82, 84], "get_predict_mask": [82, 84], "get_context_embed": [82, 85], "get_mask": [82, 85, 146, 148], "get_target_embed": [82, 85], "remove_first_dim": [82, 85], "update_teach": [82, 85], "supervis": [83, 84], "featur": [83, 84, 85, 110, 113, 114, 116, 123, 138, 143, 144, 145, 150, 151, 152, 153, 154, 173, 176], "access": [83, 84, 85], "momentum": [83, 84, 85], "exponenti": [83, 84, 85], "move": [83, 84, 85], "averag": [83, 84, 85, 126], "998": [83, 84, 85], "world": 84, "translat": 84, "latent": [84, 90], "space": [84, 90], "target_domain_kei": 84, "domain": [84, 153], "predict_target": 84, "devic": [84, 155], "tmp_dir": 87, "strict": 87, "ckpt": [87, 177], "lst": 88, "o2_spharm_encod": [89, 95], "abstract_prior": [89, 99], "diagonalgaussianprior": [89, 99, 101], "isotropicgaussianprior": [89, 99, 101], "compute_tc_penalti": [89, 99, 101], "identity_prior": [89, 99], "identityprior": [89, 99, 102], "joint_prior": [89, 99], "jointprior": [89, 99, 103], "calculate_elbo": [89, 90], "calculate_rcl": [89, 90], "calculate_rcl_dict": [89, 90], "decod": [89, 90, 105, 141, 143, 152, 154], "sample_z": [89, 90], "beta": [90, 130], "kld": 90, "term": [90, 138], "reconstruction_loss": [90, 111, 112], "reconstruct": [90, 111, 112, 160, 169], "_loss": [90, 119, 121, 122, 124, 130, 131], "abstractprior": 90, "decoder_latent_part": 90, "decoer": 90, "what": [90, 167], "depend": [90, 174], "xhat": 90, "input_kei": 90, "return_param": 90, "z_parts_param": 90, "kl_diverg": [99, 100, 101, 103], "param_s": [99, 100, 101, 102, 103], "kl": [100, 101, 102, 103], "properti": [100, 101, 102, 103, 147, 148, 149, 152], "logvar": 101, "learn_mean": 101, "learn_logvar": 101, "clamp_logvar": 101, "tc_penalty_weight": 101, "mu1": 101, "mu2": 101, "logvar1": 101, "logvar2": 101, "reduct": [101, 103, 119, 121, 122, 130, 131], "sum": [101, 103, 130, 144, 151, 154], "kullback": 101, "leibler": 101, "diverg": 101, "diagon": 101, "necessarili": 101, "isotrop": 101, "wise": 101, "second": [101, 151], "doesn": [102, 174], "contribut": 102, "effect": 102, "dirac": 102, "delta": 102, "z_param": 103, "multi_scale_discrimin": [105, 106], "multiscalediscrimin": [105, 106, 107], "n_layer_discrimin": [105, 106], "nlayerdiscrimin": [105, 106, 107, 108], "base_head": [105, 109], "basehead": [105, 109, 110, 111, 113, 114, 115, 116], "gan_head": [105, 109], "ganhead": [105, 109, 111, 112], "gan_head_superr": [105, 109], "ganhead_res": [105, 109, 112], "mae_head": [105, 109], "maehead": [105, 109, 113], "mask_head": [105, 109], "maskhead": [105, 109, 114], "res_blocks_head": [105, 109], "resblockshead": [105, 109, 112, 115], "vic_reg": [105, 109, 118], "vicreghead": [105, 109, 116], "adversarial_loss": [105, 118], "adversarialloss": [105, 118, 119], "chamfer_loss": [105, 118], "chamferloss": [105, 118, 120], "continuous_bernoulli": [105, 118], "cblogloss": [105, 118, 121], "cosine_loss": [105, 118], "cosineloss": [105, 118, 122], "gan_loss": [105, 111, 112, 118], "ganloss": [105, 111, 112, 118, 123], "pix2pixhd": [105, 107, 111, 112, 118, 123], "gaussian_nll_loss": [105, 118], "gaussiannllloss": [105, 118, 124], "geomloss": [105, 118], "loss_wrapp": [105, 118], "cmap_loss": [105, 118, 126], "losswrapp": [105, 118, 126], "spharm_loss": [105, 118], "threshold_loss": [105, 118], "thresholdloss": [105, 118, 128], "vicregloss": [105, 118, 129], "off_diagon": [105, 118, 129], "weibul": [105, 118], "weibulllogloss": [105, 118, 130], "weibull_log_prob": [105, 118, 130], "weighted_mse_loss": [105, 118], "weightedmseloss": [105, 118, 131], "dgcnn": [105, 133], "folding_net": [105, 133], "graph_funct": [105, 133], "vnn": [105, 133], "crossmae_decod": [105, 141, 150], "mae_decod": [105, 141, 150], "hieraencod": [105, 141, 151], "jepaencod": [105, 141, 151], "mae_encod": [105, 141, 151], "spatialmerg": [105, 141, 151], "mae": [105, 141], "hierama": [105, 141, 152], "mae_bas": [105, 141, 152], "iwmpredictor": [105, 141, 153], "jepapredictor": [105, 141, 153], "encodedskip": [105, 141, 154], "seg_vit": [105, 141, 154], "superresdecod": [105, 141, 154], "get_positional_embed": [105, 141, 155], "match_tuple_dimens": [105, 141, 155], "random_index": [105, 141, 155], "take_index": [105, 141, 155], "n_scale": 107, "input_nc": 108, "ndf": 108, "n_layer": 108, "norm_lay": [108, 143, 144, 145], "instancenorm": 108, "instancenorm3d": 108, "defin": [108, 123], "patchgan": 108, "n_channel": 108, "input_im": 108, "model_output": 108, "conv_fn": 108, "later": 108, "requires_featur": 108, "standard": 108, "generate_io_map": [109, 110], "update_param": [109, 110], "input_filenam": 110, "backbone_featur": [110, 111, 113, 114, 116], "y_hat": [110, 111, 113, 114, 116], "calcul": [110, 113, 114, 116, 126], "bcewithlogitsloss": [111, 112, 123], "feature_matching_loss": [111, 112], "l1loss": [111, 112], "mseloss": [111, 112], "reconstruction_loss_weight": [111, 112], "in_channel": [112, 115, 117, 138, 177], "out_channel": [112, 115, 117, 138, 154, 177], "final_act": [112, 115], "lr": [112, 115], "n_conv": [112, 115], "dropout": [112, 115, 117, 138, 145], "upsample_method": [112, 115], "pixelshuffl": [112, 115], "upsample_ratio": [112, 115], "first_lay": [112, 115], "dens": [112, 115], "inherit": 112, "__init__": 112, "calculate_metr": 114, "convolut": [115, 138, 148, 149, 151, 152, 154, 173, 176], "final": 115, "done": 115, "hr": 115, "length": [115, 126, 140, 155], "convprojectionlay": 115, "2048": 116, "8192": 116, "hidden_lay": [116, 132], "projectornetwork": 116, "hidden": [116, 145], "scale_factor": 117, "align_corn": 117, "antialia": 117, "batch_pairwise_dist": [118, 120], "batch_pairwise_dist2": [118, 120], "get_target_tensor": [118, 123], "get_feature_matching_loss": [118, 123], "get_gan_loss": [118, 123], "argmax": 119, "squeez": 119, "return_pr": 119, "sentinal4d": 120, "cellshap": 120, "blob": 120, "cellshape_cloud": 120, "vendor": 120, "chamfer_dist": 120, "py": [120, 177], "licens": 120, "license_antao": 120, "y": [120, 129, 143], "continu": 121, "bernoulli": 121, "propos": 121, "arxiv": 121, "ab": 121, "1907": 121, "06845": 121, "gan_mod": 123, "vanilla": 123, "target_real_label": 123, "target_fake_label": 123, "abstract": [123, 148, 152, 173, 176], "awai": 123, "need": [123, 173, 176, 177], "lsgan": 123, "wgangp": 123, "note": [123, 177], "sigmoid": [123, 173, 176], "target_is_r": 123, "fill": 123, "loss_weight": 123, "fm": 123, "10": [123, 124, 130, 138], "feature_typ": 123, "mean_dim": 124, "ep": [124, 130, 138, 168], "1e": [124, 130, 138, 168], "costmap": 126, "differenti": [126, 173, 176], "emphas": 126, "throughout": 126, "loss_fn": [126, 128], "channel_weight": 126, "_supportsarrai": 126, "_nestedsequ": 126, "complex": [126, 173, 176], "byte": 126, "loss_scal": 126, "arraylik": 126, "abov": 128, "num_featur": 129, "sim_coeff": 129, "std_coeff": 129, "cov_coeff": 129, "alpha": 130, "256": [132, 139], "scale_output": 132, "stride": [138, 145, 151, 152], "subunit": 138, "adn_ord": 138, "nda": 138, "prelu": 138, "norm": [138, 144], "dropout_dim": 138, "dilat": 138, "bia": [138, 145], "last_conv_onli": 138, "residu": 138, "conv": 138, "init": 138, "normalized_shap": 138, "print": [138, 167, 172], "sequenti": 138, "unit0": 138, "conv3d": 138, "adn": 138, "num_paramet": 138, "n": [138, 173, 174, 176], "layernorm": [138, 143, 144, 145], "05": 138, "elementwise_affin": 138, "unit1": 138, "zero": [138, 172], "entir": 138, "larger": 138, "control": [138, 172], "implicit": 138, "both": [138, 177], "side": 138, "in_filt": 139, "out_filt": 139, "n_input_ch": 139, "512": 139, "n_conv_filt": 139, "pos_embedding_length": 140, "120": 140, "emb_dim": [140, 143, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155], "num_lay": [140, 144, 150, 151, 153, 154], "num_head": [140, 143, 145, 150, 151, 152, 153], "num_class": 140, "token": [140, 145, 147, 149, 150, 155], "classifi": 140, "them": [140, 173, 176], "d_model": 140, "matrix": [140, 168], "patchifi": [141, 142, 151], "cross_attent": [141, 142], "crossattent": [141, 142, 143], "crossattentionblock": [141, 142, 143], "crossselfblock": [141, 142, 143], "intermediate_weigh": [141, 142], "intermediateweigh": [141, 142, 144], "masked_unit_attent": [141, 142], "hierablock": [141, 142, 145], "maskunitattent": [141, 142, 145], "add_mask_token": [141, 150], "adjust_indices_for_cl": [141, 150], "init_weight": [141, 150, 151, 153], "init_decod": [141, 152], "init_encod": [141, 152], "predict_target_featur": [141, 153], "patchify_bas": [142, 146], "patchifybas": [142, 146, 147, 148, 149], "patchify_hiera": [142, 146], "patchifyhiera": [142, 146, 149], "take_indexes_mask": [142, 146, 149], "encoder_dim": 143, "decoder_dim": [143, 152], "qkv_bia": [143, 145], "qk_scale": 143, "attn_drop": [143, 145], "proj_drop": [143, 145], "mlp_ratio": [143, 145], "drop_path": [143, 145], "act_lay": [143, 145], "gelu": [143, 145], "in_featur": 143, "hidden_featur": 143, "out_featur": 143, "embed_dim": [144, 151], "n_output": 144, "intermedi": [144, 151, 154], "dim_out": 145, "q_stride": [145, 151, 152], "patches_per_mask_unit": 145, "attent": [145, 151, 152], "unit": [145, 149, 151, 152], "learnabl": [145, 148, 150, 151, 152, 153, 155], "create_img2token": [146, 147, 149], "extract_visible_token": [146, 147, 148, 149], "get_mask_arg": [146, 147, 148, 149], "img2token": [146, 147, 148, 149], "create_conv": [146, 148], "create_patch2img": [146, 148], "n_patch": [147, 148, 149], "context_pixel": [147, 148, 149, 151, 152], "learnable_pos_embed": [147, 148, 150, 151, 152, 153], "rearrang": 147, "forward_index": [147, 149, 150], "n_visible_patch": [147, 148, 149], "pix": [148, 149], "sin": [148, 150, 151, 152, 153, 168], "co": [148, 150, 151, 152, 153, 168], "empir": [148, 150, 151, 152, 153], "brightfield": [148, 150, 151, 152, 153], "boolean": [148, 177], "mask_units_per_dim": 149, "mask_unit": 149, "enc_dim": 150, "768": [150, 152, 154], "192": [150, 151, 152, 153, 154], "has_cls_token": 150, "inspir": 150, "crossma": [150, 152], "attend": 150, "visibl": 150, "cl": [150, 155], "backward_index": 150, "num_mask_unit": [151, 152], "architectur": [151, 152, 177], "save_lay": 151, "multihead": [151, 152], "self_attent": [151, 152], "On": 151, "doubl": 151, "pool": 151, "doe": [151, 177], "while": [151, 177], "produc": [151, 173, 176], "context": [151, 153], "n_intermediate_weight": 151, "75": [151, 152, 154], "downsample_factor": 151, "in_dim": 151, "out_dim": 151, "hiera": 151, "lowest": 151, "decoder_lay": [152, 154], "decoder_head": 152, "use_crossma": 152, "features_onli": 152, "encoder_lay": [152, 154], "encoder_head": [152, 154], "input_dim": 153, "special": 153, "condition": 153, "fluoresc": 153, "context_emb": 153, "target_mask": 153, "target_domain": 153, "n_decoder_filt": 154, "base_patch_s": 154, "upsample_factor": 154, "6134": 154, "5005": 154, "encoder_ckpt": 154, "freeze_encod": 154, "encoder_kwarg": 154, "simpl": [154, 173, 176], "aic": 154, "20x": 154, "100x": 154, "unet": 154, "like": [154, 177], "fed": [154, 173, 176], "skip": 154, "use_cls_token": 155, "ensur": [155, 177], "read_pcloud": [156, 157], "generate_img": [160, 165], "compute_rotation_matrix": [160, 168], "data_cfg": 161, "load_param": 162, "kv": 163, "item": 163, "nest": 163, "form": [163, 168], "interpol": [163, 173, 176], "limit": [164, 173, 176, 177], "fn": 164, "num_sampl": 165, "dummy_metadata": 165, "10000": 165, "mani": [165, 173, 176, 177], "k": 165, "friendli": 166, "save_to_fil": 167, "prompt": 167, "user": [167, 173, 176], "print_ord": 167, "resolv": 167, "content": [167, 172], "rich": [167, 172], "librari": 167, "tree": 167, "hydra": [167, 173, 176, 177], "refer": [167, 173, 176], "export": 167, "pad_valu": 168, "06": 168, "pose": 168, "about": 168, "equivari": [168, 174], "theta": 168, "indic": 168, "pair": 168, "precis": 168, "sure": 172, "close": 172, "properli": 172, "prevent": 172, "failur": 172, "multirun": 172, "warn": 172, "metric_dict": 172, "metric_nam": 172, "safe": 172, "callbacks_cfg": 172, "logger_cfg": 172, "object_dict": 172, "rank": 172, "task_func": 172, "decor": 172, "resist": 172, "finish": 172, "except": 172, "occur": 172, "total": 172, "execut": 172, "dir": 172, "aim": [173, 176], "common": [173, 176], "manner": [173, 176], "easi": [173, 176], "beginn": [173, 176], "flexibl": [173, 176, 177], "custom": [173, 176], "templat": [173, 176, 177], "conveni": [173, 176], "syntax": [173, 176], "repo": [173, 176], "best": [173, 176], "familiar": [173, 176, 177], "mechan": [173, 176], "topic": [173, 176], "cover": [173, 176], "becaus": [173, 176], "help": [173, 176], "These": [173, 176], "reli": [173, 176], "heavili": [173, 176], "omegaconf": [173, 176], "expos": [173, 176], "_aux": [173, 176], "section": [173, 176], "persist": [173, 176], "speed": [173, 176], "up": [173, 176, 177], "divid": [173, 176], "manifest": [173, 176], "preprocess": [173, 176], "offer": [173, 176], "wide": [173, 176], "often": [173, 176, 177], "primari": [173, 176], "due": [173, 176], "score": [173, 176], "v": [173, 176], "binar": [173, 176, 177], "own": [173, 176], "thing": [173, 176], "pin_memori": [173, 176], "workhors": [173, 176], "source_col": [173, 176, 177], "target_col": [173, 176], "organ": [173, 176], "arrang": [173, 176], "through": [173, 176], "further": [173, 176], "lr_schedul": [173, 176], "alwai": [173, 176], "well": [173, 176], "_target_": [173, 176], "_partial_": [173, 176], "sigmoid_rescal": [173, 176], "tell": [173, 176], "bit": [173, 176], "encourag": [173, 176], "few": [173, 176, 177], "git": 174, "clone": 174, "allencellmodel": 174, "cd": 174, "conda": 174, "environ": 174, "myenv": 174, "pip": 174, "txt": 174, "relat": 174, "equiv": 174, "upgrad": 174, "subpackag": 175, "outlin": 177, "modif": 177, "suit": 177, "aicsimageio": 177, "recommend": 177, "suitabl": 177, "robust": 177, "again": 177, "excel": 177, "naiv": 177, "incorrect": 177, "neural": 177, "signal": 177, "cut": 177, "crucial": 177, "bottleneck": 177, "suggest": 177, "look": 177, "consider": 177, "primarili": 177, "As": 177, "thumb": 177, "enough": 177, "entiti": 177, "establish": 177, "too": 177, "downsampl": 177, "decreas": 177, "usag": 177, "ckpt_path": 177, "dictat": 177, "omit": 177}, "objects": {"": [[1, 0, 0, "-", "cyto_dl"]], "cyto_dl.api": [[3, 0, 0, "-", "cyto_dl_model"], [6, 0, 0, "-", "data"], [7, 0, 0, "-", "model"]], "cyto_dl.api.cyto_dl_model": [[4, 0, 0, "-", "cyto_dl_base_model"], [5, 0, 0, "-", "segmentation_plugin_model"]], "cyto_dl.api.cyto_dl_model.cyto_dl_base_model": [[4, 1, 1, "", "CytoDLBaseModel"]], "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel": [[4, 2, 1, "", "from_default_config"], [4, 2, 1, "", "from_existing_config"], [4, 2, 1, "", "get_config"], [4, 2, 1, "", "get_experiment_name"], [4, 2, 1, "", "predict"], [4, 2, 1, "", "save_config"], [4, 2, 1, "", "set_experiment_name"], [4, 2, 1, "", "train"]], "cyto_dl.api.cyto_dl_model.segmentation_plugin_model": [[5, 1, 1, "", "SegmentationPluginModel"]], "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel": [[5, 2, 1, "", "get_hardware_type"], [5, 2, 1, "", "get_input_channel"], [5, 2, 1, "", "get_manifest_column_names"], [5, 2, 1, "", "get_patch_size"], [5, 2, 1, "", "get_raw_image_channels"], [5, 2, 1, "", "get_split_column"], [5, 2, 1, "", "remove_split_column"], [5, 2, 1, "", "set_hardware_type"], [5, 2, 1, "", "set_input_channel"], [5, 2, 1, "", "set_manifest_column_names"], [5, 2, 1, "", "set_patch_size"], [5, 2, 1, "", "set_raw_image_channels"], [5, 2, 1, "", "set_split_column"]], "cyto_dl.api.data": [[6, 1, 1, "", "AutoThresholdMethod"], [6, 1, 1, "", "ExperimentType"], [6, 1, 1, "", "HardwareType"], [6, 1, 1, "", "PatchSize"]], "cyto_dl.api.data.AutoThresholdMethod": [[6, 3, 1, "", "ISODATA"], [6, 3, 1, "", "LI"], [6, 3, 1, "", "LOCAL"], [6, 3, 1, "", "MEAN"], [6, 3, 1, "", "MINIMUM"], [6, 3, 1, "", "MULTIOTSU"], [6, 3, 1, "", "NIBLACK"], [6, 3, 1, "", "OTSU"], [6, 3, 1, "", "SAUVOLA"], [6, 3, 1, "", "TRIANGLE"], [6, 3, 1, "", "YEN"]], "cyto_dl.api.data.ExperimentType": [[6, 3, 1, "", "GAN"], [6, 3, 1, "", "INSTANCE_SEG"], [6, 3, 1, "", "LABEL_FREE"], [6, 3, 1, "", "SEGMENTATION"], [6, 3, 1, "", "SEGMENTATION_ARRAY"], [6, 3, 1, "", "SEGMENTATION_PLUGIN"]], "cyto_dl.api.data.HardwareType": [[6, 3, 1, "", "CPU"], [6, 3, 1, "", "GPU"]], "cyto_dl.api.data.PatchSize": [[6, 3, 1, "", "LARGE"], [6, 3, 1, "", "MEDIUM"], [6, 3, 1, "", "SMALL"]], "cyto_dl.api.model": [[7, 1, 1, "", "CytoDLModel"]], "cyto_dl.api.model.CytoDLModel": [[7, 2, 1, "", "download_example_data"], [7, 2, 1, "", "load_config_from_dict"], [7, 2, 1, "", "load_config_from_file"], [7, 2, 1, "", "load_default_experiment"], [7, 2, 1, "", "override_config"], [7, 2, 1, "", "predict"], [7, 2, 1, "", "print_config"], [7, 2, 1, "", "save_config"], [7, 2, 1, "", "train"]], "cyto_dl": [[8, 0, 0, "-", "callbacks"], [17, 0, 0, "-", "dataframe"], [24, 0, 0, "-", "datamodules"], [35, 0, 0, "-", "eval"], [36, 0, 0, "-", "image"], [58, 0, 0, "-", "loggers"], [60, 0, 0, "-", "models"], [105, 0, 0, "-", "nn"], [156, 0, 0, "-", "point_cloud"], [159, 0, 0, "-", "train"], [160, 0, 0, "-", "utils"]], "cyto_dl.callbacks": [[10, 0, 0, "-", "csv_saver"], [11, 0, 0, "-", "image_saver"], [13, 0, 0, "-", "layer_freeze"], [14, 0, 0, "-", "model_utils"], [15, 0, 0, "-", "outlier_detection"]], "cyto_dl.callbacks.csv_saver": [[10, 1, 1, "", "CSVSaver"], [10, 1, 1, "", "JEPASaver"]], "cyto_dl.callbacks.csv_saver.CSVSaver": [[10, 2, 1, "", "on_predict_epoch_end"], [10, 2, 1, "", "pred_to_csv"], [10, 2, 1, "", "save_feats"]], "cyto_dl.callbacks.csv_saver.JEPASaver": [[10, 2, 1, "", "pred_to_csv"]], "cyto_dl.callbacks.image_saver": [[11, 1, 1, "", "ImageSaver"]], "cyto_dl.callbacks.image_saver.ImageSaver": [[11, 2, 1, "", "on_predict_batch_end"], [11, 2, 1, "", "on_test_batch_end"], [11, 2, 1, "", "on_train_batch_end"], [11, 2, 1, "", "on_validation_batch_end"], [11, 2, 1, "", "save"]], "cyto_dl.callbacks.layer_freeze": [[13, 1, 1, "", "LayerFreeze"]], "cyto_dl.callbacks.layer_freeze.LayerFreeze": [[13, 2, 1, "", "setup"]], "cyto_dl.callbacks.model_utils": [[14, 1, 1, "", "GetEmbeddings"], [14, 4, 1, "", "get_all_embeddings"], [14, 4, 1, "", "save_predictions_classifier"]], "cyto_dl.callbacks.model_utils.GetEmbeddings": [[14, 2, 1, "", "on_test_epoch_end"]], "cyto_dl.callbacks.outlier_detection": [[15, 1, 1, "", "OutlierDetection"]], "cyto_dl.callbacks.outlier_detection.OutlierDetection": [[15, 2, 1, "", "calculate_mahalanobis_hook"], [15, 2, 1, "", "flatten_activations"], [15, 2, 1, "", "on_load_checkpoint"], [15, 2, 1, "", "on_predict_batch_end"], [15, 2, 1, "", "on_predict_epoch_start"], [15, 2, 1, "", "on_save_checkpoint"], [15, 2, 1, "", "on_test_batch_end"], [15, 2, 1, "", "on_test_epoch_start"], [15, 2, 1, "", "on_train_epoch_start"], [15, 2, 1, "", "update_covariance_hook"]], "cyto_dl.dataframe": [[18, 0, 0, "-", "readers"], [19, 0, 0, "-", "transforms"]], "cyto_dl.dataframe.readers": [[18, 4, 1, "", "filter_columns"], [18, 4, 1, "", "read_csv"], [18, 4, 1, "", "read_dataframe"], [18, 4, 1, "", "read_h5ad"], [18, 4, 1, "", "read_parquet"]], "cyto_dl.dataframe.transforms": [[20, 0, 0, "-", "filter"], [21, 0, 0, "-", "group_cols"], [22, 0, 0, "-", "misc"], [23, 0, 0, "-", "split"]], "cyto_dl.dataframe.transforms.filter": [[20, 4, 1, "", "filter_columns"], [20, 4, 1, "", "filter_rows"]], "cyto_dl.dataframe.transforms.group_cols": [[21, 1, 1, "", "GroupCols"]], "cyto_dl.dataframe.transforms.misc": [[22, 4, 1, "", "append_class_weights"], [22, 4, 1, "", "append_labels_to_integers"], [22, 4, 1, "", "append_one_hot"], [22, 4, 1, "", "make_random_df"]], "cyto_dl.dataframe.transforms.split": [[23, 4, 1, "", "sample_n_each"], [23, 4, 1, "", "split_dataframe"]], "cyto_dl.datamodules": [[25, 0, 0, "-", "array"], [26, 0, 0, "-", "data_dict"], [27, 0, 0, "-", "dataframe"], [31, 0, 0, "-", "folder"], [32, 0, 0, "-", "multidim_image"], [33, 0, 0, "-", "smartcache"], [34, 0, 0, "-", "torchvision"]], "cyto_dl.datamodules.array": [[25, 4, 1, "", "make_array_dataloader"]], "cyto_dl.datamodules.data_dict": [[26, 4, 1, "", "make_data_dict_dataloader"]], "cyto_dl.datamodules.dataframe": [[28, 0, 0, "-", "dataframe_datamodule"], [29, 0, 0, "-", "grouped_dataframe_datamodule"], [30, 0, 0, "-", "utils"]], "cyto_dl.datamodules.dataframe.dataframe_datamodule": [[28, 1, 1, "", "DataframeDatamodule"]], "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule": [[28, 2, 1, "", "get_dataloader"], [28, 2, 1, "", "get_dataset"], [28, 2, 1, "", "make_dataloader"], [28, 2, 1, "", "predict_dataloader"], [28, 2, 1, "", "test_dataloader"], [28, 2, 1, "", "train_dataloader"], [28, 2, 1, "", "val_dataloader"]], "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule": [[29, 1, 1, "", "GroupedDataframeDatamodule"]], "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule.GroupedDataframeDatamodule": [[29, 2, 1, "", "make_dataloader"]], "cyto_dl.datamodules.dataframe.utils": [[30, 1, 1, "", "AlternatingBatchSampler"], [30, 1, 1, "", "RemoveNaNKeysd"], [30, 4, 1, "", "get_canonical_split_name"], [30, 4, 1, "", "get_dataset"], [30, 4, 1, "", "make_multiple_dataframe_splits"], [30, 4, 1, "", "make_single_dataframe_splits"], [30, 4, 1, "", "parse_transforms"]], "cyto_dl.datamodules.folder": [[31, 4, 1, "", "make_folder_dataloader"]], "cyto_dl.datamodules.multidim_image": [[32, 1, 1, "", "MultiDimImageDataset"], [32, 4, 1, "", "make_multidim_image_dataloader"]], "cyto_dl.datamodules.multidim_image.MultiDimImageDataset": [[32, 2, 1, "", "create_metatensor"], [32, 2, 1, "", "get_per_file_args"], [32, 2, 1, "", "is_batch"]], "cyto_dl.datamodules.smartcache": [[33, 1, 1, "", "SmartcacheDatamodule"]], "cyto_dl.datamodules.smartcache.SmartcacheDatamodule": [[33, 2, 1, "", "get_per_file_args"], [33, 2, 1, "", "make_dataloader"], [33, 2, 1, "", "predict_dataloader"], [33, 2, 1, "", "prepare_data"], [33, 2, 1, "", "setup"], [33, 2, 1, "", "test_dataloader"], [33, 2, 1, "", "train_dataloader"], [33, 2, 1, "", "val_dataloader"]], "cyto_dl.datamodules.torchvision": [[34, 1, 1, "", "CIFAR10"], [34, 1, 1, "", "CIFAR100"], [34, 1, 1, "", "CLEVRClassification"], [34, 1, 1, "", "Caltech101"], [34, 1, 1, "", "Caltech256"], [34, 1, 1, "", "CelebA"], [34, 1, 1, "", "Cityscapes"], [34, 1, 1, "", "CocoCaptions"], [34, 1, 1, "", "CocoDetection"], [34, 1, 1, "", "Country211"], [34, 1, 1, "", "DTD"], [34, 1, 1, "", "DatasetFolder"], [34, 1, 1, "", "EMNIST"], [34, 1, 1, "", "EuroSAT"], [34, 1, 1, "", "FER2013"], [34, 1, 1, "", "FGVCAircraft"], [34, 1, 1, "", "FakeData"], [34, 1, 1, "", "FashionMNIST"], [34, 1, 1, "", "Flickr30k"], [34, 1, 1, "", "Flickr8k"], [34, 1, 1, "", "Flowers102"], [34, 1, 1, "", "Food101"], [34, 1, 1, "", "GTSRB"], [34, 1, 1, "", "HMDB51"], [34, 1, 1, "", "INaturalist"], [34, 1, 1, "", "ImageFolder"], [34, 1, 1, "", "ImageNet"], [34, 1, 1, "", "KMNIST"], [34, 1, 1, "", "Kinetics"], [34, 1, 1, "", "Kitti"], [34, 1, 1, "", "LFWPairs"], [34, 1, 1, "", "LFWPeople"], [34, 1, 1, "", "LSUN"], [34, 1, 1, "", "LSUNClass"], [34, 1, 1, "", "MNIST"], [34, 1, 1, "", "MovingMNIST"], [34, 1, 1, "", "Omniglot"], [34, 1, 1, "", "OxfordIIITPet"], [34, 1, 1, "", "PCAM"], [34, 1, 1, "", "PhotoTour"], [34, 1, 1, "", "Places365"], [34, 1, 1, "", "QMNIST"], [34, 1, 1, "", "RenderedSST2"], [34, 1, 1, "", "SBDataset"], [34, 1, 1, "", "SBU"], [34, 1, 1, "", "SEMEION"], [34, 1, 1, "", "STL10"], [34, 1, 1, "", "SUN397"], [34, 1, 1, "", "SVHN"], [34, 1, 1, "", "StanfordCars"], [34, 1, 1, "", "UCF101"], [34, 1, 1, "", "USPS"], [34, 1, 1, "", "VOCDetection"], [34, 1, 1, "", "VOCSegmentation"], [34, 1, 1, "", "VisionDataset"], [34, 1, 1, "", "WIDERFace"]], "cyto_dl.eval": [[35, 4, 1, "", "main"]], "cyto_dl.image": [[37, 0, 0, "-", "io"], [44, 0, 0, "-", "transforms"]], "cyto_dl.image.io": [[38, 0, 0, "-", "aicsimage_loader"], [39, 0, 0, "-", "monai_bio_reader"], [40, 0, 0, "-", "numpy_reader"], [41, 0, 0, "-", "ome_zarr_reader"], [42, 0, 0, "-", "polygon_loader"], [43, 0, 0, "-", "skimage_reader"]], "cyto_dl.image.io.aicsimage_loader": [[38, 1, 1, "", "AICSImageLoaderd"]], "cyto_dl.image.io.aicsimage_loader.AICSImageLoaderd": [[38, 2, 1, "", "split_args"]], "cyto_dl.image.io.monai_bio_reader": [[39, 1, 1, "", "MonaiBioReader"]], "cyto_dl.image.io.monai_bio_reader.MonaiBioReader": [[39, 2, 1, "", "get_data"], [39, 2, 1, "", "read"], [39, 2, 1, "", "verify_suffix"]], "cyto_dl.image.io.numpy_reader": [[40, 1, 1, "", "ReadNumpyFile"]], "cyto_dl.image.io.ome_zarr_reader": [[41, 1, 1, "", "OmeZarrReader"]], "cyto_dl.image.io.ome_zarr_reader.OmeZarrReader": [[41, 2, 1, "", "get_data"], [41, 2, 1, "", "read"], [41, 2, 1, "", "verify_suffix"]], "cyto_dl.image.io.polygon_loader": [[42, 1, 1, "", "PolygonLoaderd"]], "cyto_dl.image.io.skimage_reader": [[43, 1, 1, "", "SkimageReader"]], "cyto_dl.image.io.skimage_reader.SkimageReader": [[43, 2, 1, "", "get_data"], [43, 2, 1, "", "read"], [43, 2, 1, "", "verify_suffix"]], "cyto_dl.image.transforms": [[45, 0, 0, "-", "add_meta"], [46, 0, 0, "-", "bright_sampler"], [47, 0, 0, "-", "clip"], [48, 0, 0, "-", "contrastadjust"], [49, 0, 0, "-", "generate_jepa_masks"], [50, 0, 0, "-", "merge"], [51, 0, 0, "-", "multiscale_cropper"], [52, 0, 0, "-", "pad"], [53, 0, 0, "-", "project"], [55, 0, 0, "-", "save"], [57, 0, 0, "-", "track_transforms"]], "cyto_dl.image.transforms.add_meta": [[45, 1, 1, "", "AddMeta"]], "cyto_dl.image.transforms.bright_sampler": [[46, 1, 1, "", "BrightSampler"]], "cyto_dl.image.transforms.clip": [[47, 1, 1, "", "Clip"], [47, 1, 1, "", "Clipd"]], "cyto_dl.image.transforms.contrastadjust": [[48, 1, 1, "", "ContrastAdjust"], [48, 1, 1, "", "ContrastAdjustd"]], "cyto_dl.image.transforms.generate_jepa_masks": [[49, 1, 1, "", "JEPAMaskGenerator"]], "cyto_dl.image.transforms.generate_jepa_masks.JEPAMaskGenerator": [[49, 2, 1, "", "remove_excess_pixels"]], "cyto_dl.image.transforms.merge": [[50, 1, 1, "", "Merged"]], "cyto_dl.image.transforms.multiscale_cropper": [[51, 1, 1, "", "RandomMultiScaleCropd"]], "cyto_dl.image.transforms.multiscale_cropper.RandomMultiScaleCropd": [[51, 2, 1, "", "generate_slices"]], "cyto_dl.image.transforms.pad": [[52, 1, 1, "", "PadZd"]], "cyto_dl.image.transforms.project": [[53, 1, 1, "", "MaxProjectd"]], "cyto_dl.image.transforms.save": [[55, 1, 1, "", "Save"], [55, 1, 1, "", "Saved"]], "cyto_dl.image.transforms.track_transforms": [[57, 1, 1, "", "CropResize"], [57, 1, 1, "", "GenerateTrackLabels"], [57, 1, 1, "", "PerChannel"], [57, 1, 1, "", "SplitTrackd"]], "cyto_dl.loggers": [[59, 0, 0, "-", "mlflow"]], "cyto_dl.loggers.mlflow": [[59, 1, 1, "", "MLFlowLogger"]], "cyto_dl.loggers.mlflow.MLFlowLogger": [[59, 2, 1, "", "after_save_checkpoint"], [59, 2, 1, "", "log_hyperparams"], [59, 2, 1, "", "log_metrics"]], "cyto_dl.models": [[61, 0, 0, "-", "base_model"], [62, 0, 0, "-", "basic_model"], [63, 0, 0, "-", "classification"], [66, 0, 0, "-", "contrastive"], [72, 0, 0, "-", "im2im"], [82, 0, 0, "-", "jepa"], [86, 0, 0, "-", "utils"]], "cyto_dl.models.base_model": [[61, 1, 1, "", "BaseModel"], [61, 1, 1, "", "BaseModelMeta"]], "cyto_dl.models.base_model.BaseModel": [[61, 2, 1, "", "compute_metrics"], [61, 2, 1, "", "configure_optimizers"], [61, 2, 1, "", "forward"], [61, 2, 1, "", "model_step"], [61, 2, 1, "", "on_train_start"], [61, 2, 1, "", "parse_batch"], [61, 2, 1, "", "predict_step"], [61, 2, 1, "", "test_step"], [61, 2, 1, "", "training_step"], [61, 2, 1, "", "validation_step"]], "cyto_dl.models.basic_model": [[62, 1, 1, "", "BasicModel"]], "cyto_dl.models.basic_model.BasicModel": [[62, 2, 1, "", "forward"], [62, 2, 1, "", "model_step"]], "cyto_dl.models.classification": [[64, 0, 0, "-", "classification"], [65, 0, 0, "-", "timepoint_classification"]], "cyto_dl.models.classification.classification": [[64, 1, 1, "", "Classification"]], "cyto_dl.models.classification.classification.Classification": [[64, 2, 1, "", "forward"], [64, 2, 1, "", "model_step"], [64, 2, 1, "", "predict_step"], [64, 2, 1, "", "save_images"], [64, 2, 1, "", "should_save_image"]], "cyto_dl.models.classification.timepoint_classification": [[65, 1, 1, "", "TimepointClassification"]], "cyto_dl.models.classification.timepoint_classification.TimepointClassification": [[65, 2, 1, "", "predict_step"]], "cyto_dl.models.contrastive": [[67, 0, 0, "-", "contrastive"]], "cyto_dl.models.contrastive.contrastive": [[67, 1, 1, "", "Contrastive"]], "cyto_dl.models.contrastive.contrastive.Contrastive": [[67, 2, 1, "", "forward"], [67, 2, 1, "", "model_step"], [67, 2, 1, "", "plot_classes"], [67, 2, 1, "", "plot_neighbors"], [67, 2, 1, "", "predict_step"]], "cyto_dl.models.im2im": [[73, 0, 0, "-", "gan"], [74, 0, 0, "-", "multi_task"], [75, 0, 0, "-", "utils"]], "cyto_dl.models.im2im.gan": [[73, 1, 1, "", "GAN"]], "cyto_dl.models.im2im.gan.GAN": [[73, 2, 1, "", "configure_optimizers"], [73, 2, 1, "", "model_step"], [73, 2, 1, "", "predict_step"]], "cyto_dl.models.im2im.multi_task": [[74, 1, 1, "", "MultiTaskIm2Im"]], "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im": [[74, 2, 1, "", "configure_optimizers"], [74, 2, 1, "", "forward"], [74, 2, 1, "", "get_n_postprocess_image"], [74, 2, 1, "", "get_per_head"], [74, 2, 1, "", "model_step"], [74, 2, 1, "", "predict_step"], [74, 2, 1, "", "run_forward"], [74, 2, 1, "", "test_step"], [74, 2, 1, "", "training_step"], [74, 2, 1, "", "validation_step"]], "cyto_dl.models.im2im.utils": [[76, 0, 0, "-", "instance_seg"], [77, 0, 0, "-", "noise_annealer"], [78, 0, 0, "-", "postprocessing"]], "cyto_dl.models.im2im.utils.instance_seg": [[76, 1, 1, "", "InstanceSegCluster"], [76, 1, 1, "", "InstanceSegLoss"], [76, 1, 1, "", "InstanceSegPreprocessd"], [76, 1, 1, "", "InstanceSegRandFlipd"], [76, 4, 1, "", "pad_slice"]], "cyto_dl.models.im2im.utils.instance_seg.InstanceSegCluster": [[76, 2, 1, "", "cluster_object"], [76, 2, 1, "", "kd_clustering"], [76, 2, 1, "", "remove_small_skeletons"]], "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd": [[76, 2, 1, "", "embed_from_skel"], [76, 2, 1, "", "keep_largest_cc"], [76, 2, 1, "", "label_2d"], [76, 2, 1, "", "shrink"], [76, 2, 1, "", "skeleton_tall"], [76, 2, 1, "", "smooth_embedding"], [76, 2, 1, "", "topology_preserving_thinning"]], "cyto_dl.models.im2im.utils.noise_annealer": [[77, 1, 1, "", "NoiseAnnealer"]], "cyto_dl.models.im2im.utils.noise_annealer.NoiseAnnealer": [[77, 2, 1, "", "update_noise"]], "cyto_dl.models.im2im.utils.postprocessing": [[79, 0, 0, "-", "act_thresh_label"], [80, 0, 0, "-", "auto_thresh"], [81, 0, 0, "-", "dict_to_im"]], "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label": [[79, 1, 1, "", "ActThreshLabel"]], "cyto_dl.models.im2im.utils.postprocessing.auto_thresh": [[80, 1, 1, "", "AutoThreshold"]], "cyto_dl.models.im2im.utils.postprocessing.dict_to_im": [[81, 1, 1, "", "DictToIm"], [81, 4, 1, "", "detach"]], "cyto_dl.models.jepa": [[83, 0, 0, "-", "ijepa"], [84, 0, 0, "-", "iwm"], [85, 0, 0, "-", "jepa_base"]], "cyto_dl.models.jepa.ijepa": [[83, 1, 1, "", "IJEPA"]], "cyto_dl.models.jepa.ijepa.IJEPA": [[83, 2, 1, "", "model_step"]], "cyto_dl.models.jepa.iwm": [[84, 1, 1, "", "IWM"]], "cyto_dl.models.jepa.iwm.IWM": [[84, 2, 1, "", "extract_embeddings"], [84, 2, 1, "", "get_predict_masks"], [84, 2, 1, "", "model_step"], [84, 2, 1, "", "predict_step"], [84, 2, 1, "", "test_step"]], "cyto_dl.models.jepa.jepa_base": [[85, 1, 1, "", "JEPABase"]], "cyto_dl.models.jepa.jepa_base.JEPABase": [[85, 2, 1, "", "configure_optimizers"], [85, 2, 1, "", "forward"], [85, 2, 1, "", "get_context_embeddings"], [85, 2, 1, "", "get_mask"], [85, 2, 1, "", "get_target_embeddings"], [85, 2, 1, "", "model_step"], [85, 2, 1, "", "predict_step"], [85, 2, 1, "", "remove_first_dim"], [85, 2, 1, "", "update_teacher"]], "cyto_dl.models.utils": [[87, 0, 0, "-", "mlflow"], [88, 0, 0, "-", "utils"]], "cyto_dl.models.utils.mlflow": [[87, 4, 1, "", "get_config"], [87, 4, 1, "", "load_model_from_checkpoint"]], "cyto_dl.models.utils.utils": [[88, 4, 1, "", "find_indices"]], "cyto_dl.models.vae": [[90, 0, 0, "-", "base_vae"], [99, 0, 0, "-", "priors"]], "cyto_dl.models.vae.base_vae": [[90, 1, 1, "", "BaseVAE"]], "cyto_dl.models.vae.base_vae.BaseVAE": [[90, 2, 1, "", "calculate_elbo"], [90, 2, 1, "", "calculate_rcl"], [90, 2, 1, "", "calculate_rcl_dict"], [90, 2, 1, "", "decode"], [90, 2, 1, "", "encode"], [90, 2, 1, "", "forward"], [90, 2, 1, "", "model_step"], [90, 2, 1, "", "sample_z"]], "cyto_dl.models.vae.priors": [[100, 0, 0, "-", "abstract_prior"], [101, 0, 0, "-", "gaussian"], [102, 0, 0, "-", "identity_prior"], [103, 0, 0, "-", "joint_prior"]], "cyto_dl.models.vae.priors.abstract_prior": [[100, 1, 1, "", "Prior"]], "cyto_dl.models.vae.priors.abstract_prior.Prior": [[100, 2, 1, "", "forward"], [100, 2, 1, "", "kl_divergence"], [100, 5, 1, "", "param_size"], [100, 2, 1, "", "sample"]], "cyto_dl.models.vae.priors.gaussian": [[101, 1, 1, "", "DiagonalGaussianPrior"], [101, 1, 1, "", "IsotropicGaussianPrior"], [101, 4, 1, "", "compute_tc_penalty"]], "cyto_dl.models.vae.priors.gaussian.DiagonalGaussianPrior": [[101, 2, 1, "", "forward"], [101, 2, 1, "", "kl_divergence"], [101, 5, 1, "", "param_size"]], "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior": [[101, 2, 1, "", "forward"], [101, 2, 1, "", "kl_divergence"], [101, 5, 1, "", "param_size"], [101, 2, 1, "", "sample"]], "cyto_dl.models.vae.priors.identity_prior": [[102, 1, 1, "", "IdentityPrior"]], "cyto_dl.models.vae.priors.identity_prior.IdentityPrior": [[102, 2, 1, "", "forward"], [102, 5, 1, "", "param_size"]], "cyto_dl.models.vae.priors.joint_prior": [[103, 1, 1, "", "JointPrior"]], "cyto_dl.models.vae.priors.joint_prior.JointPrior": [[103, 2, 1, "", "forward"], [103, 2, 1, "", "kl_divergence"], [103, 5, 1, "", "param_size"], [103, 2, 1, "", "sample"]], "cyto_dl.nn": [[106, 0, 0, "-", "discriminators"], [109, 0, 0, "-", "head"], [117, 0, 0, "-", "hr_skip"], [118, 0, 0, "-", "losses"], [132, 0, 0, "-", "mlp"], [138, 0, 0, "-", "res_unit"], [139, 0, 0, "-", "spatial_transformer"], [140, 0, 0, "-", "track_sequence_predictor"], [141, 0, 0, "-", "vits"]], "cyto_dl.nn.discriminators": [[107, 0, 0, "-", "multi_scale_discriminator"], [108, 0, 0, "-", "n_layer_discriminator"]], "cyto_dl.nn.discriminators.multi_scale_discriminator": [[107, 1, 1, "", "MultiScaleDiscriminator"]], "cyto_dl.nn.discriminators.multi_scale_discriminator.MultiScaleDiscriminator": [[107, 2, 1, "", "forward"]], "cyto_dl.nn.discriminators.n_layer_discriminator": [[108, 1, 1, "", "NLayerDiscriminator"]], "cyto_dl.nn.discriminators.n_layer_discriminator.NLayerDiscriminator": [[108, 2, 1, "", "forward"]], "cyto_dl.nn.head": [[110, 0, 0, "-", "base_head"], [111, 0, 0, "-", "gan_head"], [112, 0, 0, "-", "gan_head_superres"], [113, 0, 0, "-", "mae_head"], [114, 0, 0, "-", "mask_head"], [115, 0, 0, "-", "res_blocks_head"], [116, 0, 0, "-", "vic_reg"]], "cyto_dl.nn.head.base_head": [[110, 1, 1, "", "BaseHead"]], "cyto_dl.nn.head.base_head.BaseHead": [[110, 2, 1, "", "forward"], [110, 2, 1, "", "generate_io_map"], [110, 2, 1, "", "run_head"], [110, 2, 1, "", "update_params"]], "cyto_dl.nn.head.gan_head": [[111, 1, 1, "", "GANHead"]], "cyto_dl.nn.head.gan_head.GANHead": [[111, 2, 1, "", "forward"], [111, 2, 1, "", "run_head"]], "cyto_dl.nn.head.gan_head_superres": [[112, 1, 1, "", "GANHead_resize"]], "cyto_dl.nn.head.gan_head_superres.GANHead_resize": [[112, 2, 1, "", "forward"]], "cyto_dl.nn.head.mae_head": [[113, 1, 1, "", "MAEHead"]], "cyto_dl.nn.head.mae_head.MAEHead": [[113, 2, 1, "", "run_head"]], "cyto_dl.nn.head.mask_head": [[114, 1, 1, "", "MaskHead"]], "cyto_dl.nn.head.mask_head.MaskHead": [[114, 2, 1, "", "run_head"]], "cyto_dl.nn.head.res_blocks_head": [[115, 1, 1, "", "ResBlocksHead"]], "cyto_dl.nn.head.res_blocks_head.ResBlocksHead": [[115, 2, 1, "", "forward"]], "cyto_dl.nn.head.vic_reg": [[116, 1, 1, "", "VICRegHead"]], "cyto_dl.nn.head.vic_reg.VICRegHead": [[116, 2, 1, "", "run_head"]], "cyto_dl.nn.hr_skip": [[117, 1, 1, "", "HRSkip"]], "cyto_dl.nn.hr_skip.HRSkip": [[117, 2, 1, "", "forward"]], "cyto_dl.nn.losses": [[119, 0, 0, "-", "adversarial_loss"], [120, 0, 0, "-", "chamfer_loss"], [121, 0, 0, "-", "continuous_bernoulli"], [122, 0, 0, "-", "cosine_loss"], [123, 0, 0, "-", "gan_loss"], [124, 0, 0, "-", "gaussian_nll_loss"], [126, 0, 0, "-", "loss_wrapper"], [128, 0, 0, "-", "threshold_loss"], [129, 0, 0, "-", "vic_reg"], [130, 0, 0, "-", "weibull"], [131, 0, 0, "-", "weighted_mse_loss"]], "cyto_dl.nn.losses.adversarial_loss": [[119, 1, 1, "", "AdversarialLoss"]], "cyto_dl.nn.losses.adversarial_loss.AdversarialLoss": [[119, 2, 1, "", "forward"]], "cyto_dl.nn.losses.chamfer_loss": [[120, 1, 1, "", "ChamferLoss"]], "cyto_dl.nn.losses.chamfer_loss.ChamferLoss": [[120, 2, 1, "", "batch_pairwise_dist"], [120, 2, 1, "", "batch_pairwise_dist2"], [120, 2, 1, "", "forward"]], "cyto_dl.nn.losses.continuous_bernoulli": [[121, 1, 1, "", "CBLogLoss"]], "cyto_dl.nn.losses.continuous_bernoulli.CBLogLoss": [[121, 2, 1, "", "forward"]], "cyto_dl.nn.losses.cosine_loss": [[122, 1, 1, "", "CosineLoss"]], "cyto_dl.nn.losses.cosine_loss.CosineLoss": [[122, 2, 1, "", "forward"]], "cyto_dl.nn.losses.gan_loss": [[123, 1, 1, "", "GANLoss"], [123, 1, 1, "", "Pix2PixHD"]], "cyto_dl.nn.losses.gan_loss.GANLoss": [[123, 2, 1, "", "get_target_tensor"]], "cyto_dl.nn.losses.gan_loss.Pix2PixHD": [[123, 2, 1, "", "get_feature_matching_loss"], [123, 2, 1, "", "get_gan_loss"]], "cyto_dl.nn.losses.gaussian_nll_loss": [[124, 1, 1, "", "GaussianNLLLoss"]], "cyto_dl.nn.losses.gaussian_nll_loss.GaussianNLLLoss": [[124, 2, 1, "", "forward"]], "cyto_dl.nn.losses.loss_wrapper": [[126, 1, 1, "", "CMAP_loss"], [126, 1, 1, "", "LossWrapper"]], "cyto_dl.nn.losses.threshold_loss": [[128, 1, 1, "", "ThresholdLoss"]], "cyto_dl.nn.losses.vic_reg": [[129, 1, 1, "", "VICRegLoss"], [129, 4, 1, "", "off_diagonal"]], "cyto_dl.nn.losses.vic_reg.VICRegLoss": [[129, 2, 1, "", "forward"]], "cyto_dl.nn.losses.weibull": [[130, 1, 1, "", "WeibullLogLoss"], [130, 4, 1, "", "weibull_log_probs"]], "cyto_dl.nn.losses.weibull.WeibullLogLoss": [[130, 2, 1, "", "forward"]], "cyto_dl.nn.losses.weighted_mse_loss": [[131, 1, 1, "", "WeightedMSELoss"]], "cyto_dl.nn.losses.weighted_mse_loss.WeightedMSELoss": [[131, 2, 1, "", "forward"]], "cyto_dl.nn.mlp": [[132, 1, 1, "", "MLP"]], "cyto_dl.nn.mlp.MLP": [[132, 2, 1, "", "forward"]], "cyto_dl.nn.res_unit": [[138, 1, 1, "", "ResidualUnit"]], "cyto_dl.nn.res_unit.ResidualUnit": [[138, 2, 1, "", "forward"]], "cyto_dl.nn.spatial_transformer": [[139, 1, 1, "", "ConvPoolReLU"], [139, 1, 1, "", "STN"]], "cyto_dl.nn.spatial_transformer.ConvPoolReLU": [[139, 2, 1, "", "forward"]], "cyto_dl.nn.spatial_transformer.STN": [[139, 2, 1, "", "forward"], [139, 2, 1, "", "toggle"]], "cyto_dl.nn.track_sequence_predictor": [[140, 1, 1, "", "TrackClassifier"], [140, 4, 1, "", "positionalencoding1d"]], "cyto_dl.nn.track_sequence_predictor.TrackClassifier": [[140, 2, 1, "", "forward"]], "cyto_dl.nn.vits": [[142, 0, 0, "-", "blocks"], [150, 0, 0, "-", "decoder"], [151, 0, 0, "-", "encoder"], [152, 0, 0, "-", "mae"], [153, 0, 0, "-", "predictor"], [154, 0, 0, "-", "seg"], [155, 0, 0, "-", "utils"]], "cyto_dl.nn.vits.blocks": [[143, 0, 0, "-", "cross_attention"], [144, 0, 0, "-", "intermediate_weigher"], [145, 0, 0, "-", "masked_unit_attention"], [146, 0, 0, "-", "patchify"]], "cyto_dl.nn.vits.blocks.cross_attention": [[143, 1, 1, "", "CrossAttention"], [143, 1, 1, "", "CrossAttentionBlock"], [143, 1, 1, "", "CrossSelfBlock"], [143, 1, 1, "", "Mlp"]], "cyto_dl.nn.vits.blocks.cross_attention.CrossAttention": [[143, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.cross_attention.CrossAttentionBlock": [[143, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.cross_attention.CrossSelfBlock": [[143, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.cross_attention.Mlp": [[143, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.intermediate_weigher": [[144, 1, 1, "", "IntermediateWeigher"]], "cyto_dl.nn.vits.blocks.intermediate_weigher.IntermediateWeigher": [[144, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.masked_unit_attention": [[145, 1, 1, "", "HieraBlock"], [145, 1, 1, "", "MaskUnitAttention"]], "cyto_dl.nn.vits.blocks.masked_unit_attention.HieraBlock": [[145, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.masked_unit_attention.MaskUnitAttention": [[145, 2, 1, "", "forward"]], "cyto_dl.nn.vits.blocks.patchify": [[147, 0, 0, "-", "patchify"], [148, 0, 0, "-", "patchify_base"], [149, 0, 0, "-", "patchify_hiera"]], "cyto_dl.nn.vits.blocks.patchify.patchify": [[147, 1, 1, "", "Patchify"]], "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify": [[147, 2, 1, "", "create_img2token"], [147, 2, 1, "", "extract_visible_tokens"], [147, 2, 1, "", "get_mask_args"], [147, 5, 1, "", "img2token"]], "cyto_dl.nn.vits.blocks.patchify.patchify_base": [[148, 1, 1, "", "PatchifyBase"]], "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase": [[148, 2, 1, "", "create_conv"], [148, 2, 1, "", "create_patch2img"], [148, 2, 1, "", "extract_visible_tokens"], [148, 2, 1, "", "forward"], [148, 2, 1, "", "get_mask"], [148, 2, 1, "", "get_mask_args"], [148, 5, 1, "", "img2token"]], "cyto_dl.nn.vits.blocks.patchify.patchify_hiera": [[149, 1, 1, "", "PatchifyHiera"], [149, 4, 1, "", "take_indexes_mask"]], "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera": [[149, 2, 1, "", "create_img2token"], [149, 2, 1, "", "extract_visible_tokens"], [149, 2, 1, "", "get_mask_args"], [149, 5, 1, "", "img2token"]], "cyto_dl.nn.vits.decoder": [[150, 1, 1, "", "CrossMAE_Decoder"], [150, 1, 1, "", "MAE_Decoder"]], "cyto_dl.nn.vits.decoder.CrossMAE_Decoder": [[150, 2, 1, "", "forward"]], "cyto_dl.nn.vits.decoder.MAE_Decoder": [[150, 2, 1, "", "add_mask_tokens"], [150, 2, 1, "", "adjust_indices_for_cls"], [150, 2, 1, "", "forward"], [150, 2, 1, "", "init_weight"]], "cyto_dl.nn.vits.encoder": [[151, 1, 1, "", "HieraEncoder"], [151, 1, 1, "", "JEPAEncoder"], [151, 1, 1, "", "MAE_Encoder"], [151, 1, 1, "", "SpatialMerger"]], "cyto_dl.nn.vits.encoder.HieraEncoder": [[151, 2, 1, "", "forward"]], "cyto_dl.nn.vits.encoder.JEPAEncoder": [[151, 2, 1, "", "forward"]], "cyto_dl.nn.vits.encoder.MAE_Encoder": [[151, 2, 1, "", "forward"], [151, 2, 1, "", "init_weight"]], "cyto_dl.nn.vits.encoder.SpatialMerger": [[151, 2, 1, "", "forward"]], "cyto_dl.nn.vits.mae": [[152, 1, 1, "", "HieraMAE"], [152, 1, 1, "", "MAE"], [152, 1, 1, "", "MAE_Base"]], "cyto_dl.nn.vits.mae.HieraMAE": [[152, 5, 1, "", "decoder"], [152, 5, 1, "", "encoder"]], "cyto_dl.nn.vits.mae.MAE": [[152, 5, 1, "", "decoder"], [152, 5, 1, "", "encoder"]], "cyto_dl.nn.vits.mae.MAE_Base": [[152, 5, 1, "", "decoder"], [152, 5, 1, "", "encoder"], [152, 2, 1, "", "forward"], [152, 2, 1, "", "init_decoder"], [152, 2, 1, "", "init_encoder"]], "cyto_dl.nn.vits.predictor": [[153, 1, 1, "", "IWMPredictor"], [153, 1, 1, "", "JEPAPredictor"]], "cyto_dl.nn.vits.predictor.IWMPredictor": [[153, 2, 1, "", "forward"]], "cyto_dl.nn.vits.predictor.JEPAPredictor": [[153, 2, 1, "", "forward"], [153, 2, 1, "", "init_weight"], [153, 2, 1, "", "predict_target_features"]], "cyto_dl.nn.vits.seg": [[154, 1, 1, "", "EncodedSkip"], [154, 1, 1, "", "Seg_ViT"], [154, 1, 1, "", "SuperresDecoder"]], "cyto_dl.nn.vits.seg.EncodedSkip": [[154, 2, 1, "", "forward"]], "cyto_dl.nn.vits.seg.Seg_ViT": [[154, 2, 1, "", "forward"]], "cyto_dl.nn.vits.seg.SuperresDecoder": [[154, 2, 1, "", "forward"]], "cyto_dl.nn.vits.utils": [[155, 4, 1, "", "get_positional_embedding"], [155, 4, 1, "", "match_tuple_dimensions"], [155, 4, 1, "", "random_indexes"], [155, 4, 1, "", "take_indexes"]], "cyto_dl.train": [[159, 4, 1, "", "main"]], "cyto_dl.utils": [[161, 0, 0, "-", "array"], [162, 0, 0, "-", "checkpoint"], [163, 0, 0, "-", "config"], [164, 0, 0, "-", "download_test_data"], [165, 0, 0, "-", "dummy_dataset"], [166, 0, 0, "-", "pylogger"], [167, 0, 0, "-", "rich_utils"], [168, 0, 0, "-", "rotation"], [172, 0, 0, "-", "template_utils"]], "cyto_dl.utils.array": [[161, 4, 1, "", "create_dataloader"], [161, 4, 1, "", "extract_array_predictions"]], "cyto_dl.utils.checkpoint": [[162, 4, 1, "", "load_checkpoint"]], "cyto_dl.utils.config": [[163, 4, 1, "", "is_config"], [163, 4, 1, "", "kv_to_dict"], [163, 4, 1, "", "remove_aux_key"]], "cyto_dl.utils.download_test_data": [[164, 4, 1, "", "delete_test_data"], [164, 4, 1, "", "download_test_data"], [164, 4, 1, "", "parse_s3_path"], [164, 4, 1, "", "setup_paths"]], "cyto_dl.utils.dummy_dataset": [[165, 1, 1, "", "DummyDatamodule"], [165, 1, 1, "", "DummyDataset"]], "cyto_dl.utils.dummy_dataset.DummyDatamodule": [[165, 2, 1, "", "get_dataloader"], [165, 2, 1, "", "predict_dataloader"], [165, 2, 1, "", "test_dataloader"], [165, 2, 1, "", "train_dataloader"], [165, 2, 1, "", "val_dataloader"]], "cyto_dl.utils.dummy_dataset.DummyDataset": [[165, 2, 1, "", "generate_img"]], "cyto_dl.utils.pylogger": [[166, 4, 1, "", "get_pylogger"]], "cyto_dl.utils.rich_utils": [[167, 4, 1, "", "enforce_tags"], [167, 4, 1, "", "print_config_tree"]], "cyto_dl.utils.rotation": [[168, 1, 1, "", "RotationModule"], [168, 4, 1, "", "get_rotation_matrix_so2"], [168, 4, 1, "", "get_rotation_matrix_so3"]], "cyto_dl.utils.rotation.RotationModule": [[168, 2, 1, "", "compute_rotation_matrix"]], "cyto_dl.utils.template_utils": [[172, 4, 1, "", "close_loggers"], [172, 4, 1, "", "extras"], [172, 4, 1, "", "get_metric_value"], [172, 4, 1, "", "instantiate_callbacks"], [172, 4, 1, "", "instantiate_loggers"], [172, 4, 1, "", "log_hyperparameters"], [172, 4, 1, "", "save_file"], [172, 4, 1, "", "task_wrapper"]]}, "objtypes": {"0": "py:module", "1": "py:class", "2": "py:method", "3": "py:attribute", "4": "py:function", "5": "py:property"}, "objnames": {"0": ["py", "module", "Python module"], "1": ["py", "class", "Python class"], "2": ["py", "method", "Python method"], "3": ["py", "attribute", "Python attribute"], "4": ["py", "function", "Python function"], "5": ["py", "property", "Python property"]}, "titleterms": {"us": 0, "api": [0, 2, 3, 4, 5, 6, 7], "load": 0, "config": [0, 163, 177], "train": [0, 159, 177], "predict": [0, 177], "util": [0, 30, 75, 76, 77, 78, 79, 80, 81, 86, 87, 88, 155, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172], "cyto_dl": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 175], "packag": [1, 2, 3, 8, 17, 19, 24, 27, 36, 37, 44, 58, 60, 63, 66, 68, 72, 75, 78, 82, 86, 89, 95, 99, 105, 106, 109, 118, 133, 141, 142, 146, 156, 157, 160, 169], "subpackag": [1, 2, 17, 24, 36, 60, 72, 75, 89, 105, 141, 142, 156, 160], "submodul": [1, 2, 3, 8, 17, 19, 24, 27, 37, 44, 58, 60, 63, 66, 68, 72, 75, 78, 82, 86, 89, 95, 99, 105, 106, 109, 118, 133, 141, 142, 146, 157, 160, 169], "cyto_dl_model": [3, 4, 5], "cyto_dl_base_model": 4, "modul": [4, 5, 6, 7, 9, 10, 11, 12, 13, 14, 15, 16, 18, 20, 21, 22, 23, 25, 26, 28, 29, 30, 31, 32, 33, 34, 35, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 59, 61, 62, 64, 65, 67, 69, 70, 71, 73, 74, 76, 77, 79, 80, 81, 83, 84, 85, 87, 88, 90, 91, 92, 93, 94, 96, 97, 98, 100, 101, 102, 103, 104, 107, 108, 110, 111, 112, 113, 114, 115, 116, 117, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 137, 138, 139, 140, 143, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 158, 159, 161, 162, 163, 164, 165, 166, 167, 168, 170, 171, 172], "segmentation_plugin_model": 5, "data": [6, 173, 176], "model": [7, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 173, 176], "callback": [8, 9, 10, 11, 12, 13, 14, 15], "callback_util": 9, "csv_saver": 10, "image_sav": 11, "latent_walk": 12, "layer_freez": 13, "model_util": 14, "outlier_detect": 15, "compil": 16, "datafram": [17, 18, 19, 20, 21, 22, 23, 27, 28, 29, 30], "reader": 18, "transform": [19, 20, 21, 22, 23, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57], "filter": 20, "group_col": 21, "misc": 22, "split": 23, "datamodul": [24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34], "arrai": [25, 161], "data_dict": 26, "dataframe_datamodul": 28, "grouped_dataframe_datamodul": 29, "folder": 31, "multidim_imag": 32, "smartcach": 33, "torchvis": 34, "eval": 35, "imag": [36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57], "io": [37, 38, 39, 40, 41, 42, 43, 157, 158], "aicsimage_load": 38, "monai_bio_read": 39, "numpy_read": 40, "ome_zarr_read": 41, "polygon_load": 42, "skimage_read": 43, "add_meta": 45, "bright_sampl": 46, "clip": 47, "contrastadjust": 48, "generate_jepa_mask": 49, "merg": 50, "multiscale_cropp": 51, "pad": 52, "project": 53, "rotation_mask_transform": 54, "save": 55, "so2_random_rot": 56, "track_transform": 57, "logger": [58, 59], "mlflow": [59, 87], "base_model": 61, "basic_model": 62, "classif": [63, 64, 65], "timepoint_classif": 65, "contrast": [66, 67], "handler": [68, 69, 70, 71], "base_handl": 69, "image_handl": 70, "load_image_patch": 71, "im2im": [72, 73, 74, 75, 76, 77, 78, 79, 80, 81], "gan": 73, "multi_task": 74, "instance_seg": 76, "noise_anneal": 77, "postprocess": [78, 79, 80, 81], "act_thresh_label": 79, "auto_thresh": 80, "dict_to_im": 81, "jepa": [82, 83, 84, 85], "ijepa": 83, "iwm": 84, "jepa_bas": 85, "vae": [89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104], "base_va": 90, "image_canon_va": 91, "image_encod": 92, "image_va": 93, "latent_loss_va": 94, "o2_spharm_va": [95, 96, 97], "o2_spharm_encod": 96, "point_cloud_va": 98, "prior": [99, 100, 101, 102, 103], "abstract_prior": 100, "gaussian": 101, "identity_prior": 102, "joint_prior": 103, "tabular_va": 104, "nn": [105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155], "discrimin": [106, 107, 108], "multi_scale_discrimin": 107, "n_layer_discrimin": 108, "head": [109, 110, 111, 112, 113, 114, 115, 116], "base_head": 110, "gan_head": 111, "gan_head_superr": 112, "mae_head": 113, "mask_head": 114, "res_blocks_head": 115, "vic_reg": [116, 129], "hr_skip": 117, "loss": [118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131], "adversarial_loss": 119, "chamfer_loss": 120, "continuous_bernoulli": 121, "cosine_loss": 122, "gan_loss": 123, "gaussian_nll_loss": 124, "geomloss": 125, "loss_wrapp": 126, "spharm_loss": 127, "threshold_loss": 128, "weibul": 130, "weighted_mse_loss": 131, "mlp": 132, "point_cloud": [133, 134, 135, 136, 137, 156, 157, 158], "dgcnn": 134, "folding_net": 135, "graph_funct": 136, "vnn": 137, "res_unit": 138, "spatial_transform": 139, "track_sequence_predictor": 140, "vit": [141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155], "block": [142, 143, 144, 145, 146, 147, 148, 149], "cross_attent": 143, "intermediate_weigh": 144, "masked_unit_attent": 145, "patchifi": [146, 147, 148, 149], "patchify_bas": 148, "patchify_hiera": 149, "decod": 150, "encod": 151, "mae": 152, "predictor": 153, "seg": 154, "read_pcloud": 158, "checkpoint": 162, "download_test_data": 164, "dummy_dataset": 165, "pylogg": 166, "rich_util": 167, "rotat": [168, 171], "spharm": [169, 170, 171], "reconstruct": 170, "template_util": 172, "overview": [173, 176], "experi": [173, 176], "instal": 174, "modifi": 177, "exampl": 177, "test": 177}, "envversion": {"sphinx.domains.c": 2, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 8, "sphinx.domains.index": 1, "sphinx.domains.javascript": 2, "sphinx.domains.math": 2, "sphinx.domains.python": 3, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.viewcode": 1, "sphinx": 57}, "alltitles": {"Using the API": [[0, "using-the-api"]], "Loading Configs": [[0, "loading-configs"]], "Training and Prediction": [[0, "training-and-prediction"]], "Utils": [[0, "utils"]], "cyto_dl package": [[1, "module-cyto_dl"]], "Subpackages": [[1, "subpackages"], [2, "subpackages"], [17, "subpackages"], [24, "subpackages"], [36, "subpackages"], [60, "subpackages"], [72, "subpackages"], [75, "subpackages"], [89, "subpackages"], [105, "subpackages"], [141, "subpackages"], [142, "subpackages"], [156, "subpackages"], [160, "subpackages"]], "Submodules": [[1, "submodules"], [2, "submodules"], [3, "submodules"], [8, "submodules"], [17, "submodules"], [19, "submodules"], [24, "submodules"], [27, "submodules"], [37, "submodules"], [44, "submodules"], [58, "submodules"], [60, "submodules"], [63, "submodules"], [66, "submodules"], [68, "submodules"], [72, "submodules"], [75, "submodules"], [78, "submodules"], [82, "submodules"], [86, "submodules"], [89, "submodules"], [95, "submodules"], [99, "submodules"], [105, "submodules"], [106, "submodules"], [109, "submodules"], [118, "submodules"], [133, "submodules"], [141, "submodules"], [142, "submodules"], [146, "submodules"], [157, "submodules"], [160, "submodules"], [169, "submodules"]], "cyto_dl.api package": [[2, "cyto-dl-api-package"]], "cyto_dl.api.cyto_dl_model package": [[3, "module-cyto_dl.api.cyto_dl_model"]], "cyto_dl.api.cyto_dl_model.cyto_dl_base_model module": [[4, "module-cyto_dl.api.cyto_dl_model.cyto_dl_base_model"]], "cyto_dl.api.cyto_dl_model.segmentation_plugin_model module": [[5, "module-cyto_dl.api.cyto_dl_model.segmentation_plugin_model"]], "cyto_dl.api.data module": [[6, "module-cyto_dl.api.data"]], "cyto_dl.api.model module": [[7, "module-cyto_dl.api.model"]], "cyto_dl.callbacks package": [[8, "module-cyto_dl.callbacks"]], "cyto_dl.callbacks.callback_utils module": [[9, "cyto-dl-callbacks-callback-utils-module"]], "cyto_dl.callbacks.csv_saver module": [[10, "module-cyto_dl.callbacks.csv_saver"]], "cyto_dl.callbacks.image_saver module": [[11, "module-cyto_dl.callbacks.image_saver"]], "cyto_dl.callbacks.latent_walk module": [[12, "cyto-dl-callbacks-latent-walk-module"]], "cyto_dl.callbacks.layer_freeze module": [[13, "module-cyto_dl.callbacks.layer_freeze"]], "cyto_dl.callbacks.model_utils module": [[14, "module-cyto_dl.callbacks.model_utils"]], "cyto_dl.callbacks.outlier_detection module": [[15, "module-cyto_dl.callbacks.outlier_detection"]], "cyto_dl.compile module": [[16, "cyto-dl-compile-module"]], "cyto_dl.dataframe package": [[17, "module-cyto_dl.dataframe"]], "cyto_dl.dataframe.readers module": [[18, "module-cyto_dl.dataframe.readers"]], "cyto_dl.dataframe.transforms package": [[19, "module-cyto_dl.dataframe.transforms"]], "cyto_dl.dataframe.transforms.filter module": [[20, "module-cyto_dl.dataframe.transforms.filter"]], "cyto_dl.dataframe.transforms.group_cols module": [[21, "module-cyto_dl.dataframe.transforms.group_cols"]], "cyto_dl.dataframe.transforms.misc module": [[22, "module-cyto_dl.dataframe.transforms.misc"]], "cyto_dl.dataframe.transforms.split module": [[23, "module-cyto_dl.dataframe.transforms.split"]], "cyto_dl.datamodules package": [[24, "module-cyto_dl.datamodules"]], "cyto_dl.datamodules.array module": [[25, "module-cyto_dl.datamodules.array"]], "cyto_dl.datamodules.data_dict module": [[26, "module-cyto_dl.datamodules.data_dict"]], "cyto_dl.datamodules.dataframe package": [[27, "module-cyto_dl.datamodules.dataframe"]], "cyto_dl.datamodules.dataframe.dataframe_datamodule module": [[28, "module-cyto_dl.datamodules.dataframe.dataframe_datamodule"]], "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule module": [[29, "module-cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule"]], "cyto_dl.datamodules.dataframe.utils module": [[30, "module-cyto_dl.datamodules.dataframe.utils"]], "cyto_dl.datamodules.folder module": [[31, "module-cyto_dl.datamodules.folder"]], "cyto_dl.datamodules.multidim_image module": [[32, "module-cyto_dl.datamodules.multidim_image"]], "cyto_dl.datamodules.smartcache module": [[33, "module-cyto_dl.datamodules.smartcache"]], "cyto_dl.datamodules.torchvision module": [[34, "module-cyto_dl.datamodules.torchvision"]], "cyto_dl.eval module": [[35, "module-cyto_dl.eval"]], "cyto_dl.image package": [[36, "module-cyto_dl.image"]], "cyto_dl.image.io package": [[37, "module-cyto_dl.image.io"]], "cyto_dl.image.io.aicsimage_loader module": [[38, "module-cyto_dl.image.io.aicsimage_loader"]], "cyto_dl.image.io.monai_bio_reader module": [[39, "module-cyto_dl.image.io.monai_bio_reader"]], "cyto_dl.image.io.numpy_reader module": [[40, "module-cyto_dl.image.io.numpy_reader"]], "cyto_dl.image.io.ome_zarr_reader module": [[41, "module-cyto_dl.image.io.ome_zarr_reader"]], "cyto_dl.image.io.polygon_loader module": [[42, "module-cyto_dl.image.io.polygon_loader"]], "cyto_dl.image.io.skimage_reader module": [[43, "module-cyto_dl.image.io.skimage_reader"]], "cyto_dl.image.transforms package": [[44, "module-cyto_dl.image.transforms"]], "cyto_dl.image.transforms.add_meta module": [[45, "module-cyto_dl.image.transforms.add_meta"]], "cyto_dl.image.transforms.bright_sampler module": [[46, "module-cyto_dl.image.transforms.bright_sampler"]], "cyto_dl.image.transforms.clip module": [[47, "module-cyto_dl.image.transforms.clip"]], "cyto_dl.image.transforms.contrastadjust module": [[48, "module-cyto_dl.image.transforms.contrastadjust"]], "cyto_dl.image.transforms.generate_jepa_masks module": [[49, "module-cyto_dl.image.transforms.generate_jepa_masks"]], "cyto_dl.image.transforms.merge module": [[50, "module-cyto_dl.image.transforms.merge"]], "cyto_dl.image.transforms.multiscale_cropper module": [[51, "module-cyto_dl.image.transforms.multiscale_cropper"]], "cyto_dl.image.transforms.pad module": [[52, "module-cyto_dl.image.transforms.pad"]], "cyto_dl.image.transforms.project module": [[53, "module-cyto_dl.image.transforms.project"]], "cyto_dl.image.transforms.rotation_mask_transform module": [[54, "cyto-dl-image-transforms-rotation-mask-transform-module"]], "cyto_dl.image.transforms.save module": [[55, "module-cyto_dl.image.transforms.save"]], "cyto_dl.image.transforms.so2_random_rotation module": [[56, "cyto-dl-image-transforms-so2-random-rotation-module"]], "cyto_dl.image.transforms.track_transforms module": [[57, "module-cyto_dl.image.transforms.track_transforms"]], "cyto_dl.loggers package": [[58, "module-cyto_dl.loggers"]], "cyto_dl.loggers.mlflow module": [[59, "module-cyto_dl.loggers.mlflow"]], "cyto_dl.models package": [[60, "module-cyto_dl.models"]], "cyto_dl.models.base_model module": [[61, "module-cyto_dl.models.base_model"]], "cyto_dl.models.basic_model module": [[62, "module-cyto_dl.models.basic_model"]], "cyto_dl.models.classification package": [[63, "module-cyto_dl.models.classification"]], "cyto_dl.models.classification.classification module": [[64, "module-cyto_dl.models.classification.classification"]], "cyto_dl.models.classification.timepoint_classification module": [[65, "module-cyto_dl.models.classification.timepoint_classification"]], "cyto_dl.models.contrastive package": [[66, "module-cyto_dl.models.contrastive"]], "cyto_dl.models.contrastive.contrastive module": [[67, "module-cyto_dl.models.contrastive.contrastive"]], "cyto_dl.models.handlers package": [[68, "cyto-dl-models-handlers-package"]], "cyto_dl.models.handlers.base_handler module": [[69, "cyto-dl-models-handlers-base-handler-module"]], "cyto_dl.models.handlers.image_handler module": [[70, "cyto-dl-models-handlers-image-handler-module"]], "cyto_dl.models.handlers.load_image_patch module": [[71, "cyto-dl-models-handlers-load-image-patch-module"]], "cyto_dl.models.im2im package": [[72, "module-cyto_dl.models.im2im"]], "cyto_dl.models.im2im.gan module": [[73, "module-cyto_dl.models.im2im.gan"]], "cyto_dl.models.im2im.multi_task module": [[74, "module-cyto_dl.models.im2im.multi_task"]], "cyto_dl.models.im2im.utils package": [[75, "module-cyto_dl.models.im2im.utils"]], "cyto_dl.models.im2im.utils.instance_seg module": [[76, "module-cyto_dl.models.im2im.utils.instance_seg"]], "cyto_dl.models.im2im.utils.noise_annealer module": [[77, "module-cyto_dl.models.im2im.utils.noise_annealer"]], "cyto_dl.models.im2im.utils.postprocessing package": [[78, "module-cyto_dl.models.im2im.utils.postprocessing"]], "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label module": [[79, "module-cyto_dl.models.im2im.utils.postprocessing.act_thresh_label"]], "cyto_dl.models.im2im.utils.postprocessing.auto_thresh module": [[80, "module-cyto_dl.models.im2im.utils.postprocessing.auto_thresh"]], "cyto_dl.models.im2im.utils.postprocessing.dict_to_im module": [[81, "module-cyto_dl.models.im2im.utils.postprocessing.dict_to_im"]], "cyto_dl.models.jepa package": [[82, "module-cyto_dl.models.jepa"]], "cyto_dl.models.jepa.ijepa module": [[83, "module-cyto_dl.models.jepa.ijepa"]], "cyto_dl.models.jepa.iwm module": [[84, "module-cyto_dl.models.jepa.iwm"]], "cyto_dl.models.jepa.jepa_base module": [[85, "module-cyto_dl.models.jepa.jepa_base"]], "cyto_dl.models.utils package": [[86, "module-cyto_dl.models.utils"]], "cyto_dl.models.utils.mlflow module": [[87, "module-cyto_dl.models.utils.mlflow"]], "cyto_dl.models.utils.utils module": [[88, "module-cyto_dl.models.utils.utils"]], "cyto_dl.models.vae package": [[89, "cyto-dl-models-vae-package"]], "cyto_dl.models.vae.base_vae module": [[90, "module-cyto_dl.models.vae.base_vae"]], "cyto_dl.models.vae.image_canon_vae module": [[91, "cyto-dl-models-vae-image-canon-vae-module"]], "cyto_dl.models.vae.image_encoder module": [[92, "cyto-dl-models-vae-image-encoder-module"]], "cyto_dl.models.vae.image_vae module": [[93, "cyto-dl-models-vae-image-vae-module"]], "cyto_dl.models.vae.latent_loss_vae module": [[94, "cyto-dl-models-vae-latent-loss-vae-module"]], "cyto_dl.models.vae.o2_spharm_vae package": [[95, "cyto-dl-models-vae-o2-spharm-vae-package"]], "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_encoder module": [[96, "cyto-dl-models-vae-o2-spharm-vae-o2-spharm-encoder-module"]], "cyto_dl.models.vae.o2_spharm_vae.o2_spharm_vae module": [[97, "cyto-dl-models-vae-o2-spharm-vae-o2-spharm-vae-module"]], "cyto_dl.models.vae.point_cloud_vae module": [[98, "cyto-dl-models-vae-point-cloud-vae-module"]], "cyto_dl.models.vae.priors package": [[99, "module-cyto_dl.models.vae.priors"]], "cyto_dl.models.vae.priors.abstract_prior module": [[100, "module-cyto_dl.models.vae.priors.abstract_prior"]], "cyto_dl.models.vae.priors.gaussian module": [[101, "module-cyto_dl.models.vae.priors.gaussian"]], "cyto_dl.models.vae.priors.identity_prior module": [[102, "module-cyto_dl.models.vae.priors.identity_prior"]], "cyto_dl.models.vae.priors.joint_prior module": [[103, "module-cyto_dl.models.vae.priors.joint_prior"]], "cyto_dl.models.vae.tabular_vae module": [[104, "cyto-dl-models-vae-tabular-vae-module"]], "cyto_dl.nn package": [[105, "module-cyto_dl.nn"]], "cyto_dl.nn.discriminators package": [[106, "module-cyto_dl.nn.discriminators"]], "cyto_dl.nn.discriminators.multi_scale_discriminator module": [[107, "module-cyto_dl.nn.discriminators.multi_scale_discriminator"]], "cyto_dl.nn.discriminators.n_layer_discriminator module": [[108, "module-cyto_dl.nn.discriminators.n_layer_discriminator"]], "cyto_dl.nn.head package": [[109, "module-cyto_dl.nn.head"]], "cyto_dl.nn.head.base_head module": [[110, "module-cyto_dl.nn.head.base_head"]], "cyto_dl.nn.head.gan_head module": [[111, "module-cyto_dl.nn.head.gan_head"]], "cyto_dl.nn.head.gan_head_superres module": [[112, "module-cyto_dl.nn.head.gan_head_superres"]], "cyto_dl.nn.head.mae_head module": [[113, "module-cyto_dl.nn.head.mae_head"]], "cyto_dl.nn.head.mask_head module": [[114, "module-cyto_dl.nn.head.mask_head"]], "cyto_dl.nn.head.res_blocks_head module": [[115, "module-cyto_dl.nn.head.res_blocks_head"]], "cyto_dl.nn.head.vic_reg module": [[116, "module-cyto_dl.nn.head.vic_reg"]], "cyto_dl.nn.hr_skip module": [[117, "module-cyto_dl.nn.hr_skip"]], "cyto_dl.nn.losses package": [[118, "module-cyto_dl.nn.losses"]], "cyto_dl.nn.losses.adversarial_loss module": [[119, "module-cyto_dl.nn.losses.adversarial_loss"]], "cyto_dl.nn.losses.chamfer_loss module": [[120, "module-cyto_dl.nn.losses.chamfer_loss"]], "cyto_dl.nn.losses.continuous_bernoulli module": [[121, "module-cyto_dl.nn.losses.continuous_bernoulli"]], "cyto_dl.nn.losses.cosine_loss module": [[122, "module-cyto_dl.nn.losses.cosine_loss"]], "cyto_dl.nn.losses.gan_loss module": [[123, "module-cyto_dl.nn.losses.gan_loss"]], "cyto_dl.nn.losses.gaussian_nll_loss module": [[124, "module-cyto_dl.nn.losses.gaussian_nll_loss"]], "cyto_dl.nn.losses.geomloss module": [[125, "cyto-dl-nn-losses-geomloss-module"]], "cyto_dl.nn.losses.loss_wrapper module": [[126, "module-cyto_dl.nn.losses.loss_wrapper"]], "cyto_dl.nn.losses.spharm_loss module": [[127, "cyto-dl-nn-losses-spharm-loss-module"]], "cyto_dl.nn.losses.threshold_loss module": [[128, "module-cyto_dl.nn.losses.threshold_loss"]], "cyto_dl.nn.losses.vic_reg module": [[129, "module-cyto_dl.nn.losses.vic_reg"]], "cyto_dl.nn.losses.weibull module": [[130, "module-cyto_dl.nn.losses.weibull"]], "cyto_dl.nn.losses.weighted_mse_loss module": [[131, "module-cyto_dl.nn.losses.weighted_mse_loss"]], "cyto_dl.nn.mlp module": [[132, "module-cyto_dl.nn.mlp"]], "cyto_dl.nn.point_cloud package": [[133, "cyto-dl-nn-point-cloud-package"]], "cyto_dl.nn.point_cloud.dgcnn module": [[134, "cyto-dl-nn-point-cloud-dgcnn-module"]], "cyto_dl.nn.point_cloud.folding_net module": [[135, "cyto-dl-nn-point-cloud-folding-net-module"]], "cyto_dl.nn.point_cloud.graph_functions module": [[136, "cyto-dl-nn-point-cloud-graph-functions-module"]], "cyto_dl.nn.point_cloud.vnn module": [[137, "cyto-dl-nn-point-cloud-vnn-module"]], "cyto_dl.nn.res_unit module": [[138, "module-cyto_dl.nn.res_unit"]], "cyto_dl.nn.spatial_transformer module": [[139, "module-cyto_dl.nn.spatial_transformer"]], "cyto_dl.nn.track_sequence_predictor module": [[140, "module-cyto_dl.nn.track_sequence_predictor"]], "cyto_dl.nn.vits package": [[141, "module-cyto_dl.nn.vits"]], "cyto_dl.nn.vits.blocks package": [[142, "module-cyto_dl.nn.vits.blocks"]], "cyto_dl.nn.vits.blocks.cross_attention module": [[143, "module-cyto_dl.nn.vits.blocks.cross_attention"]], "cyto_dl.nn.vits.blocks.intermediate_weigher module": [[144, "module-cyto_dl.nn.vits.blocks.intermediate_weigher"]], "cyto_dl.nn.vits.blocks.masked_unit_attention module": [[145, "module-cyto_dl.nn.vits.blocks.masked_unit_attention"]], "cyto_dl.nn.vits.blocks.patchify package": [[146, "module-cyto_dl.nn.vits.blocks.patchify"]], "cyto_dl.nn.vits.blocks.patchify.patchify module": [[147, "module-cyto_dl.nn.vits.blocks.patchify.patchify"]], "cyto_dl.nn.vits.blocks.patchify.patchify_base module": [[148, "module-cyto_dl.nn.vits.blocks.patchify.patchify_base"]], "cyto_dl.nn.vits.blocks.patchify.patchify_hiera module": [[149, "module-cyto_dl.nn.vits.blocks.patchify.patchify_hiera"]], "cyto_dl.nn.vits.decoder module": [[150, "module-cyto_dl.nn.vits.decoder"]], "cyto_dl.nn.vits.encoder module": [[151, "module-cyto_dl.nn.vits.encoder"]], "cyto_dl.nn.vits.mae module": [[152, "module-cyto_dl.nn.vits.mae"]], "cyto_dl.nn.vits.predictor module": [[153, "module-cyto_dl.nn.vits.predictor"]], "cyto_dl.nn.vits.seg module": [[154, "module-cyto_dl.nn.vits.seg"]], "cyto_dl.nn.vits.utils module": [[155, "module-cyto_dl.nn.vits.utils"]], "cyto_dl.point_cloud package": [[156, "module-cyto_dl.point_cloud"]], "cyto_dl.point_cloud.io package": [[157, "cyto-dl-point-cloud-io-package"]], "cyto_dl.point_cloud.io.read_pcloud module": [[158, "cyto-dl-point-cloud-io-read-pcloud-module"]], "cyto_dl.train module": [[159, "module-cyto_dl.train"]], "cyto_dl.utils package": [[160, "module-cyto_dl.utils"]], "cyto_dl.utils.array module": [[161, "module-cyto_dl.utils.array"]], "cyto_dl.utils.checkpoint module": [[162, "module-cyto_dl.utils.checkpoint"]], "cyto_dl.utils.config module": [[163, "module-cyto_dl.utils.config"]], "cyto_dl.utils.download_test_data module": [[164, "module-cyto_dl.utils.download_test_data"]], "cyto_dl.utils.dummy_dataset module": [[165, "module-cyto_dl.utils.dummy_dataset"]], "cyto_dl.utils.pylogger module": [[166, "module-cyto_dl.utils.pylogger"]], "cyto_dl.utils.rich_utils module": [[167, "module-cyto_dl.utils.rich_utils"]], "cyto_dl.utils.rotation module": [[168, "module-cyto_dl.utils.rotation"]], "cyto_dl.utils.spharm package": [[169, "cyto-dl-utils-spharm-package"]], "cyto_dl.utils.spharm.reconstruction module": [[170, "cyto-dl-utils-spharm-reconstruction-module"]], "cyto_dl.utils.spharm.rotation module": [[171, "cyto-dl-utils-spharm-rotation-module"]], "cyto_dl.utils.template_utils module": [[172, "module-cyto_dl.utils.template_utils"]], "Overview": [[173, "overview"], [176, "overview"]], "data": [[173, "data"], [176, "data"]], "experiment": [[173, "experiment"], [176, "experiment"]], "model": [[173, "model"], [176, "model"]], "Installation": [[174, "installation"]], "cyto_dl": [[175, "cyto-dl"]], "Modifying the Example Configs": [[177, "modifying-the-example-configs"]], "Training": [[177, "training"]], "Testing/Prediction": [[177, "testing-prediction"]]}, "indexentries": {"cyto_dl": [[1, "module-cyto_dl"]], "module": [[1, "module-cyto_dl"], [3, "module-cyto_dl.api.cyto_dl_model"], [4, "module-cyto_dl.api.cyto_dl_model.cyto_dl_base_model"], [5, "module-cyto_dl.api.cyto_dl_model.segmentation_plugin_model"], [6, "module-cyto_dl.api.data"], [7, "module-cyto_dl.api.model"], [8, "module-cyto_dl.callbacks"], [10, "module-cyto_dl.callbacks.csv_saver"], [11, "module-cyto_dl.callbacks.image_saver"], [13, "module-cyto_dl.callbacks.layer_freeze"], [14, "module-cyto_dl.callbacks.model_utils"], [15, "module-cyto_dl.callbacks.outlier_detection"], [17, "module-cyto_dl.dataframe"], [18, "module-cyto_dl.dataframe.readers"], [19, "module-cyto_dl.dataframe.transforms"], [20, "module-cyto_dl.dataframe.transforms.filter"], [21, "module-cyto_dl.dataframe.transforms.group_cols"], [22, "module-cyto_dl.dataframe.transforms.misc"], [23, "module-cyto_dl.dataframe.transforms.split"], [24, "module-cyto_dl.datamodules"], [25, "module-cyto_dl.datamodules.array"], [26, "module-cyto_dl.datamodules.data_dict"], [27, "module-cyto_dl.datamodules.dataframe"], [28, "module-cyto_dl.datamodules.dataframe.dataframe_datamodule"], [29, "module-cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule"], [30, "module-cyto_dl.datamodules.dataframe.utils"], [31, "module-cyto_dl.datamodules.folder"], [32, "module-cyto_dl.datamodules.multidim_image"], [33, "module-cyto_dl.datamodules.smartcache"], [34, "module-cyto_dl.datamodules.torchvision"], [35, "module-cyto_dl.eval"], [36, "module-cyto_dl.image"], [37, "module-cyto_dl.image.io"], [38, "module-cyto_dl.image.io.aicsimage_loader"], [39, "module-cyto_dl.image.io.monai_bio_reader"], [40, "module-cyto_dl.image.io.numpy_reader"], [41, "module-cyto_dl.image.io.ome_zarr_reader"], [42, "module-cyto_dl.image.io.polygon_loader"], [43, "module-cyto_dl.image.io.skimage_reader"], [44, "module-cyto_dl.image.transforms"], [45, "module-cyto_dl.image.transforms.add_meta"], [46, "module-cyto_dl.image.transforms.bright_sampler"], [47, "module-cyto_dl.image.transforms.clip"], [48, "module-cyto_dl.image.transforms.contrastadjust"], [49, "module-cyto_dl.image.transforms.generate_jepa_masks"], [50, "module-cyto_dl.image.transforms.merge"], [51, "module-cyto_dl.image.transforms.multiscale_cropper"], [52, "module-cyto_dl.image.transforms.pad"], [53, "module-cyto_dl.image.transforms.project"], [55, "module-cyto_dl.image.transforms.save"], [57, "module-cyto_dl.image.transforms.track_transforms"], [58, "module-cyto_dl.loggers"], [59, "module-cyto_dl.loggers.mlflow"], [60, "module-cyto_dl.models"], [61, "module-cyto_dl.models.base_model"], [62, "module-cyto_dl.models.basic_model"], [63, "module-cyto_dl.models.classification"], [64, "module-cyto_dl.models.classification.classification"], [65, "module-cyto_dl.models.classification.timepoint_classification"], [66, "module-cyto_dl.models.contrastive"], [67, "module-cyto_dl.models.contrastive.contrastive"], [72, "module-cyto_dl.models.im2im"], [73, "module-cyto_dl.models.im2im.gan"], [74, "module-cyto_dl.models.im2im.multi_task"], [75, "module-cyto_dl.models.im2im.utils"], [76, "module-cyto_dl.models.im2im.utils.instance_seg"], [77, "module-cyto_dl.models.im2im.utils.noise_annealer"], [78, "module-cyto_dl.models.im2im.utils.postprocessing"], [79, "module-cyto_dl.models.im2im.utils.postprocessing.act_thresh_label"], [80, "module-cyto_dl.models.im2im.utils.postprocessing.auto_thresh"], [81, "module-cyto_dl.models.im2im.utils.postprocessing.dict_to_im"], [82, "module-cyto_dl.models.jepa"], [83, "module-cyto_dl.models.jepa.ijepa"], [84, "module-cyto_dl.models.jepa.iwm"], [85, "module-cyto_dl.models.jepa.jepa_base"], [86, "module-cyto_dl.models.utils"], [87, "module-cyto_dl.models.utils.mlflow"], [88, "module-cyto_dl.models.utils.utils"], [90, "module-cyto_dl.models.vae.base_vae"], [99, "module-cyto_dl.models.vae.priors"], [100, "module-cyto_dl.models.vae.priors.abstract_prior"], [101, "module-cyto_dl.models.vae.priors.gaussian"], [102, "module-cyto_dl.models.vae.priors.identity_prior"], [103, "module-cyto_dl.models.vae.priors.joint_prior"], [105, "module-cyto_dl.nn"], [106, "module-cyto_dl.nn.discriminators"], [107, "module-cyto_dl.nn.discriminators.multi_scale_discriminator"], [108, "module-cyto_dl.nn.discriminators.n_layer_discriminator"], [109, "module-cyto_dl.nn.head"], [110, "module-cyto_dl.nn.head.base_head"], [111, "module-cyto_dl.nn.head.gan_head"], [112, "module-cyto_dl.nn.head.gan_head_superres"], [113, "module-cyto_dl.nn.head.mae_head"], [114, "module-cyto_dl.nn.head.mask_head"], [115, "module-cyto_dl.nn.head.res_blocks_head"], [116, "module-cyto_dl.nn.head.vic_reg"], [117, "module-cyto_dl.nn.hr_skip"], [118, "module-cyto_dl.nn.losses"], [119, "module-cyto_dl.nn.losses.adversarial_loss"], [120, "module-cyto_dl.nn.losses.chamfer_loss"], [121, "module-cyto_dl.nn.losses.continuous_bernoulli"], [122, "module-cyto_dl.nn.losses.cosine_loss"], [123, "module-cyto_dl.nn.losses.gan_loss"], [124, "module-cyto_dl.nn.losses.gaussian_nll_loss"], [126, "module-cyto_dl.nn.losses.loss_wrapper"], [128, "module-cyto_dl.nn.losses.threshold_loss"], [129, "module-cyto_dl.nn.losses.vic_reg"], [130, "module-cyto_dl.nn.losses.weibull"], [131, "module-cyto_dl.nn.losses.weighted_mse_loss"], [132, "module-cyto_dl.nn.mlp"], [138, "module-cyto_dl.nn.res_unit"], [139, "module-cyto_dl.nn.spatial_transformer"], [140, "module-cyto_dl.nn.track_sequence_predictor"], [141, "module-cyto_dl.nn.vits"], [142, "module-cyto_dl.nn.vits.blocks"], [143, "module-cyto_dl.nn.vits.blocks.cross_attention"], [144, "module-cyto_dl.nn.vits.blocks.intermediate_weigher"], [145, "module-cyto_dl.nn.vits.blocks.masked_unit_attention"], [146, "module-cyto_dl.nn.vits.blocks.patchify"], [147, "module-cyto_dl.nn.vits.blocks.patchify.patchify"], [148, "module-cyto_dl.nn.vits.blocks.patchify.patchify_base"], [149, "module-cyto_dl.nn.vits.blocks.patchify.patchify_hiera"], [150, "module-cyto_dl.nn.vits.decoder"], [151, "module-cyto_dl.nn.vits.encoder"], [152, "module-cyto_dl.nn.vits.mae"], [153, "module-cyto_dl.nn.vits.predictor"], [154, "module-cyto_dl.nn.vits.seg"], [155, "module-cyto_dl.nn.vits.utils"], [156, "module-cyto_dl.point_cloud"], [159, "module-cyto_dl.train"], [160, "module-cyto_dl.utils"], [161, "module-cyto_dl.utils.array"], [162, "module-cyto_dl.utils.checkpoint"], [163, "module-cyto_dl.utils.config"], [164, "module-cyto_dl.utils.download_test_data"], [165, "module-cyto_dl.utils.dummy_dataset"], [166, "module-cyto_dl.utils.pylogger"], [167, "module-cyto_dl.utils.rich_utils"], [168, "module-cyto_dl.utils.rotation"], [172, "module-cyto_dl.utils.template_utils"]], "cyto_dl.api.cyto_dl_model": [[3, "module-cyto_dl.api.cyto_dl_model"]], "cytodlbasemodel (class in cyto_dl.api.cyto_dl_model.cyto_dl_base_model)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel"]], "cyto_dl.api.cyto_dl_model.cyto_dl_base_model": [[4, "module-cyto_dl.api.cyto_dl_model.cyto_dl_base_model"]], "from_default_config() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel class method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.from_default_config"]], "from_existing_config() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel class method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.from_existing_config"]], "get_config() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.get_config"]], "get_experiment_name() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.get_experiment_name"]], "predict() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.predict"]], "save_config() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.save_config"]], "set_experiment_name() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.set_experiment_name"]], "train() (cyto_dl.api.cyto_dl_model.cyto_dl_base_model.cytodlbasemodel method)": [[4, "cyto_dl.api.cyto_dl_model.cyto_dl_base_model.CytoDLBaseModel.train"]], "segmentationpluginmodel (class in cyto_dl.api.cyto_dl_model.segmentation_plugin_model)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel"]], "cyto_dl.api.cyto_dl_model.segmentation_plugin_model": [[5, "module-cyto_dl.api.cyto_dl_model.segmentation_plugin_model"]], "get_hardware_type() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_hardware_type"]], "get_input_channel() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_input_channel"]], "get_manifest_column_names() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_manifest_column_names"]], "get_patch_size() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_patch_size"]], "get_raw_image_channels() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_raw_image_channels"]], "get_split_column() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.get_split_column"]], "remove_split_column() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.remove_split_column"]], "set_hardware_type() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_hardware_type"]], "set_input_channel() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_input_channel"]], "set_manifest_column_names() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_manifest_column_names"]], "set_patch_size() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_patch_size"]], "set_raw_image_channels() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_raw_image_channels"]], "set_split_column() (cyto_dl.api.cyto_dl_model.segmentation_plugin_model.segmentationpluginmodel method)": [[5, "cyto_dl.api.cyto_dl_model.segmentation_plugin_model.SegmentationPluginModel.set_split_column"]], "autothresholdmethod (class in cyto_dl.api.data)": [[6, "cyto_dl.api.data.AutoThresholdMethod"]], "cpu (cyto_dl.api.data.hardwaretype attribute)": [[6, "cyto_dl.api.data.HardwareType.CPU"]], "experimenttype (class in cyto_dl.api.data)": [[6, "cyto_dl.api.data.ExperimentType"]], "gan (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.GAN"]], "gpu (cyto_dl.api.data.hardwaretype attribute)": [[6, "cyto_dl.api.data.HardwareType.GPU"]], "hardwaretype (class in cyto_dl.api.data)": [[6, "cyto_dl.api.data.HardwareType"]], "instance_seg (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.INSTANCE_SEG"]], "isodata (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.ISODATA"]], "label_free (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.LABEL_FREE"]], "large (cyto_dl.api.data.patchsize attribute)": [[6, "cyto_dl.api.data.PatchSize.LARGE"]], "li (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.LI"]], "local (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.LOCAL"]], "mean (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.MEAN"]], "medium (cyto_dl.api.data.patchsize attribute)": [[6, "cyto_dl.api.data.PatchSize.MEDIUM"]], "minimum (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.MINIMUM"]], "multiotsu (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.MULTIOTSU"]], "niblack (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.NIBLACK"]], "otsu (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.OTSU"]], "patchsize (class in cyto_dl.api.data)": [[6, "cyto_dl.api.data.PatchSize"]], "sauvola (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.SAUVOLA"]], "segmentation (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.SEGMENTATION"]], "segmentation_array (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.SEGMENTATION_ARRAY"]], "segmentation_plugin (cyto_dl.api.data.experimenttype attribute)": [[6, "cyto_dl.api.data.ExperimentType.SEGMENTATION_PLUGIN"]], "small (cyto_dl.api.data.patchsize attribute)": [[6, "cyto_dl.api.data.PatchSize.SMALL"]], "triangle (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.TRIANGLE"]], "yen (cyto_dl.api.data.autothresholdmethod attribute)": [[6, "cyto_dl.api.data.AutoThresholdMethod.YEN"]], "cyto_dl.api.data": [[6, "module-cyto_dl.api.data"]], "cytodlmodel (class in cyto_dl.api.model)": [[7, "cyto_dl.api.model.CytoDLModel"]], "cyto_dl.api.model": [[7, "module-cyto_dl.api.model"]], "download_example_data() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.download_example_data"]], "load_config_from_dict() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.load_config_from_dict"]], "load_config_from_file() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.load_config_from_file"]], "load_default_experiment() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.load_default_experiment"]], "override_config() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.override_config"]], "predict() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.predict"]], "print_config() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.print_config"]], "save_config() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.save_config"]], "train() (cyto_dl.api.model.cytodlmodel method)": [[7, "cyto_dl.api.model.CytoDLModel.train"]], "cyto_dl.callbacks": [[8, "module-cyto_dl.callbacks"]], "csvsaver (class in cyto_dl.callbacks.csv_saver)": [[10, "cyto_dl.callbacks.csv_saver.CSVSaver"]], "jepasaver (class in cyto_dl.callbacks.csv_saver)": [[10, "cyto_dl.callbacks.csv_saver.JEPASaver"]], "cyto_dl.callbacks.csv_saver": [[10, "module-cyto_dl.callbacks.csv_saver"]], "on_predict_epoch_end() (cyto_dl.callbacks.csv_saver.csvsaver method)": [[10, "cyto_dl.callbacks.csv_saver.CSVSaver.on_predict_epoch_end"]], "pred_to_csv() (cyto_dl.callbacks.csv_saver.csvsaver method)": [[10, "cyto_dl.callbacks.csv_saver.CSVSaver.pred_to_csv"]], "pred_to_csv() (cyto_dl.callbacks.csv_saver.jepasaver method)": [[10, "cyto_dl.callbacks.csv_saver.JEPASaver.pred_to_csv"]], "save_feats() (cyto_dl.callbacks.csv_saver.csvsaver method)": [[10, "cyto_dl.callbacks.csv_saver.CSVSaver.save_feats"]], "imagesaver (class in cyto_dl.callbacks.image_saver)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver"]], "cyto_dl.callbacks.image_saver": [[11, "module-cyto_dl.callbacks.image_saver"]], "on_predict_batch_end() (cyto_dl.callbacks.image_saver.imagesaver method)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver.on_predict_batch_end"]], "on_test_batch_end() (cyto_dl.callbacks.image_saver.imagesaver method)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver.on_test_batch_end"]], "on_train_batch_end() (cyto_dl.callbacks.image_saver.imagesaver method)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver.on_train_batch_end"]], "on_validation_batch_end() (cyto_dl.callbacks.image_saver.imagesaver method)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver.on_validation_batch_end"]], "save() (cyto_dl.callbacks.image_saver.imagesaver method)": [[11, "cyto_dl.callbacks.image_saver.ImageSaver.save"]], "layerfreeze (class in cyto_dl.callbacks.layer_freeze)": [[13, "cyto_dl.callbacks.layer_freeze.LayerFreeze"]], "cyto_dl.callbacks.layer_freeze": [[13, "module-cyto_dl.callbacks.layer_freeze"]], "setup() (cyto_dl.callbacks.layer_freeze.layerfreeze method)": [[13, "cyto_dl.callbacks.layer_freeze.LayerFreeze.setup"]], "getembeddings (class in cyto_dl.callbacks.model_utils)": [[14, "cyto_dl.callbacks.model_utils.GetEmbeddings"]], "cyto_dl.callbacks.model_utils": [[14, "module-cyto_dl.callbacks.model_utils"]], "get_all_embeddings() (in module cyto_dl.callbacks.model_utils)": [[14, "cyto_dl.callbacks.model_utils.get_all_embeddings"]], "on_test_epoch_end() (cyto_dl.callbacks.model_utils.getembeddings method)": [[14, "cyto_dl.callbacks.model_utils.GetEmbeddings.on_test_epoch_end"]], "save_predictions_classifier() (in module cyto_dl.callbacks.model_utils)": [[14, "cyto_dl.callbacks.model_utils.save_predictions_classifier"]], "outlierdetection (class in cyto_dl.callbacks.outlier_detection)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection"]], "calculate_mahalanobis_hook() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.calculate_mahalanobis_hook"]], "cyto_dl.callbacks.outlier_detection": [[15, "module-cyto_dl.callbacks.outlier_detection"]], "flatten_activations() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.flatten_activations"]], "on_load_checkpoint() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_load_checkpoint"]], "on_predict_batch_end() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_predict_batch_end"]], "on_predict_epoch_start() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_predict_epoch_start"]], "on_save_checkpoint() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_save_checkpoint"]], "on_test_batch_end() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_test_batch_end"]], "on_test_epoch_start() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_test_epoch_start"]], "on_train_epoch_start() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.on_train_epoch_start"]], "update_covariance_hook() (cyto_dl.callbacks.outlier_detection.outlierdetection method)": [[15, "cyto_dl.callbacks.outlier_detection.OutlierDetection.update_covariance_hook"]], "cyto_dl.dataframe": [[17, "module-cyto_dl.dataframe"]], "cyto_dl.dataframe.readers": [[18, "module-cyto_dl.dataframe.readers"]], "filter_columns() (in module cyto_dl.dataframe.readers)": [[18, "cyto_dl.dataframe.readers.filter_columns"]], "read_csv() (in module cyto_dl.dataframe.readers)": [[18, "cyto_dl.dataframe.readers.read_csv"]], "read_dataframe() (in module cyto_dl.dataframe.readers)": [[18, "cyto_dl.dataframe.readers.read_dataframe"]], "read_h5ad() (in module cyto_dl.dataframe.readers)": [[18, "cyto_dl.dataframe.readers.read_h5ad"]], "read_parquet() (in module cyto_dl.dataframe.readers)": [[18, "cyto_dl.dataframe.readers.read_parquet"]], "cyto_dl.dataframe.transforms": [[19, "module-cyto_dl.dataframe.transforms"]], "cyto_dl.dataframe.transforms.filter": [[20, "module-cyto_dl.dataframe.transforms.filter"]], "filter_columns() (in module cyto_dl.dataframe.transforms.filter)": [[20, "cyto_dl.dataframe.transforms.filter.filter_columns"]], "filter_rows() (in module cyto_dl.dataframe.transforms.filter)": [[20, "cyto_dl.dataframe.transforms.filter.filter_rows"]], "groupcols (class in cyto_dl.dataframe.transforms.group_cols)": [[21, "cyto_dl.dataframe.transforms.group_cols.GroupCols"]], "cyto_dl.dataframe.transforms.group_cols": [[21, "module-cyto_dl.dataframe.transforms.group_cols"]], "append_class_weights() (in module cyto_dl.dataframe.transforms.misc)": [[22, "cyto_dl.dataframe.transforms.misc.append_class_weights"]], "append_labels_to_integers() (in module cyto_dl.dataframe.transforms.misc)": [[22, "cyto_dl.dataframe.transforms.misc.append_labels_to_integers"]], "append_one_hot() (in module cyto_dl.dataframe.transforms.misc)": [[22, "cyto_dl.dataframe.transforms.misc.append_one_hot"]], "cyto_dl.dataframe.transforms.misc": [[22, "module-cyto_dl.dataframe.transforms.misc"]], "make_random_df() (in module cyto_dl.dataframe.transforms.misc)": [[22, "cyto_dl.dataframe.transforms.misc.make_random_df"]], "cyto_dl.dataframe.transforms.split": [[23, "module-cyto_dl.dataframe.transforms.split"]], "sample_n_each() (in module cyto_dl.dataframe.transforms.split)": [[23, "cyto_dl.dataframe.transforms.split.sample_n_each"]], "split_dataframe() (in module cyto_dl.dataframe.transforms.split)": [[23, "cyto_dl.dataframe.transforms.split.split_dataframe"]], "cyto_dl.datamodules": [[24, "module-cyto_dl.datamodules"]], "cyto_dl.datamodules.array": [[25, "module-cyto_dl.datamodules.array"]], "make_array_dataloader() (in module cyto_dl.datamodules.array)": [[25, "cyto_dl.datamodules.array.make_array_dataloader"]], "cyto_dl.datamodules.data_dict": [[26, "module-cyto_dl.datamodules.data_dict"]], "make_data_dict_dataloader() (in module cyto_dl.datamodules.data_dict)": [[26, "cyto_dl.datamodules.data_dict.make_data_dict_dataloader"]], "cyto_dl.datamodules.dataframe": [[27, "module-cyto_dl.datamodules.dataframe"]], "dataframedatamodule (class in cyto_dl.datamodules.dataframe.dataframe_datamodule)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule"]], "cyto_dl.datamodules.dataframe.dataframe_datamodule": [[28, "module-cyto_dl.datamodules.dataframe.dataframe_datamodule"]], "get_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.get_dataloader"]], "get_dataset() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.get_dataset"]], "make_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.make_dataloader"]], "predict_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.predict_dataloader"]], "test_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.test_dataloader"]], "train_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.train_dataloader"]], "val_dataloader() (cyto_dl.datamodules.dataframe.dataframe_datamodule.dataframedatamodule method)": [[28, "cyto_dl.datamodules.dataframe.dataframe_datamodule.DataframeDatamodule.val_dataloader"]], "groupeddataframedatamodule (class in cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule)": [[29, "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule.GroupedDataframeDatamodule"]], "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule": [[29, "module-cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule"]], "make_dataloader() (cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule.groupeddataframedatamodule method)": [[29, "cyto_dl.datamodules.dataframe.grouped_dataframe_datamodule.GroupedDataframeDatamodule.make_dataloader"]], "alternatingbatchsampler (class in cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.AlternatingBatchSampler"]], "removenankeysd (class in cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.RemoveNaNKeysd"]], "cyto_dl.datamodules.dataframe.utils": [[30, "module-cyto_dl.datamodules.dataframe.utils"]], "get_canonical_split_name() (in module cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.get_canonical_split_name"]], "get_dataset() (in module cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.get_dataset"]], "make_multiple_dataframe_splits() (in module cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.make_multiple_dataframe_splits"]], "make_single_dataframe_splits() (in module cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.make_single_dataframe_splits"]], "parse_transforms() (in module cyto_dl.datamodules.dataframe.utils)": [[30, "cyto_dl.datamodules.dataframe.utils.parse_transforms"]], "cyto_dl.datamodules.folder": [[31, "module-cyto_dl.datamodules.folder"]], "make_folder_dataloader() (in module cyto_dl.datamodules.folder)": [[31, "cyto_dl.datamodules.folder.make_folder_dataloader"]], "multidimimagedataset (class in cyto_dl.datamodules.multidim_image)": [[32, "cyto_dl.datamodules.multidim_image.MultiDimImageDataset"]], "create_metatensor() (cyto_dl.datamodules.multidim_image.multidimimagedataset method)": [[32, "cyto_dl.datamodules.multidim_image.MultiDimImageDataset.create_metatensor"]], "cyto_dl.datamodules.multidim_image": [[32, "module-cyto_dl.datamodules.multidim_image"]], "get_per_file_args() (cyto_dl.datamodules.multidim_image.multidimimagedataset method)": [[32, "cyto_dl.datamodules.multidim_image.MultiDimImageDataset.get_per_file_args"]], "is_batch() (cyto_dl.datamodules.multidim_image.multidimimagedataset method)": [[32, "cyto_dl.datamodules.multidim_image.MultiDimImageDataset.is_batch"]], "make_multidim_image_dataloader() (in module cyto_dl.datamodules.multidim_image)": [[32, "cyto_dl.datamodules.multidim_image.make_multidim_image_dataloader"]], "smartcachedatamodule (class in cyto_dl.datamodules.smartcache)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule"]], "cyto_dl.datamodules.smartcache": [[33, "module-cyto_dl.datamodules.smartcache"]], "get_per_file_args() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.get_per_file_args"]], "make_dataloader() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.make_dataloader"]], "predict_dataloader() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.predict_dataloader"]], "prepare_data() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.prepare_data"]], "setup() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.setup"]], "test_dataloader() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.test_dataloader"]], "train_dataloader() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.train_dataloader"]], "val_dataloader() (cyto_dl.datamodules.smartcache.smartcachedatamodule method)": [[33, "cyto_dl.datamodules.smartcache.SmartcacheDatamodule.val_dataloader"]], "cifar10 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CIFAR10"]], "cifar100 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CIFAR100"]], "clevrclassification (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CLEVRClassification"]], "caltech101 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Caltech101"]], "caltech256 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Caltech256"]], "celeba (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CelebA"]], "cityscapes (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Cityscapes"]], "cococaptions (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CocoCaptions"]], "cocodetection (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.CocoDetection"]], "country211 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Country211"]], "dtd (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.DTD"]], "datasetfolder (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.DatasetFolder"]], "emnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.EMNIST"]], "eurosat (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.EuroSAT"]], "fer2013 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.FER2013"]], "fgvcaircraft (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.FGVCAircraft"]], "fakedata (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.FakeData"]], "fashionmnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.FashionMNIST"]], "flickr30k (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Flickr30k"]], "flickr8k (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Flickr8k"]], "flowers102 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Flowers102"]], "food101 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Food101"]], "gtsrb (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.GTSRB"]], "hmdb51 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.HMDB51"]], "inaturalist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.INaturalist"]], "imagefolder (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.ImageFolder"]], "imagenet (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.ImageNet"]], "kmnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.KMNIST"]], "kinetics (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Kinetics"]], "kitti (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Kitti"]], "lfwpairs (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.LFWPairs"]], "lfwpeople (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.LFWPeople"]], "lsun (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.LSUN"]], "lsunclass (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.LSUNClass"]], "mnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.MNIST"]], "movingmnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.MovingMNIST"]], "omniglot (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Omniglot"]], "oxfordiiitpet (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.OxfordIIITPet"]], "pcam (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.PCAM"]], "phototour (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.PhotoTour"]], "places365 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.Places365"]], "qmnist (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.QMNIST"]], "renderedsst2 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.RenderedSST2"]], "sbdataset (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.SBDataset"]], "sbu (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.SBU"]], "semeion (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.SEMEION"]], "stl10 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.STL10"]], "sun397 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.SUN397"]], "svhn (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.SVHN"]], "stanfordcars (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.StanfordCars"]], "ucf101 (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.UCF101"]], "usps (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.USPS"]], "vocdetection (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.VOCDetection"]], "vocsegmentation (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.VOCSegmentation"]], "visiondataset (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.VisionDataset"]], "widerface (class in cyto_dl.datamodules.torchvision)": [[34, "cyto_dl.datamodules.torchvision.WIDERFace"]], "cyto_dl.datamodules.torchvision": [[34, "module-cyto_dl.datamodules.torchvision"]], "cyto_dl.eval": [[35, "module-cyto_dl.eval"]], "main() (in module cyto_dl.eval)": [[35, "cyto_dl.eval.main"]], "cyto_dl.image": [[36, "module-cyto_dl.image"]], "cyto_dl.image.io": [[37, "module-cyto_dl.image.io"]], "aicsimageloaderd (class in cyto_dl.image.io.aicsimage_loader)": [[38, "cyto_dl.image.io.aicsimage_loader.AICSImageLoaderd"]], "cyto_dl.image.io.aicsimage_loader": [[38, "module-cyto_dl.image.io.aicsimage_loader"]], "split_args() (cyto_dl.image.io.aicsimage_loader.aicsimageloaderd method)": [[38, "cyto_dl.image.io.aicsimage_loader.AICSImageLoaderd.split_args"]], "monaibioreader (class in cyto_dl.image.io.monai_bio_reader)": [[39, "cyto_dl.image.io.monai_bio_reader.MonaiBioReader"]], "cyto_dl.image.io.monai_bio_reader": [[39, "module-cyto_dl.image.io.monai_bio_reader"]], "get_data() (cyto_dl.image.io.monai_bio_reader.monaibioreader method)": [[39, "cyto_dl.image.io.monai_bio_reader.MonaiBioReader.get_data"]], "read() (cyto_dl.image.io.monai_bio_reader.monaibioreader method)": [[39, "cyto_dl.image.io.monai_bio_reader.MonaiBioReader.read"]], "verify_suffix() (cyto_dl.image.io.monai_bio_reader.monaibioreader method)": [[39, "cyto_dl.image.io.monai_bio_reader.MonaiBioReader.verify_suffix"]], "readnumpyfile (class in cyto_dl.image.io.numpy_reader)": [[40, "cyto_dl.image.io.numpy_reader.ReadNumpyFile"]], "cyto_dl.image.io.numpy_reader": [[40, "module-cyto_dl.image.io.numpy_reader"]], "omezarrreader (class in cyto_dl.image.io.ome_zarr_reader)": [[41, "cyto_dl.image.io.ome_zarr_reader.OmeZarrReader"]], "cyto_dl.image.io.ome_zarr_reader": [[41, "module-cyto_dl.image.io.ome_zarr_reader"]], "get_data() (cyto_dl.image.io.ome_zarr_reader.omezarrreader method)": [[41, "cyto_dl.image.io.ome_zarr_reader.OmeZarrReader.get_data"]], "read() (cyto_dl.image.io.ome_zarr_reader.omezarrreader method)": [[41, "cyto_dl.image.io.ome_zarr_reader.OmeZarrReader.read"]], "verify_suffix() (cyto_dl.image.io.ome_zarr_reader.omezarrreader method)": [[41, "cyto_dl.image.io.ome_zarr_reader.OmeZarrReader.verify_suffix"]], "polygonloaderd (class in cyto_dl.image.io.polygon_loader)": [[42, "cyto_dl.image.io.polygon_loader.PolygonLoaderd"]], "cyto_dl.image.io.polygon_loader": [[42, "module-cyto_dl.image.io.polygon_loader"]], "skimagereader (class in cyto_dl.image.io.skimage_reader)": [[43, "cyto_dl.image.io.skimage_reader.SkimageReader"]], "cyto_dl.image.io.skimage_reader": [[43, "module-cyto_dl.image.io.skimage_reader"]], "get_data() (cyto_dl.image.io.skimage_reader.skimagereader method)": [[43, "cyto_dl.image.io.skimage_reader.SkimageReader.get_data"]], "read() (cyto_dl.image.io.skimage_reader.skimagereader method)": [[43, "cyto_dl.image.io.skimage_reader.SkimageReader.read"]], "verify_suffix() (cyto_dl.image.io.skimage_reader.skimagereader method)": [[43, "cyto_dl.image.io.skimage_reader.SkimageReader.verify_suffix"]], "cyto_dl.image.transforms": [[44, "module-cyto_dl.image.transforms"]], "addmeta (class in cyto_dl.image.transforms.add_meta)": [[45, "cyto_dl.image.transforms.add_meta.AddMeta"]], "cyto_dl.image.transforms.add_meta": [[45, "module-cyto_dl.image.transforms.add_meta"]], "brightsampler (class in cyto_dl.image.transforms.bright_sampler)": [[46, "cyto_dl.image.transforms.bright_sampler.BrightSampler"]], "cyto_dl.image.transforms.bright_sampler": [[46, "module-cyto_dl.image.transforms.bright_sampler"]], "clip (class in cyto_dl.image.transforms.clip)": [[47, "cyto_dl.image.transforms.clip.Clip"]], "clipd (class in cyto_dl.image.transforms.clip)": [[47, "cyto_dl.image.transforms.clip.Clipd"]], "cyto_dl.image.transforms.clip": [[47, "module-cyto_dl.image.transforms.clip"]], "contrastadjust (class in cyto_dl.image.transforms.contrastadjust)": [[48, "cyto_dl.image.transforms.contrastadjust.ContrastAdjust"]], "contrastadjustd (class in cyto_dl.image.transforms.contrastadjust)": [[48, "cyto_dl.image.transforms.contrastadjust.ContrastAdjustd"]], "cyto_dl.image.transforms.contrastadjust": [[48, "module-cyto_dl.image.transforms.contrastadjust"]], "jepamaskgenerator (class in cyto_dl.image.transforms.generate_jepa_masks)": [[49, "cyto_dl.image.transforms.generate_jepa_masks.JEPAMaskGenerator"]], "cyto_dl.image.transforms.generate_jepa_masks": [[49, "module-cyto_dl.image.transforms.generate_jepa_masks"]], "remove_excess_pixels() (cyto_dl.image.transforms.generate_jepa_masks.jepamaskgenerator method)": [[49, "cyto_dl.image.transforms.generate_jepa_masks.JEPAMaskGenerator.remove_excess_pixels"]], "merged (class in cyto_dl.image.transforms.merge)": [[50, "cyto_dl.image.transforms.merge.Merged"]], "cyto_dl.image.transforms.merge": [[50, "module-cyto_dl.image.transforms.merge"]], "randommultiscalecropd (class in cyto_dl.image.transforms.multiscale_cropper)": [[51, "cyto_dl.image.transforms.multiscale_cropper.RandomMultiScaleCropd"]], "cyto_dl.image.transforms.multiscale_cropper": [[51, "module-cyto_dl.image.transforms.multiscale_cropper"]], "generate_slices() (cyto_dl.image.transforms.multiscale_cropper.randommultiscalecropd method)": [[51, "cyto_dl.image.transforms.multiscale_cropper.RandomMultiScaleCropd.generate_slices"]], "padzd (class in cyto_dl.image.transforms.pad)": [[52, "cyto_dl.image.transforms.pad.PadZd"]], "cyto_dl.image.transforms.pad": [[52, "module-cyto_dl.image.transforms.pad"]], "maxprojectd (class in cyto_dl.image.transforms.project)": [[53, "cyto_dl.image.transforms.project.MaxProjectd"]], "cyto_dl.image.transforms.project": [[53, "module-cyto_dl.image.transforms.project"]], "save (class in cyto_dl.image.transforms.save)": [[55, "cyto_dl.image.transforms.save.Save"]], "saved (class in cyto_dl.image.transforms.save)": [[55, "cyto_dl.image.transforms.save.Saved"]], "cyto_dl.image.transforms.save": [[55, "module-cyto_dl.image.transforms.save"]], "cropresize (class in cyto_dl.image.transforms.track_transforms)": [[57, "cyto_dl.image.transforms.track_transforms.CropResize"]], "generatetracklabels (class in cyto_dl.image.transforms.track_transforms)": [[57, "cyto_dl.image.transforms.track_transforms.GenerateTrackLabels"]], "perchannel (class in cyto_dl.image.transforms.track_transforms)": [[57, "cyto_dl.image.transforms.track_transforms.PerChannel"]], "splittrackd (class in cyto_dl.image.transforms.track_transforms)": [[57, "cyto_dl.image.transforms.track_transforms.SplitTrackd"]], "cyto_dl.image.transforms.track_transforms": [[57, "module-cyto_dl.image.transforms.track_transforms"]], "cyto_dl.loggers": [[58, "module-cyto_dl.loggers"]], "mlflowlogger (class in cyto_dl.loggers.mlflow)": [[59, "cyto_dl.loggers.mlflow.MLFlowLogger"]], "after_save_checkpoint() (cyto_dl.loggers.mlflow.mlflowlogger method)": [[59, "cyto_dl.loggers.mlflow.MLFlowLogger.after_save_checkpoint"]], "cyto_dl.loggers.mlflow": [[59, "module-cyto_dl.loggers.mlflow"]], "log_hyperparams() (cyto_dl.loggers.mlflow.mlflowlogger method)": [[59, "cyto_dl.loggers.mlflow.MLFlowLogger.log_hyperparams"]], "log_metrics() (cyto_dl.loggers.mlflow.mlflowlogger method)": [[59, "cyto_dl.loggers.mlflow.MLFlowLogger.log_metrics"]], "cyto_dl.models": [[60, "module-cyto_dl.models"]], "basemodel (class in cyto_dl.models.base_model)": [[61, "cyto_dl.models.base_model.BaseModel"]], "basemodelmeta (class in cyto_dl.models.base_model)": [[61, "cyto_dl.models.base_model.BaseModelMeta"]], "compute_metrics() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.compute_metrics"]], "configure_optimizers() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.configure_optimizers"]], "cyto_dl.models.base_model": [[61, "module-cyto_dl.models.base_model"]], "forward() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.forward"]], "model_step() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.model_step"]], "on_train_start() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.on_train_start"]], "parse_batch() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.parse_batch"]], "predict_step() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.predict_step"]], "test_step() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.test_step"]], "training_step() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.training_step"]], "validation_step() (cyto_dl.models.base_model.basemodel method)": [[61, "cyto_dl.models.base_model.BaseModel.validation_step"]], "basicmodel (class in cyto_dl.models.basic_model)": [[62, "cyto_dl.models.basic_model.BasicModel"]], "cyto_dl.models.basic_model": [[62, "module-cyto_dl.models.basic_model"]], "forward() (cyto_dl.models.basic_model.basicmodel method)": [[62, "cyto_dl.models.basic_model.BasicModel.forward"]], "model_step() (cyto_dl.models.basic_model.basicmodel method)": [[62, "cyto_dl.models.basic_model.BasicModel.model_step"]], "cyto_dl.models.classification": [[63, "module-cyto_dl.models.classification"]], "classification (class in cyto_dl.models.classification.classification)": [[64, "cyto_dl.models.classification.classification.Classification"]], "cyto_dl.models.classification.classification": [[64, "module-cyto_dl.models.classification.classification"]], "forward() (cyto_dl.models.classification.classification.classification method)": [[64, "cyto_dl.models.classification.classification.Classification.forward"]], "model_step() (cyto_dl.models.classification.classification.classification method)": [[64, "cyto_dl.models.classification.classification.Classification.model_step"]], "predict_step() (cyto_dl.models.classification.classification.classification method)": [[64, "cyto_dl.models.classification.classification.Classification.predict_step"]], "save_images() (cyto_dl.models.classification.classification.classification method)": [[64, "cyto_dl.models.classification.classification.Classification.save_images"]], "should_save_image() (cyto_dl.models.classification.classification.classification method)": [[64, "cyto_dl.models.classification.classification.Classification.should_save_image"]], "timepointclassification (class in cyto_dl.models.classification.timepoint_classification)": [[65, "cyto_dl.models.classification.timepoint_classification.TimepointClassification"]], "cyto_dl.models.classification.timepoint_classification": [[65, "module-cyto_dl.models.classification.timepoint_classification"]], "predict_step() (cyto_dl.models.classification.timepoint_classification.timepointclassification method)": [[65, "cyto_dl.models.classification.timepoint_classification.TimepointClassification.predict_step"]], "cyto_dl.models.contrastive": [[66, "module-cyto_dl.models.contrastive"]], "contrastive (class in cyto_dl.models.contrastive.contrastive)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive"]], "cyto_dl.models.contrastive.contrastive": [[67, "module-cyto_dl.models.contrastive.contrastive"]], "forward() (cyto_dl.models.contrastive.contrastive.contrastive method)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive.forward"]], "model_step() (cyto_dl.models.contrastive.contrastive.contrastive method)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive.model_step"]], "plot_classes() (cyto_dl.models.contrastive.contrastive.contrastive method)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive.plot_classes"]], "plot_neighbors() (cyto_dl.models.contrastive.contrastive.contrastive method)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive.plot_neighbors"]], "predict_step() (cyto_dl.models.contrastive.contrastive.contrastive method)": [[67, "cyto_dl.models.contrastive.contrastive.Contrastive.predict_step"]], "cyto_dl.models.im2im": [[72, "module-cyto_dl.models.im2im"]], "gan (class in cyto_dl.models.im2im.gan)": [[73, "cyto_dl.models.im2im.gan.GAN"]], "configure_optimizers() (cyto_dl.models.im2im.gan.gan method)": [[73, "cyto_dl.models.im2im.gan.GAN.configure_optimizers"]], "cyto_dl.models.im2im.gan": [[73, "module-cyto_dl.models.im2im.gan"]], "model_step() (cyto_dl.models.im2im.gan.gan method)": [[73, "cyto_dl.models.im2im.gan.GAN.model_step"]], "predict_step() (cyto_dl.models.im2im.gan.gan method)": [[73, "cyto_dl.models.im2im.gan.GAN.predict_step"]], "multitaskim2im (class in cyto_dl.models.im2im.multi_task)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im"]], "configure_optimizers() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.configure_optimizers"]], "cyto_dl.models.im2im.multi_task": [[74, "module-cyto_dl.models.im2im.multi_task"]], "forward() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.forward"]], "get_n_postprocess_image() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.get_n_postprocess_image"]], "get_per_head() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.get_per_head"]], "model_step() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.model_step"]], "predict_step() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.predict_step"]], "run_forward() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.run_forward"]], "test_step() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.test_step"]], "training_step() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.training_step"]], "validation_step() (cyto_dl.models.im2im.multi_task.multitaskim2im method)": [[74, "cyto_dl.models.im2im.multi_task.MultiTaskIm2Im.validation_step"]], "cyto_dl.models.im2im.utils": [[75, "module-cyto_dl.models.im2im.utils"]], "instancesegcluster (class in cyto_dl.models.im2im.utils.instance_seg)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegCluster"]], "instancesegloss (class in cyto_dl.models.im2im.utils.instance_seg)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegLoss"]], "instancesegpreprocessd (class in cyto_dl.models.im2im.utils.instance_seg)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd"]], "instancesegrandflipd (class in cyto_dl.models.im2im.utils.instance_seg)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegRandFlipd"]], "cluster_object() (cyto_dl.models.im2im.utils.instance_seg.instancesegcluster method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegCluster.cluster_object"]], "cyto_dl.models.im2im.utils.instance_seg": [[76, "module-cyto_dl.models.im2im.utils.instance_seg"]], "embed_from_skel() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.embed_from_skel"]], "kd_clustering() (cyto_dl.models.im2im.utils.instance_seg.instancesegcluster method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegCluster.kd_clustering"]], "keep_largest_cc() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.keep_largest_cc"]], "label_2d() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.label_2d"]], "pad_slice() (in module cyto_dl.models.im2im.utils.instance_seg)": [[76, "cyto_dl.models.im2im.utils.instance_seg.pad_slice"]], "remove_small_skeletons() (cyto_dl.models.im2im.utils.instance_seg.instancesegcluster method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegCluster.remove_small_skeletons"]], "shrink() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.shrink"]], "skeleton_tall() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.skeleton_tall"]], "smooth_embedding() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.smooth_embedding"]], "topology_preserving_thinning() (cyto_dl.models.im2im.utils.instance_seg.instancesegpreprocessd method)": [[76, "cyto_dl.models.im2im.utils.instance_seg.InstanceSegPreprocessd.topology_preserving_thinning"]], "noiseannealer (class in cyto_dl.models.im2im.utils.noise_annealer)": [[77, "cyto_dl.models.im2im.utils.noise_annealer.NoiseAnnealer"]], "cyto_dl.models.im2im.utils.noise_annealer": [[77, "module-cyto_dl.models.im2im.utils.noise_annealer"]], "update_noise() (cyto_dl.models.im2im.utils.noise_annealer.noiseannealer method)": [[77, "cyto_dl.models.im2im.utils.noise_annealer.NoiseAnnealer.update_noise"]], "cyto_dl.models.im2im.utils.postprocessing": [[78, "module-cyto_dl.models.im2im.utils.postprocessing"]], "actthreshlabel (class in cyto_dl.models.im2im.utils.postprocessing.act_thresh_label)": [[79, "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label.ActThreshLabel"]], "cyto_dl.models.im2im.utils.postprocessing.act_thresh_label": [[79, "module-cyto_dl.models.im2im.utils.postprocessing.act_thresh_label"]], "autothreshold (class in cyto_dl.models.im2im.utils.postprocessing.auto_thresh)": [[80, "cyto_dl.models.im2im.utils.postprocessing.auto_thresh.AutoThreshold"]], "cyto_dl.models.im2im.utils.postprocessing.auto_thresh": [[80, "module-cyto_dl.models.im2im.utils.postprocessing.auto_thresh"]], "dicttoim (class in cyto_dl.models.im2im.utils.postprocessing.dict_to_im)": [[81, "cyto_dl.models.im2im.utils.postprocessing.dict_to_im.DictToIm"]], "cyto_dl.models.im2im.utils.postprocessing.dict_to_im": [[81, "module-cyto_dl.models.im2im.utils.postprocessing.dict_to_im"]], "detach() (in module cyto_dl.models.im2im.utils.postprocessing.dict_to_im)": [[81, "cyto_dl.models.im2im.utils.postprocessing.dict_to_im.detach"]], "cyto_dl.models.jepa": [[82, "module-cyto_dl.models.jepa"]], "ijepa (class in cyto_dl.models.jepa.ijepa)": [[83, "cyto_dl.models.jepa.ijepa.IJEPA"]], "cyto_dl.models.jepa.ijepa": [[83, "module-cyto_dl.models.jepa.ijepa"]], "model_step() (cyto_dl.models.jepa.ijepa.ijepa method)": [[83, "cyto_dl.models.jepa.ijepa.IJEPA.model_step"]], "iwm (class in cyto_dl.models.jepa.iwm)": [[84, "cyto_dl.models.jepa.iwm.IWM"]], "cyto_dl.models.jepa.iwm": [[84, "module-cyto_dl.models.jepa.iwm"]], "extract_embeddings() (cyto_dl.models.jepa.iwm.iwm method)": [[84, "cyto_dl.models.jepa.iwm.IWM.extract_embeddings"]], "get_predict_masks() (cyto_dl.models.jepa.iwm.iwm method)": [[84, "cyto_dl.models.jepa.iwm.IWM.get_predict_masks"]], "model_step() (cyto_dl.models.jepa.iwm.iwm method)": [[84, "cyto_dl.models.jepa.iwm.IWM.model_step"]], "predict_step() (cyto_dl.models.jepa.iwm.iwm method)": [[84, "cyto_dl.models.jepa.iwm.IWM.predict_step"]], "test_step() (cyto_dl.models.jepa.iwm.iwm method)": [[84, "cyto_dl.models.jepa.iwm.IWM.test_step"]], "jepabase (class in cyto_dl.models.jepa.jepa_base)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase"]], "configure_optimizers() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.configure_optimizers"]], "cyto_dl.models.jepa.jepa_base": [[85, "module-cyto_dl.models.jepa.jepa_base"]], "forward() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.forward"]], "get_context_embeddings() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.get_context_embeddings"]], "get_mask() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.get_mask"]], "get_target_embeddings() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.get_target_embeddings"]], "model_step() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.model_step"]], "predict_step() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.predict_step"]], "remove_first_dim() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.remove_first_dim"]], "update_teacher() (cyto_dl.models.jepa.jepa_base.jepabase method)": [[85, "cyto_dl.models.jepa.jepa_base.JEPABase.update_teacher"]], "cyto_dl.models.utils": [[86, "module-cyto_dl.models.utils"]], "cyto_dl.models.utils.mlflow": [[87, "module-cyto_dl.models.utils.mlflow"]], "get_config() (in module cyto_dl.models.utils.mlflow)": [[87, "cyto_dl.models.utils.mlflow.get_config"]], "load_model_from_checkpoint() (in module cyto_dl.models.utils.mlflow)": [[87, "cyto_dl.models.utils.mlflow.load_model_from_checkpoint"]], "cyto_dl.models.utils.utils": [[88, "module-cyto_dl.models.utils.utils"]], "find_indices() (in module cyto_dl.models.utils.utils)": [[88, "cyto_dl.models.utils.utils.find_indices"]], "basevae (class in cyto_dl.models.vae.base_vae)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE"]], "calculate_elbo() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.calculate_elbo"]], "calculate_rcl() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.calculate_rcl"]], "calculate_rcl_dict() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.calculate_rcl_dict"]], "cyto_dl.models.vae.base_vae": [[90, "module-cyto_dl.models.vae.base_vae"]], "decode() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.decode"]], "encode() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.encode"]], "forward() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.forward"]], "model_step() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.model_step"]], "sample_z() (cyto_dl.models.vae.base_vae.basevae method)": [[90, "cyto_dl.models.vae.base_vae.BaseVAE.sample_z"]], "cyto_dl.models.vae.priors": [[99, "module-cyto_dl.models.vae.priors"]], "prior (class in cyto_dl.models.vae.priors.abstract_prior)": [[100, "cyto_dl.models.vae.priors.abstract_prior.Prior"]], "cyto_dl.models.vae.priors.abstract_prior": [[100, "module-cyto_dl.models.vae.priors.abstract_prior"]], "forward() (cyto_dl.models.vae.priors.abstract_prior.prior method)": [[100, "cyto_dl.models.vae.priors.abstract_prior.Prior.forward"]], "kl_divergence() (cyto_dl.models.vae.priors.abstract_prior.prior method)": [[100, "cyto_dl.models.vae.priors.abstract_prior.Prior.kl_divergence"]], "param_size (cyto_dl.models.vae.priors.abstract_prior.prior property)": [[100, "cyto_dl.models.vae.priors.abstract_prior.Prior.param_size"]], "sample() (cyto_dl.models.vae.priors.abstract_prior.prior method)": [[100, "cyto_dl.models.vae.priors.abstract_prior.Prior.sample"]], "diagonalgaussianprior (class in cyto_dl.models.vae.priors.gaussian)": [[101, "cyto_dl.models.vae.priors.gaussian.DiagonalGaussianPrior"]], "isotropicgaussianprior (class in cyto_dl.models.vae.priors.gaussian)": [[101, "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior"]], "compute_tc_penalty() (in module cyto_dl.models.vae.priors.gaussian)": [[101, "cyto_dl.models.vae.priors.gaussian.compute_tc_penalty"]], "cyto_dl.models.vae.priors.gaussian": [[101, "module-cyto_dl.models.vae.priors.gaussian"]], "forward() (cyto_dl.models.vae.priors.gaussian.diagonalgaussianprior method)": [[101, "cyto_dl.models.vae.priors.gaussian.DiagonalGaussianPrior.forward"]], "forward() (cyto_dl.models.vae.priors.gaussian.isotropicgaussianprior method)": [[101, "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior.forward"]], "kl_divergence() (cyto_dl.models.vae.priors.gaussian.diagonalgaussianprior class method)": [[101, "cyto_dl.models.vae.priors.gaussian.DiagonalGaussianPrior.kl_divergence"]], "kl_divergence() (cyto_dl.models.vae.priors.gaussian.isotropicgaussianprior class method)": [[101, "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior.kl_divergence"]], "param_size (cyto_dl.models.vae.priors.gaussian.diagonalgaussianprior property)": [[101, "cyto_dl.models.vae.priors.gaussian.DiagonalGaussianPrior.param_size"]], "param_size (cyto_dl.models.vae.priors.gaussian.isotropicgaussianprior property)": [[101, "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior.param_size"]], "sample() (cyto_dl.models.vae.priors.gaussian.isotropicgaussianprior class method)": [[101, "cyto_dl.models.vae.priors.gaussian.IsotropicGaussianPrior.sample"]], "identityprior (class in cyto_dl.models.vae.priors.identity_prior)": [[102, "cyto_dl.models.vae.priors.identity_prior.IdentityPrior"]], "cyto_dl.models.vae.priors.identity_prior": [[102, "module-cyto_dl.models.vae.priors.identity_prior"]], "forward() (cyto_dl.models.vae.priors.identity_prior.identityprior method)": [[102, "cyto_dl.models.vae.priors.identity_prior.IdentityPrior.forward"]], "param_size (cyto_dl.models.vae.priors.identity_prior.identityprior property)": [[102, "cyto_dl.models.vae.priors.identity_prior.IdentityPrior.param_size"]], "jointprior (class in cyto_dl.models.vae.priors.joint_prior)": [[103, "cyto_dl.models.vae.priors.joint_prior.JointPrior"]], "cyto_dl.models.vae.priors.joint_prior": [[103, "module-cyto_dl.models.vae.priors.joint_prior"]], "forward() (cyto_dl.models.vae.priors.joint_prior.jointprior method)": [[103, "cyto_dl.models.vae.priors.joint_prior.JointPrior.forward"]], "kl_divergence() (cyto_dl.models.vae.priors.joint_prior.jointprior method)": [[103, "cyto_dl.models.vae.priors.joint_prior.JointPrior.kl_divergence"]], "param_size (cyto_dl.models.vae.priors.joint_prior.jointprior property)": [[103, "cyto_dl.models.vae.priors.joint_prior.JointPrior.param_size"]], "sample() (cyto_dl.models.vae.priors.joint_prior.jointprior method)": [[103, "cyto_dl.models.vae.priors.joint_prior.JointPrior.sample"]], "cyto_dl.nn": [[105, "module-cyto_dl.nn"]], "cyto_dl.nn.discriminators": [[106, "module-cyto_dl.nn.discriminators"]], "multiscalediscriminator (class in cyto_dl.nn.discriminators.multi_scale_discriminator)": [[107, "cyto_dl.nn.discriminators.multi_scale_discriminator.MultiScaleDiscriminator"]], "cyto_dl.nn.discriminators.multi_scale_discriminator": [[107, "module-cyto_dl.nn.discriminators.multi_scale_discriminator"]], "forward() (cyto_dl.nn.discriminators.multi_scale_discriminator.multiscalediscriminator method)": [[107, "cyto_dl.nn.discriminators.multi_scale_discriminator.MultiScaleDiscriminator.forward"]], "nlayerdiscriminator (class in cyto_dl.nn.discriminators.n_layer_discriminator)": [[108, "cyto_dl.nn.discriminators.n_layer_discriminator.NLayerDiscriminator"]], "cyto_dl.nn.discriminators.n_layer_discriminator": [[108, "module-cyto_dl.nn.discriminators.n_layer_discriminator"]], "forward() (cyto_dl.nn.discriminators.n_layer_discriminator.nlayerdiscriminator method)": [[108, "cyto_dl.nn.discriminators.n_layer_discriminator.NLayerDiscriminator.forward"]], "cyto_dl.nn.head": [[109, "module-cyto_dl.nn.head"]], "basehead (class in cyto_dl.nn.head.base_head)": [[110, "cyto_dl.nn.head.base_head.BaseHead"]], "cyto_dl.nn.head.base_head": [[110, "module-cyto_dl.nn.head.base_head"]], "forward() (cyto_dl.nn.head.base_head.basehead method)": [[110, "cyto_dl.nn.head.base_head.BaseHead.forward"]], "generate_io_map() (cyto_dl.nn.head.base_head.basehead method)": [[110, "cyto_dl.nn.head.base_head.BaseHead.generate_io_map"]], "run_head() (cyto_dl.nn.head.base_head.basehead method)": [[110, "cyto_dl.nn.head.base_head.BaseHead.run_head"]], "update_params() (cyto_dl.nn.head.base_head.basehead method)": [[110, "cyto_dl.nn.head.base_head.BaseHead.update_params"]], "ganhead (class in cyto_dl.nn.head.gan_head)": [[111, "cyto_dl.nn.head.gan_head.GANHead"]], "cyto_dl.nn.head.gan_head": [[111, "module-cyto_dl.nn.head.gan_head"]], "forward() (cyto_dl.nn.head.gan_head.ganhead method)": [[111, "cyto_dl.nn.head.gan_head.GANHead.forward"]], "run_head() (cyto_dl.nn.head.gan_head.ganhead method)": [[111, "cyto_dl.nn.head.gan_head.GANHead.run_head"]], "ganhead_resize (class in cyto_dl.nn.head.gan_head_superres)": [[112, "cyto_dl.nn.head.gan_head_superres.GANHead_resize"]], "cyto_dl.nn.head.gan_head_superres": [[112, "module-cyto_dl.nn.head.gan_head_superres"]], "forward() (cyto_dl.nn.head.gan_head_superres.ganhead_resize method)": [[112, "cyto_dl.nn.head.gan_head_superres.GANHead_resize.forward"]], "maehead (class in cyto_dl.nn.head.mae_head)": [[113, "cyto_dl.nn.head.mae_head.MAEHead"]], "cyto_dl.nn.head.mae_head": [[113, "module-cyto_dl.nn.head.mae_head"]], "run_head() (cyto_dl.nn.head.mae_head.maehead method)": [[113, "cyto_dl.nn.head.mae_head.MAEHead.run_head"]], "maskhead (class in cyto_dl.nn.head.mask_head)": [[114, "cyto_dl.nn.head.mask_head.MaskHead"]], "cyto_dl.nn.head.mask_head": [[114, "module-cyto_dl.nn.head.mask_head"]], "run_head() (cyto_dl.nn.head.mask_head.maskhead method)": [[114, "cyto_dl.nn.head.mask_head.MaskHead.run_head"]], "resblockshead (class in cyto_dl.nn.head.res_blocks_head)": [[115, "cyto_dl.nn.head.res_blocks_head.ResBlocksHead"]], "cyto_dl.nn.head.res_blocks_head": [[115, "module-cyto_dl.nn.head.res_blocks_head"]], "forward() (cyto_dl.nn.head.res_blocks_head.resblockshead method)": [[115, "cyto_dl.nn.head.res_blocks_head.ResBlocksHead.forward"]], "vicreghead (class in cyto_dl.nn.head.vic_reg)": [[116, "cyto_dl.nn.head.vic_reg.VICRegHead"]], "cyto_dl.nn.head.vic_reg": [[116, "module-cyto_dl.nn.head.vic_reg"]], "run_head() (cyto_dl.nn.head.vic_reg.vicreghead method)": [[116, "cyto_dl.nn.head.vic_reg.VICRegHead.run_head"]], "hrskip (class in cyto_dl.nn.hr_skip)": [[117, "cyto_dl.nn.hr_skip.HRSkip"]], "cyto_dl.nn.hr_skip": [[117, "module-cyto_dl.nn.hr_skip"]], "forward() (cyto_dl.nn.hr_skip.hrskip method)": [[117, "cyto_dl.nn.hr_skip.HRSkip.forward"]], "cyto_dl.nn.losses": [[118, "module-cyto_dl.nn.losses"]], "adversarialloss (class in cyto_dl.nn.losses.adversarial_loss)": [[119, "cyto_dl.nn.losses.adversarial_loss.AdversarialLoss"]], "cyto_dl.nn.losses.adversarial_loss": [[119, "module-cyto_dl.nn.losses.adversarial_loss"]], "forward() (cyto_dl.nn.losses.adversarial_loss.adversarialloss method)": [[119, "cyto_dl.nn.losses.adversarial_loss.AdversarialLoss.forward"]], "chamferloss (class in cyto_dl.nn.losses.chamfer_loss)": [[120, "cyto_dl.nn.losses.chamfer_loss.ChamferLoss"]], "batch_pairwise_dist() (cyto_dl.nn.losses.chamfer_loss.chamferloss method)": [[120, "cyto_dl.nn.losses.chamfer_loss.ChamferLoss.batch_pairwise_dist"]], "batch_pairwise_dist2() (cyto_dl.nn.losses.chamfer_loss.chamferloss method)": [[120, "cyto_dl.nn.losses.chamfer_loss.ChamferLoss.batch_pairwise_dist2"]], "cyto_dl.nn.losses.chamfer_loss": [[120, "module-cyto_dl.nn.losses.chamfer_loss"]], "forward() (cyto_dl.nn.losses.chamfer_loss.chamferloss method)": [[120, "cyto_dl.nn.losses.chamfer_loss.ChamferLoss.forward"]], "cblogloss (class in cyto_dl.nn.losses.continuous_bernoulli)": [[121, "cyto_dl.nn.losses.continuous_bernoulli.CBLogLoss"]], "cyto_dl.nn.losses.continuous_bernoulli": [[121, "module-cyto_dl.nn.losses.continuous_bernoulli"]], "forward() (cyto_dl.nn.losses.continuous_bernoulli.cblogloss method)": [[121, "cyto_dl.nn.losses.continuous_bernoulli.CBLogLoss.forward"]], "cosineloss (class in cyto_dl.nn.losses.cosine_loss)": [[122, "cyto_dl.nn.losses.cosine_loss.CosineLoss"]], "cyto_dl.nn.losses.cosine_loss": [[122, "module-cyto_dl.nn.losses.cosine_loss"]], "forward() (cyto_dl.nn.losses.cosine_loss.cosineloss method)": [[122, "cyto_dl.nn.losses.cosine_loss.CosineLoss.forward"]], "ganloss (class in cyto_dl.nn.losses.gan_loss)": [[123, "cyto_dl.nn.losses.gan_loss.GANLoss"]], "pix2pixhd (class in cyto_dl.nn.losses.gan_loss)": [[123, "cyto_dl.nn.losses.gan_loss.Pix2PixHD"]], "cyto_dl.nn.losses.gan_loss": [[123, "module-cyto_dl.nn.losses.gan_loss"]], "get_feature_matching_loss() (cyto_dl.nn.losses.gan_loss.pix2pixhd method)": [[123, "cyto_dl.nn.losses.gan_loss.Pix2PixHD.get_feature_matching_loss"]], "get_gan_loss() (cyto_dl.nn.losses.gan_loss.pix2pixhd method)": [[123, "cyto_dl.nn.losses.gan_loss.Pix2PixHD.get_gan_loss"]], "get_target_tensor() (cyto_dl.nn.losses.gan_loss.ganloss method)": [[123, "cyto_dl.nn.losses.gan_loss.GANLoss.get_target_tensor"]], "gaussiannllloss (class in cyto_dl.nn.losses.gaussian_nll_loss)": [[124, "cyto_dl.nn.losses.gaussian_nll_loss.GaussianNLLLoss"]], "cyto_dl.nn.losses.gaussian_nll_loss": [[124, "module-cyto_dl.nn.losses.gaussian_nll_loss"]], "forward() (cyto_dl.nn.losses.gaussian_nll_loss.gaussiannllloss method)": [[124, "cyto_dl.nn.losses.gaussian_nll_loss.GaussianNLLLoss.forward"]], "cmap_loss (class in cyto_dl.nn.losses.loss_wrapper)": [[126, "cyto_dl.nn.losses.loss_wrapper.CMAP_loss"]], "losswrapper (class in cyto_dl.nn.losses.loss_wrapper)": [[126, "cyto_dl.nn.losses.loss_wrapper.LossWrapper"]], "cyto_dl.nn.losses.loss_wrapper": [[126, "module-cyto_dl.nn.losses.loss_wrapper"]], "thresholdloss (class in cyto_dl.nn.losses.threshold_loss)": [[128, "cyto_dl.nn.losses.threshold_loss.ThresholdLoss"]], "cyto_dl.nn.losses.threshold_loss": [[128, "module-cyto_dl.nn.losses.threshold_loss"]], "vicregloss (class in cyto_dl.nn.losses.vic_reg)": [[129, "cyto_dl.nn.losses.vic_reg.VICRegLoss"]], "cyto_dl.nn.losses.vic_reg": [[129, "module-cyto_dl.nn.losses.vic_reg"]], "forward() (cyto_dl.nn.losses.vic_reg.vicregloss method)": [[129, "cyto_dl.nn.losses.vic_reg.VICRegLoss.forward"]], "off_diagonal() (in module cyto_dl.nn.losses.vic_reg)": [[129, "cyto_dl.nn.losses.vic_reg.off_diagonal"]], "weibulllogloss (class in cyto_dl.nn.losses.weibull)": [[130, "cyto_dl.nn.losses.weibull.WeibullLogLoss"]], "cyto_dl.nn.losses.weibull": [[130, "module-cyto_dl.nn.losses.weibull"]], "forward() (cyto_dl.nn.losses.weibull.weibulllogloss method)": [[130, "cyto_dl.nn.losses.weibull.WeibullLogLoss.forward"]], "weibull_log_probs() (in module cyto_dl.nn.losses.weibull)": [[130, "cyto_dl.nn.losses.weibull.weibull_log_probs"]], "weightedmseloss (class in cyto_dl.nn.losses.weighted_mse_loss)": [[131, "cyto_dl.nn.losses.weighted_mse_loss.WeightedMSELoss"]], "cyto_dl.nn.losses.weighted_mse_loss": [[131, "module-cyto_dl.nn.losses.weighted_mse_loss"]], "forward() (cyto_dl.nn.losses.weighted_mse_loss.weightedmseloss method)": [[131, "cyto_dl.nn.losses.weighted_mse_loss.WeightedMSELoss.forward"]], "mlp (class in cyto_dl.nn.mlp)": [[132, "cyto_dl.nn.mlp.MLP"]], "cyto_dl.nn.mlp": [[132, "module-cyto_dl.nn.mlp"]], "forward() (cyto_dl.nn.mlp.mlp method)": [[132, "cyto_dl.nn.mlp.MLP.forward"]], "residualunit (class in cyto_dl.nn.res_unit)": [[138, "cyto_dl.nn.res_unit.ResidualUnit"]], "cyto_dl.nn.res_unit": [[138, "module-cyto_dl.nn.res_unit"]], "forward() (cyto_dl.nn.res_unit.residualunit method)": [[138, "cyto_dl.nn.res_unit.ResidualUnit.forward"]], "convpoolrelu (class in cyto_dl.nn.spatial_transformer)": [[139, "cyto_dl.nn.spatial_transformer.ConvPoolReLU"]], "stn (class in cyto_dl.nn.spatial_transformer)": [[139, "cyto_dl.nn.spatial_transformer.STN"]], "cyto_dl.nn.spatial_transformer": [[139, "module-cyto_dl.nn.spatial_transformer"]], "forward() (cyto_dl.nn.spatial_transformer.convpoolrelu method)": [[139, "cyto_dl.nn.spatial_transformer.ConvPoolReLU.forward"]], "forward() (cyto_dl.nn.spatial_transformer.stn method)": [[139, "cyto_dl.nn.spatial_transformer.STN.forward"]], "toggle() (cyto_dl.nn.spatial_transformer.stn method)": [[139, "cyto_dl.nn.spatial_transformer.STN.toggle"]], "trackclassifier (class in cyto_dl.nn.track_sequence_predictor)": [[140, "cyto_dl.nn.track_sequence_predictor.TrackClassifier"]], "cyto_dl.nn.track_sequence_predictor": [[140, "module-cyto_dl.nn.track_sequence_predictor"]], "forward() (cyto_dl.nn.track_sequence_predictor.trackclassifier method)": [[140, "cyto_dl.nn.track_sequence_predictor.TrackClassifier.forward"]], "positionalencoding1d() (in module cyto_dl.nn.track_sequence_predictor)": [[140, "cyto_dl.nn.track_sequence_predictor.positionalencoding1d"]], "cyto_dl.nn.vits": [[141, "module-cyto_dl.nn.vits"]], "cyto_dl.nn.vits.blocks": [[142, "module-cyto_dl.nn.vits.blocks"]], "crossattention (class in cyto_dl.nn.vits.blocks.cross_attention)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossAttention"]], "crossattentionblock (class in cyto_dl.nn.vits.blocks.cross_attention)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossAttentionBlock"]], "crossselfblock (class in cyto_dl.nn.vits.blocks.cross_attention)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossSelfBlock"]], "mlp (class in cyto_dl.nn.vits.blocks.cross_attention)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.Mlp"]], "cyto_dl.nn.vits.blocks.cross_attention": [[143, "module-cyto_dl.nn.vits.blocks.cross_attention"]], "forward() (cyto_dl.nn.vits.blocks.cross_attention.crossattention method)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossAttention.forward"]], "forward() (cyto_dl.nn.vits.blocks.cross_attention.crossattentionblock method)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossAttentionBlock.forward"]], "forward() (cyto_dl.nn.vits.blocks.cross_attention.crossselfblock method)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.CrossSelfBlock.forward"]], "forward() (cyto_dl.nn.vits.blocks.cross_attention.mlp method)": [[143, "cyto_dl.nn.vits.blocks.cross_attention.Mlp.forward"]], "intermediateweigher (class in cyto_dl.nn.vits.blocks.intermediate_weigher)": [[144, "cyto_dl.nn.vits.blocks.intermediate_weigher.IntermediateWeigher"]], "cyto_dl.nn.vits.blocks.intermediate_weigher": [[144, "module-cyto_dl.nn.vits.blocks.intermediate_weigher"]], "forward() (cyto_dl.nn.vits.blocks.intermediate_weigher.intermediateweigher method)": [[144, "cyto_dl.nn.vits.blocks.intermediate_weigher.IntermediateWeigher.forward"]], "hierablock (class in cyto_dl.nn.vits.blocks.masked_unit_attention)": [[145, "cyto_dl.nn.vits.blocks.masked_unit_attention.HieraBlock"]], "maskunitattention (class in cyto_dl.nn.vits.blocks.masked_unit_attention)": [[145, "cyto_dl.nn.vits.blocks.masked_unit_attention.MaskUnitAttention"]], "cyto_dl.nn.vits.blocks.masked_unit_attention": [[145, "module-cyto_dl.nn.vits.blocks.masked_unit_attention"]], "forward() (cyto_dl.nn.vits.blocks.masked_unit_attention.hierablock method)": [[145, "cyto_dl.nn.vits.blocks.masked_unit_attention.HieraBlock.forward"]], "forward() (cyto_dl.nn.vits.blocks.masked_unit_attention.maskunitattention method)": [[145, "cyto_dl.nn.vits.blocks.masked_unit_attention.MaskUnitAttention.forward"]], "cyto_dl.nn.vits.blocks.patchify": [[146, "module-cyto_dl.nn.vits.blocks.patchify"]], "patchify (class in cyto_dl.nn.vits.blocks.patchify.patchify)": [[147, "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify"]], "create_img2token() (cyto_dl.nn.vits.blocks.patchify.patchify.patchify method)": [[147, "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify.create_img2token"]], "cyto_dl.nn.vits.blocks.patchify.patchify": [[147, "module-cyto_dl.nn.vits.blocks.patchify.patchify"]], "extract_visible_tokens() (cyto_dl.nn.vits.blocks.patchify.patchify.patchify method)": [[147, "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify.extract_visible_tokens"]], "get_mask_args() (cyto_dl.nn.vits.blocks.patchify.patchify.patchify method)": [[147, "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify.get_mask_args"]], "img2token (cyto_dl.nn.vits.blocks.patchify.patchify.patchify property)": [[147, "cyto_dl.nn.vits.blocks.patchify.patchify.Patchify.img2token"]], "patchifybase (class in cyto_dl.nn.vits.blocks.patchify.patchify_base)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase"]], "create_conv() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.create_conv"]], "create_patch2img() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.create_patch2img"]], "cyto_dl.nn.vits.blocks.patchify.patchify_base": [[148, "module-cyto_dl.nn.vits.blocks.patchify.patchify_base"]], "extract_visible_tokens() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.extract_visible_tokens"]], "forward() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.forward"]], "get_mask() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.get_mask"]], "get_mask_args() (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase method)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.get_mask_args"]], "img2token (cyto_dl.nn.vits.blocks.patchify.patchify_base.patchifybase property)": [[148, "cyto_dl.nn.vits.blocks.patchify.patchify_base.PatchifyBase.img2token"]], "patchifyhiera (class in cyto_dl.nn.vits.blocks.patchify.patchify_hiera)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera"]], "create_img2token() (cyto_dl.nn.vits.blocks.patchify.patchify_hiera.patchifyhiera method)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera.create_img2token"]], "cyto_dl.nn.vits.blocks.patchify.patchify_hiera": [[149, "module-cyto_dl.nn.vits.blocks.patchify.patchify_hiera"]], "extract_visible_tokens() (cyto_dl.nn.vits.blocks.patchify.patchify_hiera.patchifyhiera method)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera.extract_visible_tokens"]], "get_mask_args() (cyto_dl.nn.vits.blocks.patchify.patchify_hiera.patchifyhiera method)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera.get_mask_args"]], "img2token (cyto_dl.nn.vits.blocks.patchify.patchify_hiera.patchifyhiera property)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.PatchifyHiera.img2token"]], "take_indexes_mask() (in module cyto_dl.nn.vits.blocks.patchify.patchify_hiera)": [[149, "cyto_dl.nn.vits.blocks.patchify.patchify_hiera.take_indexes_mask"]], "crossmae_decoder (class in cyto_dl.nn.vits.decoder)": [[150, "cyto_dl.nn.vits.decoder.CrossMAE_Decoder"]], "mae_decoder (class in cyto_dl.nn.vits.decoder)": [[150, "cyto_dl.nn.vits.decoder.MAE_Decoder"]], "add_mask_tokens() (cyto_dl.nn.vits.decoder.mae_decoder method)": [[150, "cyto_dl.nn.vits.decoder.MAE_Decoder.add_mask_tokens"]], "adjust_indices_for_cls() (cyto_dl.nn.vits.decoder.mae_decoder method)": [[150, "cyto_dl.nn.vits.decoder.MAE_Decoder.adjust_indices_for_cls"]], "cyto_dl.nn.vits.decoder": [[150, "module-cyto_dl.nn.vits.decoder"]], "forward() (cyto_dl.nn.vits.decoder.crossmae_decoder method)": [[150, "cyto_dl.nn.vits.decoder.CrossMAE_Decoder.forward"]], "forward() (cyto_dl.nn.vits.decoder.mae_decoder method)": [[150, "cyto_dl.nn.vits.decoder.MAE_Decoder.forward"]], "init_weight() (cyto_dl.nn.vits.decoder.mae_decoder method)": [[150, "cyto_dl.nn.vits.decoder.MAE_Decoder.init_weight"]], "hieraencoder (class in cyto_dl.nn.vits.encoder)": [[151, "cyto_dl.nn.vits.encoder.HieraEncoder"]], "jepaencoder (class in cyto_dl.nn.vits.encoder)": [[151, "cyto_dl.nn.vits.encoder.JEPAEncoder"]], "mae_encoder (class in cyto_dl.nn.vits.encoder)": [[151, "cyto_dl.nn.vits.encoder.MAE_Encoder"]], "spatialmerger (class in cyto_dl.nn.vits.encoder)": [[151, "cyto_dl.nn.vits.encoder.SpatialMerger"]], "cyto_dl.nn.vits.encoder": [[151, "module-cyto_dl.nn.vits.encoder"]], "forward() (cyto_dl.nn.vits.encoder.hieraencoder method)": [[151, "cyto_dl.nn.vits.encoder.HieraEncoder.forward"]], "forward() (cyto_dl.nn.vits.encoder.jepaencoder method)": [[151, "cyto_dl.nn.vits.encoder.JEPAEncoder.forward"]], "forward() (cyto_dl.nn.vits.encoder.mae_encoder method)": [[151, "cyto_dl.nn.vits.encoder.MAE_Encoder.forward"]], "forward() (cyto_dl.nn.vits.encoder.spatialmerger method)": [[151, "cyto_dl.nn.vits.encoder.SpatialMerger.forward"]], "init_weight() (cyto_dl.nn.vits.encoder.mae_encoder method)": [[151, "cyto_dl.nn.vits.encoder.MAE_Encoder.init_weight"]], "hieramae (class in cyto_dl.nn.vits.mae)": [[152, "cyto_dl.nn.vits.mae.HieraMAE"]], "mae (class in cyto_dl.nn.vits.mae)": [[152, "cyto_dl.nn.vits.mae.MAE"]], "mae_base (class in cyto_dl.nn.vits.mae)": [[152, "cyto_dl.nn.vits.mae.MAE_Base"]], "cyto_dl.nn.vits.mae": [[152, "module-cyto_dl.nn.vits.mae"]], "decoder (cyto_dl.nn.vits.mae.hieramae property)": [[152, "cyto_dl.nn.vits.mae.HieraMAE.decoder"]], "decoder (cyto_dl.nn.vits.mae.mae property)": [[152, "cyto_dl.nn.vits.mae.MAE.decoder"]], "decoder (cyto_dl.nn.vits.mae.mae_base property)": [[152, "cyto_dl.nn.vits.mae.MAE_Base.decoder"]], "encoder (cyto_dl.nn.vits.mae.hieramae property)": [[152, "cyto_dl.nn.vits.mae.HieraMAE.encoder"]], "encoder (cyto_dl.nn.vits.mae.mae property)": [[152, "cyto_dl.nn.vits.mae.MAE.encoder"]], "encoder (cyto_dl.nn.vits.mae.mae_base property)": [[152, "cyto_dl.nn.vits.mae.MAE_Base.encoder"]], "forward() (cyto_dl.nn.vits.mae.mae_base method)": [[152, "cyto_dl.nn.vits.mae.MAE_Base.forward"]], "init_decoder() (cyto_dl.nn.vits.mae.mae_base method)": [[152, "cyto_dl.nn.vits.mae.MAE_Base.init_decoder"]], "init_encoder() (cyto_dl.nn.vits.mae.mae_base method)": [[152, "cyto_dl.nn.vits.mae.MAE_Base.init_encoder"]], "iwmpredictor (class in cyto_dl.nn.vits.predictor)": [[153, "cyto_dl.nn.vits.predictor.IWMPredictor"]], "jepapredictor (class in cyto_dl.nn.vits.predictor)": [[153, "cyto_dl.nn.vits.predictor.JEPAPredictor"]], "cyto_dl.nn.vits.predictor": [[153, "module-cyto_dl.nn.vits.predictor"]], "forward() (cyto_dl.nn.vits.predictor.iwmpredictor method)": [[153, "cyto_dl.nn.vits.predictor.IWMPredictor.forward"]], "forward() (cyto_dl.nn.vits.predictor.jepapredictor method)": [[153, "cyto_dl.nn.vits.predictor.JEPAPredictor.forward"]], "init_weight() (cyto_dl.nn.vits.predictor.jepapredictor method)": [[153, "cyto_dl.nn.vits.predictor.JEPAPredictor.init_weight"]], "predict_target_features() (cyto_dl.nn.vits.predictor.jepapredictor method)": [[153, "cyto_dl.nn.vits.predictor.JEPAPredictor.predict_target_features"]], "encodedskip (class in cyto_dl.nn.vits.seg)": [[154, "cyto_dl.nn.vits.seg.EncodedSkip"]], "seg_vit (class in cyto_dl.nn.vits.seg)": [[154, "cyto_dl.nn.vits.seg.Seg_ViT"]], "superresdecoder (class in cyto_dl.nn.vits.seg)": [[154, "cyto_dl.nn.vits.seg.SuperresDecoder"]], "cyto_dl.nn.vits.seg": [[154, "module-cyto_dl.nn.vits.seg"]], "forward() (cyto_dl.nn.vits.seg.encodedskip method)": [[154, "cyto_dl.nn.vits.seg.EncodedSkip.forward"]], "forward() (cyto_dl.nn.vits.seg.seg_vit method)": [[154, "cyto_dl.nn.vits.seg.Seg_ViT.forward"]], "forward() (cyto_dl.nn.vits.seg.superresdecoder method)": [[154, "cyto_dl.nn.vits.seg.SuperresDecoder.forward"]], "cyto_dl.nn.vits.utils": [[155, "module-cyto_dl.nn.vits.utils"]], "get_positional_embedding() (in module cyto_dl.nn.vits.utils)": [[155, "cyto_dl.nn.vits.utils.get_positional_embedding"]], "match_tuple_dimensions() (in module cyto_dl.nn.vits.utils)": [[155, "cyto_dl.nn.vits.utils.match_tuple_dimensions"]], "random_indexes() (in module cyto_dl.nn.vits.utils)": [[155, "cyto_dl.nn.vits.utils.random_indexes"]], "take_indexes() (in module cyto_dl.nn.vits.utils)": [[155, "cyto_dl.nn.vits.utils.take_indexes"]], "cyto_dl.point_cloud": [[156, "module-cyto_dl.point_cloud"]], "cyto_dl.train": [[159, "module-cyto_dl.train"]], "main() (in module cyto_dl.train)": [[159, "cyto_dl.train.main"]], "cyto_dl.utils": [[160, "module-cyto_dl.utils"]], "create_dataloader() (in module cyto_dl.utils.array)": [[161, "cyto_dl.utils.array.create_dataloader"]], "cyto_dl.utils.array": [[161, "module-cyto_dl.utils.array"]], "extract_array_predictions() (in module cyto_dl.utils.array)": [[161, "cyto_dl.utils.array.extract_array_predictions"]], "cyto_dl.utils.checkpoint": [[162, "module-cyto_dl.utils.checkpoint"]], "load_checkpoint() (in module cyto_dl.utils.checkpoint)": [[162, "cyto_dl.utils.checkpoint.load_checkpoint"]], "cyto_dl.utils.config": [[163, "module-cyto_dl.utils.config"]], "is_config() (in module cyto_dl.utils.config)": [[163, "cyto_dl.utils.config.is_config"]], "kv_to_dict() (in module cyto_dl.utils.config)": [[163, "cyto_dl.utils.config.kv_to_dict"]], "remove_aux_key() (in module cyto_dl.utils.config)": [[163, "cyto_dl.utils.config.remove_aux_key"]], "cyto_dl.utils.download_test_data": [[164, "module-cyto_dl.utils.download_test_data"]], "delete_test_data() (in module cyto_dl.utils.download_test_data)": [[164, "cyto_dl.utils.download_test_data.delete_test_data"]], "download_test_data() (in module cyto_dl.utils.download_test_data)": [[164, "cyto_dl.utils.download_test_data.download_test_data"]], "parse_s3_path() (in module cyto_dl.utils.download_test_data)": [[164, "cyto_dl.utils.download_test_data.parse_s3_path"]], "setup_paths() (in module cyto_dl.utils.download_test_data)": [[164, "cyto_dl.utils.download_test_data.setup_paths"]], "dummydatamodule (class in cyto_dl.utils.dummy_dataset)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule"]], "dummydataset (class in cyto_dl.utils.dummy_dataset)": [[165, "cyto_dl.utils.dummy_dataset.DummyDataset"]], "cyto_dl.utils.dummy_dataset": [[165, "module-cyto_dl.utils.dummy_dataset"]], "generate_img() (cyto_dl.utils.dummy_dataset.dummydataset method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDataset.generate_img"]], "get_dataloader() (cyto_dl.utils.dummy_dataset.dummydatamodule method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule.get_dataloader"]], "predict_dataloader() (cyto_dl.utils.dummy_dataset.dummydatamodule method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule.predict_dataloader"]], "test_dataloader() (cyto_dl.utils.dummy_dataset.dummydatamodule method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule.test_dataloader"]], "train_dataloader() (cyto_dl.utils.dummy_dataset.dummydatamodule method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule.train_dataloader"]], "val_dataloader() (cyto_dl.utils.dummy_dataset.dummydatamodule method)": [[165, "cyto_dl.utils.dummy_dataset.DummyDatamodule.val_dataloader"]], "cyto_dl.utils.pylogger": [[166, "module-cyto_dl.utils.pylogger"]], "get_pylogger() (in module cyto_dl.utils.pylogger)": [[166, "cyto_dl.utils.pylogger.get_pylogger"]], "cyto_dl.utils.rich_utils": [[167, "module-cyto_dl.utils.rich_utils"]], "enforce_tags() (in module cyto_dl.utils.rich_utils)": [[167, "cyto_dl.utils.rich_utils.enforce_tags"]], "print_config_tree() (in module cyto_dl.utils.rich_utils)": [[167, "cyto_dl.utils.rich_utils.print_config_tree"]], "rotationmodule (class in cyto_dl.utils.rotation)": [[168, "cyto_dl.utils.rotation.RotationModule"]], "compute_rotation_matrix() (cyto_dl.utils.rotation.rotationmodule method)": [[168, "cyto_dl.utils.rotation.RotationModule.compute_rotation_matrix"]], "cyto_dl.utils.rotation": [[168, "module-cyto_dl.utils.rotation"]], "get_rotation_matrix_so2() (in module cyto_dl.utils.rotation)": [[168, "cyto_dl.utils.rotation.get_rotation_matrix_so2"]], "get_rotation_matrix_so3() (in module cyto_dl.utils.rotation)": [[168, "cyto_dl.utils.rotation.get_rotation_matrix_so3"]], "close_loggers() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.close_loggers"]], "cyto_dl.utils.template_utils": [[172, "module-cyto_dl.utils.template_utils"]], "extras() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.extras"]], "get_metric_value() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.get_metric_value"]], "instantiate_callbacks() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.instantiate_callbacks"]], "instantiate_loggers() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.instantiate_loggers"]], "log_hyperparameters() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.log_hyperparameters"]], "save_file() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.save_file"]], "task_wrapper() (in module cyto_dl.utils.template_utils)": [[172, "cyto_dl.utils.template_utils.task_wrapper"]]}})
\ No newline at end of file