MANIFEST.in
README.md
pyproject.toml
src/vesuvius/__init__.py
src/vesuvius/_version.py
src/vesuvius/vesuvius.py
src/vesuvius.egg-info/PKG-INFO
src/vesuvius.egg-info/SOURCES.txt
src/vesuvius.egg-info/dependency_links.txt
src/vesuvius.egg-info/entry_points.txt
src/vesuvius.egg-info/requires.txt
src/vesuvius.egg-info/top_level.txt
src/vesuvius/data/__init__.py
src/vesuvius/data/chunks_filter.py
src/vesuvius/data/utils.py
src/vesuvius/data/vc_dataset.py
src/vesuvius/data/volume.py
src/vesuvius/data/paths/__init__.py
src/vesuvius/data/paths/fetcher.py
src/vesuvius/data/paths/local.py
src/vesuvius/data/paths/parser.py
src/vesuvius/examples/__init__.py
src/vesuvius/exps_2d_model/batch_fit_cosine_grid.py
src/vesuvius/exps_2d_model/batch_unet_tiled_infer.py
src/vesuvius/exps_2d_model/cli_data.py
src/vesuvius/exps_2d_model/cli_json.py
src/vesuvius/exps_2d_model/cli_model.py
src/vesuvius/exps_2d_model/cli_opt.py
src/vesuvius/exps_2d_model/cli_vis.py
src/vesuvius/exps_2d_model/common.py
src/vesuvius/exps_2d_model/convert_fit_zarr_to_vc3d_omezarr.py
src/vesuvius/exps_2d_model/fit.py
src/vesuvius/exps_2d_model/fit2tifxyz.py
src/vesuvius/exps_2d_model/fit_data.py
src/vesuvius/exps_2d_model/fit_service.py
src/vesuvius/exps_2d_model/gen_post_data.py
src/vesuvius/exps_2d_model/inv_map.py
src/vesuvius/exps_2d_model/mask_schedule.py
src/vesuvius/exps_2d_model/model.py
src/vesuvius/exps_2d_model/opt_loss_corr.py
src/vesuvius/exps_2d_model/opt_loss_data.py
src/vesuvius/exps_2d_model/opt_loss_dir.py
src/vesuvius/exps_2d_model/opt_loss_geom.py
src/vesuvius/exps_2d_model/opt_loss_gradmag.py
src/vesuvius/exps_2d_model/opt_loss_min_dist.py
src/vesuvius/exps_2d_model/opt_loss_mod.py
src/vesuvius/exps_2d_model/opt_loss_pred_dt.py
src/vesuvius/exps_2d_model/opt_loss_step.py
src/vesuvius/exps_2d_model/optimizer.py
src/vesuvius/exps_2d_model/point_constraints.py
src/vesuvius/exps_2d_model/preprocess_cos_omezarr.py
src/vesuvius/exps_2d_model/review_tiffs.py
src/vesuvius/exps_2d_model/setup.py
src/vesuvius/exps_2d_model/tiled_infer.py
src/vesuvius/exps_2d_model/train_unet.py
src/vesuvius/exps_2d_model/vis.py
src/vesuvius/exps_2d_model/old/fit.py
src/vesuvius/exps_2d_model/old/fit_cosine_grid.py
src/vesuvius/exps_2d_model/old/fit_data.py
src/vesuvius/exps_2d_model/old/fit_helpers.py
src/vesuvius/exps_2d_model/old/fit_loss_aux.py
src/vesuvius/exps_2d_model/old/fit_loss_data.py
src/vesuvius/exps_2d_model/old/fit_loss_geom.py
src/vesuvius/exps_2d_model/old/fit_loss_gradmag.py
src/vesuvius/exps_2d_model/old/fit_mask.py
src/vesuvius/exps_2d_model/old/fit_model.py
src/vesuvius/exps_2d_model/old/fit_vis.py
src/vesuvius/image_proc/__init__.py
src/vesuvius/image_proc/distance.py
src/vesuvius/image_proc/helpers.py
src/vesuvius/image_proc/resampling.py
src/vesuvius/image_proc/blur/__init__.py
src/vesuvius/image_proc/blur/gaussian.py
src/vesuvius/image_proc/features/__init__.py
src/vesuvius/image_proc/features/curvature.py
src/vesuvius/image_proc/features/ridges_vessels.py
src/vesuvius/image_proc/features/skeletonization.py
src/vesuvius/image_proc/geometry/__init__.py
src/vesuvius/image_proc/geometry/alpha_wrap.py
src/vesuvius/image_proc/geometry/structure_tensor.py
src/vesuvius/image_proc/geometry/diffusion/__init__.py
src/vesuvius/image_proc/geometry/diffusion/coherence_enhancing_diffusion.py
src/vesuvius/image_proc/intensity/__init__.py
src/vesuvius/image_proc/intensity/normalization.py
src/vesuvius/image_proc/mesh/__init__.py
src/vesuvius/image_proc/mesh/affine.py
src/vesuvius/image_proc/run/__init__.py
src/vesuvius/image_proc/run/alpha_wrap_labels.py
src/vesuvius/image_proc/run/apply_affine.py
src/vesuvius/image_proc/run/apply_masks.py
src/vesuvius/image_proc/run/ced.py
src/vesuvius/image_proc/run/edt_frangi_label.py
src/vesuvius/image_proc/run/filter_branching_labels.py
src/vesuvius/image_proc/run/generate_mutex_graph.py
src/vesuvius/image_proc/run/prediction_grid.py
src/vesuvius/image_proc/run/remap_values.py
src/vesuvius/image_proc/run/stack_composite_tifs.py
src/vesuvius/image_proc/run/voxelize_objs.py
src/vesuvius/image_proc/run/zarr_tasks/__init__.py
src/vesuvius/image_proc/run/zarr_tasks/__main__.py
src/vesuvius/image_proc/run/zarr_tasks/base.py
src/vesuvius/image_proc/run/zarr_tasks/registry.py
src/vesuvius/image_proc/run/zarr_tasks/utils.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/__init__.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/edt_dilate.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/export_tifs.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/merge.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/recompress.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/remap.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/resize.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/scale.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/threshold.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/transpose.py
src/vesuvius/image_proc/run/zarr_tasks/tasks/zero_range.py
src/vesuvius/install/__init__.py
src/vesuvius/install/accept_terms.py
src/vesuvius/install/custom_install.py
src/vesuvius/install/configs/cubes.yaml
src/vesuvius/install/configs/scrolls.yaml
src/vesuvius/models/__init__.py
src/vesuvius/models/utils.py
src/vesuvius/models/augmentation/__init__.py
src/vesuvius/models/augmentation/dataloading/__init__.py
src/vesuvius/models/augmentation/helpers/__init__.py
src/vesuvius/models/augmentation/helpers/scalar_type.py
src/vesuvius/models/augmentation/pipelines/__init__.py
src/vesuvius/models/augmentation/pipelines/training_transforms.py
src/vesuvius/models/augmentation/transforms/__init__.py
src/vesuvius/models/augmentation/transforms/base/__init__.py
src/vesuvius/models/augmentation/transforms/base/basic_transform.py
src/vesuvius/models/augmentation/transforms/intensity/__init__.py
src/vesuvius/models/augmentation/transforms/intensity/brightness.py
src/vesuvius/models/augmentation/transforms/intensity/contrast.py
src/vesuvius/models/augmentation/transforms/intensity/gamma.py
src/vesuvius/models/augmentation/transforms/intensity/gaussian_noise.py
src/vesuvius/models/augmentation/transforms/intensity/illumination.py
src/vesuvius/models/augmentation/transforms/intensity/inversion.py
src/vesuvius/models/augmentation/transforms/intensity/random_clip.py
src/vesuvius/models/augmentation/transforms/local/__init__.py
src/vesuvius/models/augmentation/transforms/local/brightness_gradient.py
src/vesuvius/models/augmentation/transforms/local/local_contrast.py
src/vesuvius/models/augmentation/transforms/local/local_gamma.py
src/vesuvius/models/augmentation/transforms/local/local_smoothing.py
src/vesuvius/models/augmentation/transforms/local/local_transform.py
src/vesuvius/models/augmentation/transforms/nnunet/__init__.py
src/vesuvius/models/augmentation/transforms/nnunet/random_binary_operator.py
src/vesuvius/models/augmentation/transforms/nnunet/remove_connected_components.py
src/vesuvius/models/augmentation/transforms/nnunet/seg_to_onehot.py
src/vesuvius/models/augmentation/transforms/noise/__init__.py
src/vesuvius/models/augmentation/transforms/noise/extranoisetransforms.py
src/vesuvius/models/augmentation/transforms/noise/gaussian_blur.py
src/vesuvius/models/augmentation/transforms/noise/median_filter.py
src/vesuvius/models/augmentation/transforms/noise/sharpen.py
src/vesuvius/models/augmentation/transforms/spatial/__init__.py
src/vesuvius/models/augmentation/transforms/spatial/low_resolution.py
src/vesuvius/models/augmentation/transforms/spatial/mirroring.py
src/vesuvius/models/augmentation/transforms/spatial/rot90.py
src/vesuvius/models/augmentation/transforms/spatial/sheet_compression.py
src/vesuvius/models/augmentation/transforms/spatial/spatial.py
src/vesuvius/models/augmentation/transforms/spatial/thick_slice.py
src/vesuvius/models/augmentation/transforms/spatial/transpose.py
src/vesuvius/models/augmentation/transforms/utils/__init__.py
src/vesuvius/models/augmentation/transforms/utils/compose.py
src/vesuvius/models/augmentation/transforms/utils/cropping.py
src/vesuvius/models/augmentation/transforms/utils/deep_supervision_downsampling.py
src/vesuvius/models/augmentation/transforms/utils/morphological_closing.py
src/vesuvius/models/augmentation/transforms/utils/nnunet_masking.py
src/vesuvius/models/augmentation/transforms/utils/oneoftransform.py
src/vesuvius/models/augmentation/transforms/utils/perf.py
src/vesuvius/models/augmentation/transforms/utils/pseudo2d.py
src/vesuvius/models/augmentation/transforms/utils/random.py
src/vesuvius/models/augmentation/transforms/utils/remove_label.py
src/vesuvius/models/augmentation/transforms/utils/seg_to_regions.py
src/vesuvius/models/augmentation/transforms/utils/skeleton_transform.py
src/vesuvius/models/build/__init__.py
src/vesuvius/models/build/activations.py
src/vesuvius/models/build/build_network_from_config.py
src/vesuvius/models/build/decoder.py
src/vesuvius/models/build/encoder.py
src/vesuvius/models/build/primus_wrapper.py
src/vesuvius/models/build/resblocks.py
src/vesuvius/models/build/simple_conv_blocks.py
src/vesuvius/models/build/transformers/__init__.py
src/vesuvius/models/build/transformers/abstract_arch.py
src/vesuvius/models/build/transformers/eva.py
src/vesuvius/models/build/transformers/flash_rope.py
src/vesuvius/models/build/transformers/patch_encode_decode.py
src/vesuvius/models/build/transformers/pope.py
src/vesuvius/models/build/transformers/primus.py
src/vesuvius/models/configuration/__init__.py
src/vesuvius/models/configuration/config_manager.py
src/vesuvius/models/configuration/config_utils.py
src/vesuvius/models/datasets/__init__.py
src/vesuvius/models/datasets/find_valid_patches.py
src/vesuvius/models/datasets/intensity_properties.py
src/vesuvius/models/datasets/intensity_sampling.py
src/vesuvius/models/datasets/mutex_affinity_dataset.py
src/vesuvius/models/datasets/save_valid_patches.py
src/vesuvius/models/datasets/zarr_dataset.py
src/vesuvius/models/evaluation/__init__.py
src/vesuvius/models/evaluation/affinity_metrics.py
src/vesuvius/models/evaluation/base_metric.py
src/vesuvius/models/evaluation/connected_components.py
src/vesuvius/models/evaluation/critical_components.py
src/vesuvius/models/evaluation/hausdorff.py
src/vesuvius/models/evaluation/iou_dice.py
src/vesuvius/models/evaluation/mean_average_precision.py
src/vesuvius/models/evaluation/precision_recall_f1.py
src/vesuvius/models/evaluation/skeleton_branch_points.py
src/vesuvius/models/evaluation/voi.py
src/vesuvius/models/preprocessing/pack_tifs_to_sparse_zarr.py
src/vesuvius/models/preprocessing/patches/__init__.py
src/vesuvius/models/preprocessing/patches/cache.py
src/vesuvius/models/preprocessing/patches/cli.py
src/vesuvius/models/preprocessing/patches/generate.py
src/vesuvius/models/run/__init__.py
src/vesuvius/models/run/blending.py
src/vesuvius/models/run/finalize_outputs.py
src/vesuvius/models/run/inference.py
src/vesuvius/models/run/tta.py
src/vesuvius/models/run/external_models/__init__.py
src/vesuvius/models/run/external_models/load_resnet.py
src/vesuvius/models/training/__init__.py
src/vesuvius/models/training/cli.py
src/vesuvius/models/training/lr_schedulers.py
src/vesuvius/models/training/normalization.py
src/vesuvius/models/training/optimizers.py
src/vesuvius/models/training/save_checkpoint.py
src/vesuvius/models/training/train.py
src/vesuvius/models/training/wandb_logging.py
src/vesuvius/models/training/auxiliary_tasks/__init__.py
src/vesuvius/models/training/auxiliary_tasks/apply_aux_targets.py
src/vesuvius/models/training/auxiliary_tasks/aux_distance_transform.py
src/vesuvius/models/training/auxiliary_tasks/aux_inplane_direction.py
src/vesuvius/models/training/auxiliary_tasks/aux_nearest_component.py
src/vesuvius/models/training/auxiliary_tasks/aux_structure_tensor.py
src/vesuvius/models/training/auxiliary_tasks/aux_surface_normals.py
src/vesuvius/models/training/loss/__init__.py
src/vesuvius/models/training/loss/betti_losses.py
src/vesuvius/models/training/loss/ect_loss.py
src/vesuvius/models/training/loss/loss_helpers.py
src/vesuvius/models/training/loss/losses.py
src/vesuvius/models/training/loss/nnunet_losses.py
src/vesuvius/models/training/loss/robustcrossentropy.py
src/vesuvius/models/training/loss/sigreg.py
src/vesuvius/models/training/loss/skeleton_recall.py
src/vesuvius/models/training/loss/spherical_betti_loss.py
src/vesuvius/models/training/loss/surface_frame.py
src/vesuvius/models/training/trainers/__init__.py
src/vesuvius/models/training/trainers/mutex_affinity_trainer.py
src/vesuvius/models/training/trainers/surface_frame_trainer.py
src/vesuvius/models/training/trainers/auxiliary/__init__.py
src/vesuvius/models/training/trainers/auxiliary/base_aux_trainer.py
src/vesuvius/models/training/trainers/auxiliary/distance_transform_trainer.py
src/vesuvius/models/training/trainers/auxiliary/inplane_direction_trainer.py
src/vesuvius/models/training/trainers/auxiliary/nearest_component_trainer.py
src/vesuvius/models/training/trainers/auxiliary/structure_tensor_trainer.py
src/vesuvius/models/training/trainers/auxiliary/surface_normals_trainer.py
src/vesuvius/models/training/trainers/self_supervised/__init__.py
src/vesuvius/models/training/trainers/self_supervised/train_eva_mae.py
src/vesuvius/models/training/trainers/self_supervised/train_finetune_lejepa.py
src/vesuvius/models/training/trainers/self_supervised/train_finetune_mae_unet.py
src/vesuvius/models/training/trainers/self_supervised/train_lejepa.py
src/vesuvius/models/training/trainers/self_supervised/train_unet_mae.py
src/vesuvius/models/training/trainers/self_supervised/warmup_lr.py
src/vesuvius/models/training/trainers/semi_supervised/__init__.py
src/vesuvius/models/training/trainers/semi_supervised/ramps.py
src/vesuvius/models/training/trainers/semi_supervised/train_mean_teacher.py
src/vesuvius/models/training/trainers/semi_supervised/train_uncertainty_aware_mean_teacher.py
src/vesuvius/models/training/trainers/semi_supervised/two_stream_batch_sampler.py
src/vesuvius/models/utilities/__init__.py
src/vesuvius/models/utilities/cli_utils.py
src/vesuvius/models/utilities/compute_gradient_norm.py
src/vesuvius/models/utilities/data_format_utils.py
src/vesuvius/models/utilities/get_accelerator.py
src/vesuvius/models/utilities/load_checkpoint.py
src/vesuvius/models/utilities/s3_utils.py
src/vesuvius/models/utilities/utils.py
src/vesuvius/napari_trainer/__init__.py
src/vesuvius/napari_trainer/inference.py
src/vesuvius/napari_trainer/inference_widget.py
src/vesuvius/napari_trainer/main_window.py
src/vesuvius/neural_tracing/infer.py
src/vesuvius/neural_tracing/trace_service.py
src/vesuvius/neural_tracing/datasets/__init__.py
src/vesuvius/neural_tracing/datasets/augmentation.py
src/vesuvius/neural_tracing/datasets/common.py
src/vesuvius/neural_tracing/datasets/conditioning.py
src/vesuvius/neural_tracing/datasets/dataset_defaults.py
src/vesuvius/neural_tracing/datasets/dataset_rowcol_cond.py
src/vesuvius/neural_tracing/datasets/detect_wrap_overlap_masks.py
src/vesuvius/neural_tracing/datasets/direction_helpers.py
src/vesuvius/neural_tracing/datasets/extrapolation.py
src/vesuvius/neural_tracing/datasets/patch_finding.py
src/vesuvius/neural_tracing/datasets/perturbation.py
src/vesuvius/neural_tracing/datasets/triplet_resampling.py
src/vesuvius/neural_tracing/datasets/interpolation/radial_fn.py
src/vesuvius/neural_tracing/datasets/interpolation/torch_rbf.py
src/vesuvius/neural_tracing/evaluation/evaluate.py
src/vesuvius/neural_tracing/evaluation/evaluate_vc3d.py
src/vesuvius/neural_tracing/heatmap_single_point/augmentation.py
src/vesuvius/neural_tracing/heatmap_single_point/cropping.py
src/vesuvius/neural_tracing/heatmap_single_point/dataset.py
src/vesuvius/neural_tracing/heatmap_single_point/tifxyz.py
src/vesuvius/neural_tracing/heatmap_single_point/trace.py
src/vesuvius/neural_tracing/heatmap_single_point/train.py
src/vesuvius/neural_tracing/heatmap_single_point/visualization.py
src/vesuvius/neural_tracing/inference/__init__.py
src/vesuvius/neural_tracing/inference/common.py
src/vesuvius/neural_tracing/inference/displacement_helpers.py
src/vesuvius/neural_tracing/inference/displacement_tta.py
src/vesuvius/neural_tracing/inference/extrap_lookup.py
src/vesuvius/neural_tracing/inference/generate_segment_cover_bboxes.py
src/vesuvius/neural_tracing/inference/infer_global_extrap.py
src/vesuvius/neural_tracing/inference/infer_rowcol_triplet_wraps.py
src/vesuvius/neural_tracing/loss/__init__.py
src/vesuvius/neural_tracing/loss/deep_supervision.py
src/vesuvius/neural_tracing/loss/displacement_losses.py
src/vesuvius/neural_tracing/loss/surf_overlap_loss.py
src/vesuvius/neural_tracing/loss/uv_equidist_loss.py
src/vesuvius/neural_tracing/nets/__init__.py
src/vesuvius/neural_tracing/nets/models.py
src/vesuvius/neural_tracing/nets/vesuvius_unet3d.py
src/vesuvius/neural_tracing/nets/youssef_mae.py
src/vesuvius/neural_tracing/trainers/__init__.py
src/vesuvius/neural_tracing/trainers/rowcol_cond_visualization.py
src/vesuvius/neural_tracing/trainers/train_rowcol_cond.py
src/vesuvius/scripts/crop_tif_stack.py
src/vesuvius/scripts/detect_holes.py
src/vesuvius/scripts/extract_sparse_ome_zarr.py
src/vesuvius/scripts/fill_inner_outer_labels.py
src/vesuvius/scripts/fix_holes.py
src/vesuvius/scripts/hole_viewer.py
src/vesuvius/scripts/labels_to_zarr.py
src/vesuvius/scripts/tifs_to_ome_zarr.py
src/vesuvius/structure_tensor/__init__.py
src/vesuvius/structure_tensor/_compile_utils.py
src/vesuvius/structure_tensor/compute_vf.py
src/vesuvius/structure_tensor/create_st.py
src/vesuvius/structure_tensor/create_vf.py
src/vesuvius/structure_tensor/run_create_st.py
src/vesuvius/structure_tensor/smooth_vf.py
src/vesuvius/structure_tensor/vf_format.py
src/vesuvius/structure_tensor/vf_io.py
src/vesuvius/tifxyz/__init__.py
src/vesuvius/tifxyz/hierarchical_tiling.py
src/vesuvius/tifxyz/reader.py
src/vesuvius/tifxyz/types.py
src/vesuvius/tifxyz/upsampling.py
src/vesuvius/tifxyz/writer.py
src/vesuvius/utils/__init__.py
src/vesuvius/utils/build_betti.py
src/vesuvius/utils/catalog.py
src/vesuvius/utils/k8s.py
src/vesuvius/utils/plotting.py
src/vesuvius/utils/type_conversion.py
src/vesuvius/utils/utils.py
src/vesuvius/utils/io/__init__.py
src/vesuvius/utils/io/patch_cache_utils.py
src/vesuvius/utils/io/zarr_utils.py
src/vesuvius/utils/models/__init__.py
src/vesuvius/utils/models/ddp_allgather.py
src/vesuvius/utils/models/helpers.py
src/vesuvius/utils/models/load_nnunet_model.py
tests/test_imports.py