All modules for which code is available
- evaluate
- imaginaire.config
- imaginaire.datasets.base
- imaginaire.datasets.cache
- imaginaire.datasets.dummy
- imaginaire.datasets.folder
- imaginaire.datasets.images
- imaginaire.datasets.lmdb
- imaginaire.datasets.object_store
- imaginaire.datasets.paired_few_shot_videos
- imaginaire.datasets.paired_few_shot_videos_native
- imaginaire.datasets.paired_images
- imaginaire.datasets.paired_videos
- imaginaire.datasets.unpaired_few_shot_images
- imaginaire.datasets.unpaired_images
- imaginaire.discriminators.dummy
- imaginaire.discriminators.fpse
- imaginaire.discriminators.fs_vid2vid
- imaginaire.discriminators.funit
- imaginaire.discriminators.gancraft
- imaginaire.discriminators.mlp_multiclass
- imaginaire.discriminators.multires_patch
- imaginaire.discriminators.munit
- imaginaire.discriminators.residual
- imaginaire.discriminators.spade
- imaginaire.discriminators.unit
- imaginaire.evaluation.caption.clip
- imaginaire.evaluation.caption.common
- imaginaire.evaluation.caption.r_precision
- imaginaire.evaluation.common
- imaginaire.evaluation.fid
- imaginaire.evaluation.kid
- imaginaire.evaluation.lpips
- imaginaire.evaluation.msid
- imaginaire.evaluation.prdc
- imaginaire.evaluation.pretrained
- imaginaire.evaluation.segmentation.celebamask_hq
- imaginaire.evaluation.segmentation.cocostuff
- imaginaire.evaluation.segmentation.common
- imaginaire.generators.coco_funit
- imaginaire.generators.dummy
- imaginaire.generators.fs_vid2vid
- imaginaire.generators.funit
- imaginaire.generators.gancraft
- imaginaire.generators.gancraft_base
- imaginaire.generators.munit
- imaginaire.generators.pix2pixHD
- imaginaire.generators.spade
- imaginaire.generators.unit
- imaginaire.generators.vid2vid
- imaginaire.generators.wc_vid2vid
- imaginaire.layers.activation_norm
- imaginaire.layers.conv
- imaginaire.layers.misc
- imaginaire.layers.non_local
- imaginaire.layers.nonlinearity
- imaginaire.layers.residual
- imaginaire.layers.residual_deep
- imaginaire.layers.vit
- imaginaire.layers.weight_norm
- imaginaire.losses.dict
- imaginaire.losses.feature_matching
- imaginaire.losses.flow
- imaginaire.losses.gan
- imaginaire.losses.info_nce
- imaginaire.losses.kl
- imaginaire.losses.perceptual
- imaginaire.losses.weighted_mse
- imaginaire.model_utils.fs_vid2vid
- imaginaire.model_utils.gancraft.camctl
- imaginaire.model_utils.gancraft.layers
- imaginaire.model_utils.gancraft.loss
- imaginaire.model_utils.gancraft.mc_lbl_reduction
- imaginaire.model_utils.gancraft.mc_utils
- imaginaire.model_utils.gancraft.voxlib.positional_encoding
- imaginaire.model_utils.gancraft.voxlib.sp_trilinear
- imaginaire.model_utils.gancraft.voxlib.voxlib
- imaginaire.model_utils.label
- imaginaire.model_utils.pix2pixHD
- imaginaire.model_utils.rename_inputs
- imaginaire.model_utils.wc_vid2vid.render
- imaginaire.optimizers.fromage
- imaginaire.optimizers.madam
- imaginaire.third_party.bias_act.bias_act
- imaginaire.third_party.flow_net.flow_net
- imaginaire.third_party.upfirdn2d.upfirdn2d
- imaginaire.trainers.base
- imaginaire.trainers.fs_vid2vid
- imaginaire.trainers.funit
- imaginaire.trainers.gancraft
- imaginaire.trainers.munit
- imaginaire.trainers.pix2pixHD
- imaginaire.trainers.spade
- imaginaire.trainers.unit
- imaginaire.trainers.vid2vid
- imaginaire.trainers.wc_vid2vid
- imaginaire.utils.cudnn
- imaginaire.utils.data
- imaginaire.utils.dataset
- imaginaire.utils.diff_aug
- imaginaire.utils.distributed
- imaginaire.utils.gpu_affinity
- imaginaire.utils.init_weight
- imaginaire.utils.io
- imaginaire.utils.lmdb
- imaginaire.utils.logging
- imaginaire.utils.meters
- imaginaire.utils.misc
- imaginaire.utils.model_average
- imaginaire.utils.path
- imaginaire.utils.trainer
- imaginaire.utils.visualization.common
- imaginaire.utils.visualization.face
- imaginaire.utils.visualization.pose
- inference
- train