LICENSE
MANIFEST.in
README.md
pyproject.toml
oma/__init__.py
oma/trainer.py
oma/data/datamodule/base.py
oma/data/datamodule/datamodules.py
oma/data/datasets/base.py
oma/data/datasets/numpy_dataset.py
oma/data/recipes/__init__.py
oma/data/recipes/base.py
oma/data/recipes/brats.py
oma/data/recipes/ixi.py
oma/data/recipes/utils.py
oma/evaluation/manager.py
oma/evaluation/evaluators/__init__.py
oma/evaluation/evaluators/base.py
oma/evaluation/evaluators/image.py
oma/evaluation/metrics/__init__.py
oma/evaluation/metrics/basic.py
oma/evaluation/metrics/reconstruction.py
oma/losses/__init__.py
oma/losses/base.py
oma/losses/composer.py
oma/losses/recipes/__init__.py
oma/losses/recipes/autoencoder.py
oma/losses/terms/__init__.py
oma/losses/terms/adversarial.py
oma/losses/terms/perceptual.py
oma/losses/terms/pixel.py
oma/losses/terms/regularization.py
oma/methods/__init__.py
oma/methods/autoencoder.py
oma/methods/base.py
oma/methods/diffusion_bridge_translation.py
oma/methods/translation.py
oma/models/autoencoder/ae_kl.py
oma/models/diffusion/bridge.py
oma/models/ldm/util.py
oma/models/ldm/data/__init__.py
oma/models/ldm/data/util.py
oma/models/ldm/models/autoencoder.py
oma/models/ldm/models/diffusion/__init__.py
oma/models/ldm/models/diffusion/ddim.py
oma/models/ldm/models/diffusion/ddpm.py
oma/models/ldm/models/diffusion/plms.py
oma/models/ldm/models/diffusion/sampling_util.py
oma/models/ldm/models/diffusion/dpm_solver/__init__.py
oma/models/ldm/models/diffusion/dpm_solver/dpm_solver.py
oma/models/ldm/models/diffusion/dpm_solver/sampler.py
oma/models/ldm/modules/attention.py
oma/models/ldm/modules/ema.py
oma/models/ldm/modules/diffusionmodules/__init__.py
oma/models/ldm/modules/diffusionmodules/model.py
oma/models/ldm/modules/diffusionmodules/openaimodel.py
oma/models/ldm/modules/diffusionmodules/upscaling.py
oma/models/ldm/modules/diffusionmodules/util.py
oma/models/ldm/modules/distributions/__init__.py
oma/models/ldm/modules/distributions/distributions.py
oma/models/ldm/modules/encoders/__init__.py
oma/models/ldm/modules/encoders/modules.py
oma/models/ldm/modules/image_degradation/__init__.py
oma/models/ldm/modules/image_degradation/bsrgan.py
oma/models/ldm/modules/image_degradation/bsrgan_light.py
oma/models/ldm/modules/image_degradation/utils_image.py
oma/models/ldm/modules/midas/__init__.py
oma/models/ldm/modules/midas/api.py
oma/models/ldm/modules/midas/utils.py
oma/models/ldm/modules/midas/midas/__init__.py
oma/models/ldm/modules/midas/midas/base_model.py
oma/models/ldm/modules/midas/midas/blocks.py
oma/models/ldm/modules/midas/midas/dpt_depth.py
oma/models/ldm/modules/midas/midas/midas_net.py
oma/models/ldm/modules/midas/midas/midas_net_custom.py
oma/models/ldm/modules/midas/midas/transforms.py
oma/models/ldm/modules/midas/midas/vit.py
oma/models/networks/__init__.py
oma/models/networks/backbones/dense_layer.py
oma/models/networks/backbones/discriminator.py
oma/models/networks/backbones/fp16_util.py
oma/models/networks/backbones/layers.py
oma/models/networks/backbones/layerspp.py
oma/models/networks/backbones/ncsnpp.py
oma/models/networks/backbones/up_or_down_sampling.py
oma/models/networks/backbones/op/__init__.py
oma/models/networks/backbones/op/fused_act.py
oma/models/networks/backbones/op/fused_bias_act.cpp
oma/models/networks/backbones/op/fused_bias_act_kernel.cu
oma/models/networks/backbones/op/upfirdn2d.cpp
oma/models/networks/backbones/op/upfirdn2d.py
oma/models/networks/backbones/op/upfirdn2d_kernel.cu
oma/models/taming/lr_scheduler.py
oma/models/taming/util.py
oma/models/taming/data/ade20k.py
oma/models/taming/data/annotated_objects_coco.py
oma/models/taming/data/annotated_objects_dataset.py
oma/models/taming/data/annotated_objects_open_images.py
oma/models/taming/data/base.py
oma/models/taming/data/coco.py
oma/models/taming/data/custom.py
oma/models/taming/data/faceshq.py
oma/models/taming/data/helper_types.py
oma/models/taming/data/image_transforms.py
oma/models/taming/data/imagenet.py
oma/models/taming/data/open_images_helper.py
oma/models/taming/data/sflckr.py
oma/models/taming/data/utils.py
oma/models/taming/data/conditional_builder/objects_bbox.py
oma/models/taming/data/conditional_builder/objects_center_points.py
oma/models/taming/data/conditional_builder/utils.py
oma/models/taming/models/cond_transformer.py
oma/models/taming/models/dummy_cond_stage.py
oma/models/taming/models/vqgan.py
oma/models/taming/modules/util.py
oma/models/taming/modules/diffusionmodules/model.py
oma/models/taming/modules/discriminator/model.py
oma/models/taming/modules/losses/__init__.py
oma/models/taming/modules/losses/lpips.py
oma/models/taming/modules/losses/segmentation.py
oma/models/taming/modules/losses/vqperceptual.py
oma/models/taming/modules/misc/coord.py
oma/models/taming/modules/transformer/mingpt.py
oma/models/taming/modules/transformer/permuter.py
oma/models/taming/modules/vqvae/quantize.py
oma/visualization/translation_train_vis.py
openmedaxis.egg-info/PKG-INFO
openmedaxis.egg-info/SOURCES.txt
openmedaxis.egg-info/dependency_links.txt
openmedaxis.egg-info/requires.txt
openmedaxis.egg-info/top_level.txt
tests/test_autoencoder_loss_contract.py
tests/test_grouped_loss_method.py
tests/test_grouped_loss_method_v2.py
tests/test_loss_numerics_and_gradients.py
tests/test_lsplit_datamodule_numpy.py