imaginaire_public

• evaluate module
• imaginaire package
  • Subpackages
    • imaginaire.datasets package
      • Submodules
      • imaginaire.datasets.base module
      • imaginaire.datasets.cache module
      • imaginaire.datasets.dummy module
      • imaginaire.datasets.folder module
      • imaginaire.datasets.images module
      • imaginaire.datasets.lmdb module
      • imaginaire.datasets.object_store module
      • imaginaire.datasets.paired_few_shot_videos module
      • imaginaire.datasets.paired_few_shot_videos_native module
      • imaginaire.datasets.paired_images module
      • imaginaire.datasets.paired_videos module
      • imaginaire.datasets.unpaired_few_shot_images module
      • imaginaire.datasets.unpaired_images module
      • Module contents
    • imaginaire.discriminators package
      • Submodules
      • imaginaire.discriminators.dummy module
      • imaginaire.discriminators.fpse module
      • imaginaire.discriminators.fs_vid2vid module
      • imaginaire.discriminators.funit module
      • imaginaire.discriminators.gancraft module
      • imaginaire.discriminators.mlp_multiclass module
      • imaginaire.discriminators.multires_patch module
      • imaginaire.discriminators.munit module
      • imaginaire.discriminators.residual module
      • imaginaire.discriminators.spade module
      • imaginaire.discriminators.unit module
      • Module contents
    • imaginaire.evaluation package
      • Subpackages
      • Submodules
      • imaginaire.evaluation.common module
      • imaginaire.evaluation.fid module
      • imaginaire.evaluation.kid module
      • imaginaire.evaluation.knn module
      • imaginaire.evaluation.lpips module
      • imaginaire.evaluation.msid module
      • imaginaire.evaluation.prdc module
      • imaginaire.evaluation.pretrained module
      • Module contents
    • imaginaire.generators package
      • Submodules
      • imaginaire.generators.coco_funit module
      • imaginaire.generators.dummy module
      • imaginaire.generators.fs_vid2vid module
      • imaginaire.generators.funit module
      • imaginaire.generators.gancraft module
      • imaginaire.generators.gancraft_base module
      • imaginaire.generators.munit module
      • imaginaire.generators.pix2pixHD module
      • imaginaire.generators.spade module
      • imaginaire.generators.unit module
      • imaginaire.generators.vid2vid module
      • imaginaire.generators.wc_vid2vid module
      • Module contents
    • imaginaire.layers package
      • Submodules
      • imaginaire.layers.activation_norm module
      • imaginaire.layers.conv module
      • imaginaire.layers.misc module
      • imaginaire.layers.non_local module
      • imaginaire.layers.nonlinearity module
      • imaginaire.layers.residual module
      • imaginaire.layers.residual_deep module
      • imaginaire.layers.vit module
      • imaginaire.layers.weight_norm module
      • Module contents
    • imaginaire.losses package
      • Submodules
      • imaginaire.losses.dict module
      • imaginaire.losses.feature_matching module
      • imaginaire.losses.flow module
      • imaginaire.losses.gan module
      • imaginaire.losses.info_nce module
      • imaginaire.losses.kl module
      • imaginaire.losses.perceptual module
      • imaginaire.losses.weighted_mse module
      • Module contents
    • imaginaire.model_utils package
      • Subpackages
      • Submodules
      • imaginaire.model_utils.fs_vid2vid module
      • imaginaire.model_utils.label module
      • imaginaire.model_utils.pix2pixHD module
      • imaginaire.model_utils.rename_inputs module
      • Module contents
    • imaginaire.optimizers package
      • Submodules
      • imaginaire.optimizers.fromage module
      • imaginaire.optimizers.madam module
      • Module contents
    • imaginaire.third_party package
      • Subpackages
      • Module contents
    • imaginaire.trainers package
      • Submodules
      • imaginaire.trainers.base module
      • imaginaire.trainers.fs_vid2vid module
      • imaginaire.trainers.funit module
      • imaginaire.trainers.gancraft module
      • imaginaire.trainers.munit module
      • imaginaire.trainers.pix2pixHD module
      • imaginaire.trainers.spade module
      • imaginaire.trainers.unit module
      • imaginaire.trainers.vid2vid module
      • imaginaire.trainers.wc_vid2vid module
      • Module contents
    • imaginaire.utils package
      • Subpackages
      • Submodules
      • imaginaire.utils.cudnn module
      • imaginaire.utils.data module
      • imaginaire.utils.dataset module
      • imaginaire.utils.diff_aug module
      • imaginaire.utils.distributed module
      • imaginaire.utils.gpu_affinity module
      • imaginaire.utils.init_weight module
      • imaginaire.utils.io module
      • imaginaire.utils.lmdb module
      • imaginaire.utils.logging module
      • imaginaire.utils.meters module
      • imaginaire.utils.misc module
      • imaginaire.utils.model_average module
      • imaginaire.utils.path module
      • imaginaire.utils.trainer module
      • Module contents
  • Submodules
  • imaginaire.config module
  • Module contents
• inference module
• train module