Contents:
imaginaire_public
evaluate module
imaginaire package
Subpackages
imaginaire.datasets package
imaginaire.discriminators package
imaginaire.evaluation package
imaginaire.generators package
imaginaire.layers package
imaginaire.losses package
imaginaire.model_utils package
imaginaire.optimizers package
imaginaire.third_party package
imaginaire.trainers package
imaginaire.utils package
Submodules
imaginaire.config module
Module contents
inference module
train module
Imaginaire
»
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
Y
A
AdaptiveNorm (class in imaginaire.layers.activation_norm)
add_face_keypoints() (in module imaginaire.utils.visualization.face)
AffineMod (class in imaginaire.model_utils.gancraft.layers)
apply() (imaginaire.layers.weight_norm.ScaledLR static method)
apply_diff_aug() (in module imaginaire.utils.diff_aug)
apply_imagenet_normalization() (in module imaginaire.utils.misc)
apply_ops() (imaginaire.datasets.base.BaseDataset method)
ApplyNoise (class in imaginaire.layers.misc)
attention_encode() (imaginaire.generators.fs_vid2vid.AttentionModule method)
AttentionModule (class in imaginaire.generators.fs_vid2vid)
AttrDict (class in imaginaire.config)
Augmentor (class in imaginaire.utils.data)
AutoEncoder (class in imaginaire.generators.munit)
(class in imaginaire.generators.unit)
available_models() (in module imaginaire.evaluation.caption.clip)
B
backward() (imaginaire.losses.info_nce.GatherLayer static method)
(imaginaire.model_utils.gancraft.voxlib.positional_encoding.PositionalEncodingFunction static method)
(imaginaire.model_utils.gancraft.voxlib.sp_trilinear.SparseTrilinearWorldCoordFunction static method)
Base3DGenerator (class in imaginaire.generators.gancraft_base)
base_openpose_to_npy() (in module imaginaire.utils.visualization.pose)
BaseDataset (class in imaginaire.datasets.base)
BaseNetwork (class in imaginaire.generators.vid2vid)
BaseTrainer (class in imaginaire.trainers.base)
bias (imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
Blur (class in imaginaire.third_party.upfirdn2d)
(class in imaginaire.third_party.upfirdn2d.upfirdn2d)
BlurDownsample (class in imaginaire.third_party.upfirdn2d)
(class in imaginaire.third_party.upfirdn2d.upfirdn2d)
BlurUpsample (class in imaginaire.third_party.upfirdn2d)
(class in imaginaire.third_party.upfirdn2d.upfirdn2d)
build_lmdb() (in module imaginaire.utils.lmdb)
build_model() (in module imaginaire.evaluation.caption.clip)
C
Cache (class in imaginaire.datasets.cache)
calc_height_map() (in module imaginaire.model_utils.gancraft.mc_utils)
calibrate_batch_norm_momentum() (in module imaginaire.utils.model_average)
check_and_add() (in module imaginaire.utils.lmdb)
clean_resize() (in module imaginaire.evaluation.common)
CleanInceptionV3 (class in imaginaire.evaluation.common)
cluster_features() (in module imaginaire.model_utils.pix2pixHD)
coco2reduced() (imaginaire.model_utils.gancraft.mc_utils.MCLabelTranslator method)
COCOFUNITTranslator (class in imaginaire.generators.coco_funit)
Colorize (class in imaginaire.utils.visualization.common)
colormap() (in module imaginaire.model_utils.gancraft.mc_utils)
combine_fg_mask() (in module imaginaire.model_utils.fs_vid2vid)
compute_all_metrics() (in module imaginaire.evaluation)
(in module imaginaire.evaluation.common)
compute_all_metrics_data() (in module imaginaire.evaluation)
(in module imaginaire.evaluation.common)
compute_bias() (imaginaire.layers.weight_norm.ScaledLR method)
compute_fid() (in module imaginaire.evaluation)
(in module imaginaire.evaluation.fid)
compute_fid_data() (in module imaginaire.evaluation)
(in module imaginaire.evaluation.fid)
compute_flow_and_conf() (imaginaire.third_party.flow_net.flow_net.FlowNet method)
compute_flow_loss() (imaginaire.losses.flow.FlowLoss method)
(imaginaire.losses.FlowLoss method)
compute_flow_losses() (imaginaire.losses.flow.FlowLoss method)
(imaginaire.losses.FlowLoss method)
compute_gan_losses() (imaginaire.trainers.vid2vid.Trainer method)
compute_hist() (in module imaginaire.evaluation.segmentation.common)
compute_kid() (in module imaginaire.evaluation)
(in module imaginaire.evaluation.kid)
compute_kid_data() (in module imaginaire.evaluation)
(in module imaginaire.evaluation.kid)
compute_mask_loss() (imaginaire.losses.flow.FlowLoss method)
(imaginaire.losses.FlowLoss method)
compute_mask_losses() (imaginaire.losses.flow.FlowLoss method)
(imaginaire.losses.FlowLoss method)
compute_nn() (in module imaginaire.evaluation.common)
compute_pairwise_distance() (in module imaginaire.evaluation.common)
compute_prdc() (in module imaginaire.evaluation)
(in module imaginaire.evaluation.prdc)
compute_weight() (imaginaire.layers.weight_norm.ScaledLR method)
concat_few_shot_labels() (in module imaginaire.model_utils.label)
concat_frames() (in module imaginaire.model_utils.fs_vid2vid)
concat_labels() (in module imaginaire.model_utils.label)
Config (class in imaginaire.config)
connect_face_keypoints() (in module imaginaire.utils.visualization.face)
connect_pose_keypoints() (in module imaginaire.utils.visualization.pose)
ConstantInput (class in imaginaire.layers.misc)
construct_file_path() (in module imaginaire.utils.lmdb)
construct_graph() (in module imaginaire.evaluation.msid)
construct_graph_kgraph() (in module imaginaire.evaluation.msid)
construct_graph_sparse() (in module imaginaire.evaluation.msid)
ContentEncoder (class in imaginaire.generators.funit)
(class in imaginaire.generators.unit)
Conv1dBlock (class in imaginaire.layers)
(class in imaginaire.layers.conv)
Conv2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.conv)
Conv3dBlock (class in imaginaire.layers)
(class in imaginaire.layers.conv)
conv_blocks() (imaginaire.layers.residual.ModulatedRes2dBlock method)
convert_face_landmarks_to_image() (in module imaginaire.utils.visualization.face)
copy_s2t() (imaginaire.utils.model_average.ModelAverage method)
copy_t2s() (imaginaire.utils.model_average.ModelAverage method)
create_metadata() (in module imaginaire.utils.lmdb)
create_sequence_output_dir() (imaginaire.trainers.vid2vid.Trainer method)
(imaginaire.trainers.wc_vid2vid.Trainer method)
crop_and_resize() (in module imaginaire.model_utils.fs_vid2vid)
crop_face_from_data() (in module imaginaire.model_utils.fs_vid2vid)
crop_face_from_output() (in module imaginaire.model_utils.fs_vid2vid)
crop_hand_from_output() (in module imaginaire.model_utils.fs_vid2vid)
crop_person_from_data() (in module imaginaire.model_utils.fs_vid2vid)
cumsum_exclusive() (in module imaginaire.model_utils.gancraft.mc_utils)
custom_init() (imaginaire.generators.fs_vid2vid.Generator method)
(imaginaire.generators.gancraft.Generator method)
D
Dataset (class in imaginaire.datasets.dummy)
(class in imaginaire.datasets.images)
(class in imaginaire.datasets.paired_few_shot_videos)
(class in imaginaire.datasets.paired_few_shot_videos_native)
(class in imaginaire.datasets.paired_images)
(class in imaginaire.datasets.paired_videos)
(class in imaginaire.datasets.unpaired_few_shot_images)
(class in imaginaire.datasets.unpaired_images)
decode() (imaginaire.generators.coco_funit.COCOFUNITTranslator method)
(imaginaire.generators.funit.FUNITTranslator method)
(imaginaire.generators.munit.AutoEncoder method)
decode_unprojections() (in module imaginaire.model_utils.wc_vid2vid.render)
Decoder (class in imaginaire.generators.funit)
(class in imaginaire.generators.munit)
(class in imaginaire.generators.unit)
DeepLabV2 (class in imaginaire.evaluation.segmentation.cocostuff)
DeepRes2dBlock (class in imaginaire.layers.residual_deep)
define_edge_lists() (in module imaginaire.utils.visualization.pose)
detach() (in module imaginaire.model_utils.fs_vid2vid)
device (class in imaginaire.utils.gpu_affinity)
deviceGetCount() (in module imaginaire.utils.gpu_affinity)
DictLoss (class in imaginaire.losses)
(class in imaginaire.losses.dict)
diff_aug() (in module imaginaire.utils.diff_aug)
dilation (imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
dis_forward() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.funit.Trainer method)
(imaginaire.trainers.gancraft.Trainer method)
(imaginaire.trainers.munit.Trainer method)
(imaginaire.trainers.pix2pixHD.Trainer method)
(imaginaire.trainers.spade.Trainer method)
(imaginaire.trainers.unit.Trainer method)
dis_update() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.vid2vid.Trainer method)
discriminate_video() (imaginaire.discriminators.fs_vid2vid.Discriminator method)
Discriminator (class in imaginaire.discriminators.dummy)
(class in imaginaire.discriminators.fs_vid2vid)
(class in imaginaire.discriminators.funit)
(class in imaginaire.discriminators.gancraft)
(class in imaginaire.discriminators.mlp_multiclass)
(class in imaginaire.discriminators.multires_patch)
(class in imaginaire.discriminators.munit)
(class in imaginaire.discriminators.spade)
(class in imaginaire.discriminators.unit)
discrminate_image() (imaginaire.discriminators.fs_vid2vid.Discriminator method)
dist_all_gather_tensor() (in module imaginaire.utils.distributed)
dist_all_reduce_tensor() (in module imaginaire.utils.distributed)
dist_reduce_tensor() (in module imaginaire.utils.distributed)
download_file() (in module imaginaire.utils.io)
download_file_from_google_drive() (in module imaginaire.utils.io)
DownRes2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.residual)
downsample2d() (in module imaginaire.third_party.upfirdn2d.upfirdn2d)
draw_edge() (in module imaginaire.utils.visualization.face)
draw_edges() (in module imaginaire.utils.visualization.pose)
draw_openpose_npy() (in module imaginaire.utils.visualization.pose)
DualAdaptiveNorm (class in imaginaire.layers.activation_norm)
E
Embedding2d (class in imaginaire.layers.conv)
Embedding2dBlock (class in imaginaire.layers.conv)
embedding_dim (imaginaire.layers.conv.Embedding2d attribute)
EmbeddingBlock (class in imaginaire.layers.conv)
encode() (imaginaire.generators.coco_funit.COCOFUNITTranslator method)
(imaginaire.generators.funit.FUNITTranslator method)
(imaginaire.generators.munit.AutoEncoder method)
encode_features() (in module imaginaire.model_utils.pix2pixHD)
encode_reference() (imaginaire.generators.fs_vid2vid.WeightGenerator method)
Encoder (class in imaginaire.generators.pix2pixHD)
end_of_epoch() (imaginaire.trainers.base.BaseTrainer method)
end_of_iteration() (imaginaire.trainers.base.BaseTrainer method)
eval() (imaginaire.trainers.gancraft.GauGANLoader method)
EvalCameraController (class in imaginaire.model_utils.gancraft.camctl)
evaluate (module)
extra_repr() (imaginaire.layers.activation_norm.ScaleNorm method)
(imaginaire.layers.conv.ModulatedConv2d method)
(imaginaire.third_party.upfirdn2d.Blur method)
(imaginaire.third_party.upfirdn2d.BlurDownsample method)
(imaginaire.third_party.upfirdn2d.BlurUpsample method)
(imaginaire.third_party.upfirdn2d.upfirdn2d.Blur method)
(imaginaire.third_party.upfirdn2d.upfirdn2d.BlurDownsample method)
(imaginaire.third_party.upfirdn2d.upfirdn2d.BlurUpsample method)
extract_valid_keypoints() (in module imaginaire.utils.visualization.pose)
extract_valid_pose_labels() (in module imaginaire.model_utils.fs_vid2vid)
F
FeatureMatchingLoss (class in imaginaire.losses)
(class in imaginaire.losses.feature_matching)
FIDInceptionA (class in imaginaire.evaluation.pretrained)
FIDInceptionC (class in imaginaire.evaluation.pretrained)
FIDInceptionE_1 (class in imaginaire.evaluation.pretrained)
FIDInceptionE_2 (class in imaginaire.evaluation.pretrained)
filter2d() (in module imaginaire.third_party.upfirdn2d.upfirdn2d)
filtfilt() (imaginaire.model_utils.gancraft.camctl.EvalCameraController method)
finetune() (imaginaire.trainers.fs_vid2vid.Trainer method)
flip_hdr() (imaginaire.datasets.base.BaseDataset method)
flow_generation() (imaginaire.generators.fs_vid2vid.Generator method)
FlowGenerator (class in imaginaire.generators.fs_vid2vid)
(class in imaginaire.generators.vid2vid)
FlowLoss (class in imaginaire.losses)
(class in imaginaire.losses.flow)
FlowNet (class in imaginaire.third_party.flow_net.flow_net)
flush() (imaginaire.utils.meters.Meter method)
FolderDataset (class in imaginaire.datasets.folder)
forward() (imaginaire.discriminators.dummy.Discriminator method)
(imaginaire.discriminators.fpse.FPSEDiscriminator method)
(imaginaire.discriminators.fs_vid2vid.Discriminator method)
(imaginaire.discriminators.fs_vid2vid.MultiPatchDiscriminator method)
(imaginaire.discriminators.funit.Discriminator method)
(imaginaire.discriminators.funit.ResDiscriminator method)
(imaginaire.discriminators.gancraft.Discriminator method)
(imaginaire.discriminators.gancraft.FPSEDiscriminator method)
(imaginaire.discriminators.mlp_multiclass.Discriminator method)
(imaginaire.discriminators.multires_patch.Discriminator method)
(imaginaire.discriminators.multires_patch.MultiResPatchDiscriminator method)
(imaginaire.discriminators.multires_patch.NLayerPatchDiscriminator method)
(imaginaire.discriminators.multires_patch.WeightSharedMultiResPatchDiscriminator method)
(imaginaire.discriminators.munit.Discriminator method)
(imaginaire.discriminators.residual.ResDiscriminator method)
(imaginaire.discriminators.spade.Discriminator method)
(imaginaire.discriminators.unit.Discriminator method)
(imaginaire.evaluation.caption.common.ImageEncoder method)
(imaginaire.evaluation.common.CleanInceptionV3 method)
(imaginaire.evaluation.lpips.LPIPSNet method)
(imaginaire.evaluation.lpips.LPNet method)
(imaginaire.evaluation.lpips.NetLinLayer method)
(imaginaire.evaluation.lpips.ScalingLayer method)
(imaginaire.evaluation.lpips.vgg16 method)
(imaginaire.evaluation.pretrained.FIDInceptionA method)
(imaginaire.evaluation.pretrained.FIDInceptionC method)
(imaginaire.evaluation.pretrained.FIDInceptionE_1 method)
(imaginaire.evaluation.pretrained.FIDInceptionE_2 method)
(imaginaire.evaluation.pretrained.InceptionV3 method)
(imaginaire.evaluation.pretrained.SwAV method)
(imaginaire.evaluation.pretrained.TFInceptionV3 method)
(imaginaire.evaluation.pretrained.Vgg16 method)
(imaginaire.evaluation.segmentation.celebamask_hq.Unet method)
(imaginaire.evaluation.segmentation.celebamask_hq.unetConv2 method)
(imaginaire.evaluation.segmentation.celebamask_hq.unetUp method)
(imaginaire.evaluation.segmentation.cocostuff.DeepLabV2 method)
(imaginaire.evaluation.segmentation.common.SegmentationHistModel method)
(imaginaire.generators.coco_funit.COCOFUNITTranslator method)
(imaginaire.generators.coco_funit.Generator method)
(imaginaire.generators.dummy.Generator method)
(imaginaire.generators.fs_vid2vid.AttentionModule method)
(imaginaire.generators.fs_vid2vid.FlowGenerator method)
(imaginaire.generators.fs_vid2vid.Generator method)
(imaginaire.generators.fs_vid2vid.LabelEmbedder method)
(imaginaire.generators.fs_vid2vid.WeightGenerator method)
(imaginaire.generators.funit.ContentEncoder method)
(imaginaire.generators.funit.Decoder method)
(imaginaire.generators.funit.FUNITTranslator method)
(imaginaire.generators.funit.Generator method)
(imaginaire.generators.funit.MLP method)
(imaginaire.generators.funit.StyleEncoder method)
(imaginaire.generators.gancraft.Generator method)
(imaginaire.generators.gancraft_base.RenderCNN method)
(imaginaire.generators.gancraft_base.RenderMLP method)
(imaginaire.generators.gancraft_base.SKYMLP method)
(imaginaire.generators.gancraft_base.StyleEncoder method)
(imaginaire.generators.gancraft_base.StyleMLP method)
(imaginaire.generators.munit.AutoEncoder method)
(imaginaire.generators.munit.Decoder method)
(imaginaire.generators.munit.Generator method)
(imaginaire.generators.munit.MLP method)
(imaginaire.generators.munit.StyleEncoder method)
(imaginaire.generators.pix2pixHD.Encoder method)
(imaginaire.generators.pix2pixHD.Generator method)
(imaginaire.generators.pix2pixHD.GlobalGenerator method)
(imaginaire.generators.pix2pixHD.LocalEnhancer method)
(imaginaire.generators.spade.Generator method)
(imaginaire.generators.spade.SPADEGenerator method)
(imaginaire.generators.spade.StyleEncoder method)
(imaginaire.generators.unit.AutoEncoder method)
(imaginaire.generators.unit.ContentEncoder method)
(imaginaire.generators.unit.Decoder method)
(imaginaire.generators.unit.Generator method)
(imaginaire.generators.vid2vid.FlowGenerator method)
(imaginaire.generators.vid2vid.Generator method)
(imaginaire.generators.wc_vid2vid.Generator method)
(imaginaire.layers.activation_norm.AdaptiveNorm method)
(imaginaire.layers.activation_norm.DualAdaptiveNorm method)
(imaginaire.layers.activation_norm.HyperSpatiallyAdaptiveNorm method)
(imaginaire.layers.activation_norm.LayerNorm2d method)
(imaginaire.layers.activation_norm.PixelLayerNorm method)
(imaginaire.layers.activation_norm.ScaleNorm method)
(imaginaire.layers.activation_norm.SpatiallyAdaptiveNorm method)
(imaginaire.layers.activation_norm.SplitMeanStd method)
(imaginaire.layers.conv.Embedding2d method)
(imaginaire.layers.conv.HyperConv2d method)
(imaginaire.layers.conv.ModulatedConv2d method)
(imaginaire.layers.conv.ModulatedConv2dBlock method)
(imaginaire.layers.conv.PartialConv2d method)
(imaginaire.layers.conv.PartialConv3d method)
(imaginaire.layers.misc.ApplyNoise method)
(imaginaire.layers.misc.ConstantInput method)
(imaginaire.layers.misc.PartialSequential method)
(imaginaire.layers.non_local.NonLocal2dBlock method)
(imaginaire.layers.nonlinearity.ScaledLeakyReLU method)
(imaginaire.layers.NonLocal2dBlock method)
(imaginaire.layers.vit.ViT2dBlock method)
(imaginaire.layers.weight_norm.WeightDemodulation method)
(imaginaire.losses.dict.DictLoss method)
(imaginaire.losses.DictLoss method)
(imaginaire.losses.feature_matching.FeatureMatchingLoss method)
(imaginaire.losses.FeatureMatchingLoss method)
(imaginaire.losses.flow.FlowLoss method)
(imaginaire.losses.flow.MaskedL1Loss method)
(imaginaire.losses.FlowLoss method)
(imaginaire.losses.gan.GANLoss method)
(imaginaire.losses.GANLoss method)
(imaginaire.losses.GaussianKLLoss method)
(imaginaire.losses.info_nce.GatherLayer static method)
(imaginaire.losses.info_nce.InfoNCELoss method)
(imaginaire.losses.kl.GaussianKLLoss method)
(imaginaire.losses.MaskedL1Loss method)
(imaginaire.losses.perceptual.PerceptualLoss method)
(imaginaire.losses.PerceptualLoss method)
(imaginaire.losses.weighted_mse.WeightedMSELoss method)
(imaginaire.losses.WeightedMSELoss method)
(imaginaire.model_utils.gancraft.layers.AffineMod method)
(imaginaire.model_utils.gancraft.layers.ModLinear method)
(imaginaire.model_utils.gancraft.loss.GANLoss method)
(imaginaire.model_utils.gancraft.voxlib.positional_encoding.PositionalEncodingFunction static method)
(imaginaire.model_utils.gancraft.voxlib.sp_trilinear.SparseTrilinearWorldCoordFunction static method)
(imaginaire.third_party.bias_act.bias_act.FusedNonlinearity method)
(imaginaire.third_party.bias_act.FusedNonlinearity method)
(imaginaire.third_party.flow_net.flow_net.FlowNet method)
(imaginaire.third_party.upfirdn2d.Blur method)
(imaginaire.third_party.upfirdn2d.BlurDownsample method)
(imaginaire.third_party.upfirdn2d.BlurUpsample method)
(imaginaire.third_party.upfirdn2d.upfirdn2d.Blur method)
(imaginaire.third_party.upfirdn2d.upfirdn2d.BlurDownsample method)
(imaginaire.third_party.upfirdn2d.upfirdn2d.BlurUpsample method)
(imaginaire.utils.model_average.ModelAverage method)
(imaginaire.utils.trainer.WrappedModel method)
FPSEDiscriminator (class in imaginaire.discriminators.fpse)
(class in imaginaire.discriminators.gancraft)
Fromage (class in imaginaire.optimizers)
(class in imaginaire.optimizers.fromage)
func() (in module imaginaire.utils.visualization.face)
FUNITTranslator (class in imaginaire.generators.funit)
FusedNonlinearity (class in imaginaire.third_party.bias_act)
(class in imaginaire.third_party.bias_act.bias_act)
G
GANLoss (class in imaginaire.losses)
(class in imaginaire.losses.gan)
(class in imaginaire.model_utils.gancraft.loss)
gather_tensors() (in module imaginaire.evaluation.common)
GatherLayer (class in imaginaire.losses.info_nce)
GauGANLoader (class in imaginaire.trainers.gancraft)
GaussianKLLoss (class in imaginaire.losses)
(class in imaginaire.losses.kl)
gen_corner_voxel() (in module imaginaire.model_utils.gancraft.mc_utils)
gen_forward() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.funit.Trainer method)
(imaginaire.trainers.gancraft.Trainer method)
(imaginaire.trainers.munit.Trainer method)
(imaginaire.trainers.pix2pixHD.Trainer method)
(imaginaire.trainers.spade.Trainer method)
(imaginaire.trainers.unit.Trainer method)
gen_frames() (imaginaire.trainers.vid2vid.Trainer method)
(imaginaire.trainers.wc_vid2vid.Trainer method)
gen_update() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.vid2vid.Trainer method)
Generator (class in imaginaire.generators.coco_funit)
(class in imaginaire.generators.dummy)
(class in imaginaire.generators.fs_vid2vid)
(class in imaginaire.generators.funit)
(class in imaginaire.generators.gancraft)
(class in imaginaire.generators.munit)
(class in imaginaire.generators.pix2pixHD)
(class in imaginaire.generators.spade)
(class in imaginaire.generators.unit)
(class in imaginaire.generators.vid2vid)
(class in imaginaire.generators.wc_vid2vid)
get_126_landmarks_from_image() (in module imaginaire.utils.visualization.face)
get_activation_norm_layer() (in module imaginaire.layers.activation_norm)
get_activations() (in module imaginaire.evaluation.common)
get_all_filenames_from_list() (in module imaginaire.utils.lmdb)
get_all_skipped_frames() (in module imaginaire.discriminators.fs_vid2vid)
get_and_setattr() (in module imaginaire.utils.misc)
get_checkpoint() (in module imaginaire.utils.io)
get_class_number() (in module imaginaire.utils.data)
get_cond_dims() (imaginaire.generators.vid2vid.Generator method)
(imaginaire.generators.wc_vid2vid.Generator method)
get_cond_maps() (imaginaire.generators.vid2vid.Generator method)
(imaginaire.generators.wc_vid2vid.Generator method)
get_confirm_token() (in module imaginaire.utils.io)
get_conv_weights() (imaginaire.generators.fs_vid2vid.WeightGenerator method)
get_crop_h_w() (in module imaginaire.utils.data)
get_data_t() (imaginaire.trainers.fs_vid2vid.Trainer method)
(imaginaire.trainers.vid2vid.Trainer method)
(imaginaire.trainers.wc_vid2vid.Trainer method)
get_date_uid() (in module imaginaire.utils.logging)
get_dis_losses() (imaginaire.trainers.vid2vid.Trainer method)
get_dlib_landmarks_from_image() (in module imaginaire.utils.visualization.face)
get_edges() (in module imaginaire.model_utils.pix2pixHD)
get_face_bbox_for_data() (in module imaginaire.model_utils.fs_vid2vid)
get_face_bbox_for_output() (in module imaginaire.model_utils.fs_vid2vid)
get_face_mask() (in module imaginaire.model_utils.fs_vid2vid)
get_fg_mask() (in module imaginaire.model_utils.fs_vid2vid)
get_gen_losses() (imaginaire.trainers.vid2vid.Trainer method)
get_grid() (in module imaginaire.model_utils.fs_vid2vid)
get_guidance_images_and_masks() (imaginaire.generators.wc_vid2vid.Generator method)
get_hand_bbox_for_output() (in module imaginaire.model_utils.fs_vid2vid)
get_image_encoder() (in module imaginaire.evaluation.caption)
(in module imaginaire.evaluation.caption.common)
get_image_size() (in module imaginaire.utils.data)
get_immediate_subdirectories() (in module imaginaire.utils.path)
get_kth_value() (in module imaginaire.evaluation.prdc)
get_label_lengths() (imaginaire.datasets.paired_few_shot_videos_native.Dataset method)
(imaginaire.datasets.paired_videos.Dataset method)
get_lmdb_data_types() (in module imaginaire.utils.lmdb)
get_lpips_model() (in module imaginaire.evaluation.lpips)
get_miou() (in module imaginaire.evaluation.segmentation)
(in module imaginaire.evaluation.segmentation.common)
get_model_optimizer_and_scheduler() (in module imaginaire.utils.trainer)
get_neighbor_height() (in module imaginaire.model_utils.gancraft.camctl)
get_nested_attr() (in module imaginaire.utils.misc)
get_nonlinearity_layer() (in module imaginaire.layers.nonlinearity)
get_norm_weights() (imaginaire.generators.fs_vid2vid.WeightGenerator method)
get_num_filters() (imaginaire.generators.vid2vid.BaseNetwork method)
get_num_reduced_lbls() (imaginaire.model_utils.gancraft.mc_utils.MCLabelTranslator method)
get_op() (imaginaire.datasets.base.BaseDataset method)
get_optimizer() (in module imaginaire.utils.trainer)
get_optimizer_for_params() (in module imaginaire.utils.trainer)
get_optimizer_with_params() (in module imaginaire.model_utils.pix2pixHD)
get_outputs() (in module imaginaire.evaluation.common)
get_paired_input_image_channel_number() (in module imaginaire.utils.data)
get_paired_input_label_channel_number() (in module imaginaire.utils.data)
get_param_groups() (imaginaire.generators.gancraft_base.Base3DGenerator method)
get_part_mask() (in module imaginaire.model_utils.fs_vid2vid)
get_partial() (imaginaire.generators.wc_vid2vid.Generator method)
get_person_bbox_for_data() (in module imaginaire.model_utils.fs_vid2vid)
get_pseudo_gt() (imaginaire.generators.gancraft.Generator method)
get_r_precision() (in module imaginaire.evaluation.caption)
(in module imaginaire.evaluation.caption.r_precision)
get_rank() (in module imaginaire.utils.distributed)
get_recursive_subdirectories() (in module imaginaire.utils.path)
get_scheduler() (in module imaginaire.utils.trainer)
get_segmentation_hist_model() (in module imaginaire.evaluation.segmentation)
(in module imaginaire.evaluation.segmentation.common)
get_skipped_frames() (in module imaginaire.discriminators.fs_vid2vid)
get_target_tensor() (imaginaire.losses.gan.GANLoss method)
(imaginaire.losses.GANLoss method)
get_test_dataloader() (in module imaginaire.utils.dataset)
get_test_output_images() (imaginaire.trainers.fs_vid2vid.Trainer method)
(imaginaire.trainers.vid2vid.Trainer method)
(imaginaire.trainers.wc_vid2vid.Trainer method)
get_train_and_val_dataloader() (in module imaginaire.utils.dataset)
get_train_params() (in module imaginaire.model_utils.pix2pixHD)
get_trainer() (in module imaginaire.utils.trainer)
get_video_activations() (in module imaginaire.evaluation.common)
get_weight_norm_layer() (in module imaginaire.layers.weight_norm)
get_weight_stats() (in module imaginaire.utils.meters)
get_world_size() (in module imaginaire.utils.distributed)
getCpuAffinity() (imaginaire.utils.gpu_affinity.device method)
getitem_by_path() (imaginaire.datasets.folder.FolderDataset method)
(imaginaire.datasets.lmdb.LMDBDataset method)
(imaginaire.datasets.object_store.ObjectStoreDataset method)
getName() (imaginaire.utils.gpu_affinity.device method)
gglbl2ggid() (imaginaire.model_utils.gancraft.mc_lbl_reduction.ReducedLabelMapper method)
(imaginaire.model_utils.gancraft.mc_utils.MCLabelTranslator method)
GlobalGenerator (class in imaginaire.generators.pix2pixHD)
gradient_norm() (in module imaginaire.utils.misc)
groups (imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
H
HyperConv2d (class in imaginaire.layers.conv)
HyperConv2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.conv)
HyperRes2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.residual)
HyperSpatiallyAdaptiveNorm (class in imaginaire.layers.activation_norm)
I
ImageEncoder (class in imaginaire.evaluation.caption.common)
imaginaire (module)
imaginaire.config (module)
imaginaire.datasets (module)
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)
imaginaire.discriminators (module)
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)
imaginaire.evaluation (module)
imaginaire.evaluation.caption (module)
imaginaire.evaluation.caption.clip (module)
imaginaire.evaluation.caption.common (module)
imaginaire.evaluation.caption.r_precision (module)
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)
imaginaire.evaluation.segmentation (module)
imaginaire.evaluation.segmentation.celebamask_hq (module)
imaginaire.evaluation.segmentation.cocostuff (module)
imaginaire.evaluation.segmentation.common (module)
imaginaire.generators (module)
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)
imaginaire.layers (module)
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)
imaginaire.losses (module)
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)
imaginaire.model_utils (module)
imaginaire.model_utils.fs_vid2vid (module)
imaginaire.model_utils.gancraft (module)
imaginaire.model_utils.gancraft.camctl (module)
imaginaire.model_utils.gancraft.layers (module)
imaginaire.model_utils.gancraft.loss (module)
imaginaire.model_utils.gancraft.mc_lbl_reduction (module)
imaginaire.model_utils.gancraft.mc_utils (module)
imaginaire.model_utils.gancraft.voxlib (module)
imaginaire.model_utils.gancraft.voxlib.positional_encoding (module)
imaginaire.model_utils.gancraft.voxlib.sp_trilinear (module)
imaginaire.model_utils.gancraft.voxlib.voxlib (module)
imaginaire.model_utils.label (module)
imaginaire.model_utils.pix2pixHD (module)
imaginaire.model_utils.rename_inputs (module)
imaginaire.model_utils.wc_vid2vid (module)
imaginaire.model_utils.wc_vid2vid.render (module)
imaginaire.optimizers (module)
imaginaire.optimizers.fromage (module)
imaginaire.optimizers.madam (module)
imaginaire.third_party (module)
imaginaire.third_party.bias_act (module)
imaginaire.third_party.bias_act.bias_act (module)
imaginaire.third_party.flow_net (module)
imaginaire.third_party.flow_net.flow_net (module)
imaginaire.third_party.flow_net.test_flownet2 (module)
imaginaire.third_party.upfirdn2d (module)
imaginaire.third_party.upfirdn2d.upfirdn2d (module)
imaginaire.trainers (module)
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)
imaginaire.utils (module)
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)
imaginaire.utils.visualization (module)
imaginaire.utils.visualization.common (module)
imaginaire.utils.visualization.face (module)
imaginaire.utils.visualization.pose (module)
inception_forward() (in module imaginaire.evaluation.common)
inception_init() (in module imaginaire.evaluation.common)
InceptionV3 (class in imaginaire.evaluation.pretrained)
inference (module)
inference() (imaginaire.generators.coco_funit.Generator method)
(imaginaire.generators.funit.Generator method)
(imaginaire.generators.gancraft.Generator method)
(imaginaire.generators.munit.Generator method)
(imaginaire.generators.pix2pixHD.Generator method)
(imaginaire.generators.spade.Generator method)
(imaginaire.generators.unit.Generator method)
InfoNCELoss (class in imaginaire.losses.info_nce)
init_cudnn() (in module imaginaire.utils.cudnn)
init_dist() (in module imaginaire.utils.distributed)
init_logging() (in module imaginaire.utils.logging)
init_network_weights() (imaginaire.generators.fs_vid2vid.Generator method)
init_temporal_network() (imaginaire.generators.fs_vid2vid.Generator method)
(imaginaire.generators.vid2vid.Generator method)
(imaginaire.trainers.vid2vid.Trainer method)
interp_points() (in module imaginaire.utils.visualization.face)
is_local_master() (in module imaginaire.utils.distributed)
is_master() (in module imaginaire.utils.distributed)
is_sea() (imaginaire.model_utils.gancraft.mc_utils.McVoxel method)
K
kernel_size (imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
L
labelcolormap() (in module imaginaire.utils.visualization.common)
LabelEmbedder (class in imaginaire.generators.fs_vid2vid)
LayerNorm2d (class in imaginaire.layers.activation_norm)
linear() (in module imaginaire.utils.visualization.face)
LinearBlock (class in imaginaire.layers)
(class in imaginaire.layers.conv)
LMDBDataset (class in imaginaire.datasets.lmdb)
load() (in module imaginaire.evaluation.caption.clip)
load_checkpoint() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.gancraft.Trainer method)
(imaginaire.trainers.wc_vid2vid.Trainer method)
load_from_folder() (in module imaginaire.utils.data)
load_from_lmdb() (in module imaginaire.utils.data)
load_from_object_store() (in module imaginaire.utils.data)
load_or_compute_activations() (in module imaginaire.evaluation.common)
load_pretrained_network() (imaginaire.generators.fs_vid2vid.Generator method)
(imaginaire.generators.pix2pixHD.Generator method)
load_voxel_new() (in module imaginaire.model_utils.gancraft.mc_utils)
LocalEnhancer (class in imaginaire.generators.pix2pixHD)
loss() (imaginaire.losses.gan.GANLoss method)
(imaginaire.losses.GANLoss method)
(imaginaire.model_utils.gancraft.loss.GANLoss method)
LPIPSNet (class in imaginaire.evaluation.lpips)
LPNet (class in imaginaire.evaluation.lpips)
M
Madam (class in imaginaire.optimizers)
(class in imaginaire.optimizers.madam)
main() (in module evaluate)
(in module inference)
(in module train)
make_logging_dir() (in module imaginaire.utils.logging)
make_one_hot() (in module imaginaire.model_utils.label)
MaskedL1Loss (class in imaginaire.losses)
(class in imaginaire.losses.flow)
master_only() (in module imaginaire.utils.distributed)
master_only_print() (in module imaginaire.utils.distributed)
max_norm (imaginaire.layers.conv.Embedding2d attribute)
mc2coco() (imaginaire.model_utils.gancraft.mc_utils.MCLabelTranslator method)
mc2reduced() (imaginaire.model_utils.gancraft.mc_utils.MCLabelTranslator method)
mc_color() (imaginaire.model_utils.gancraft.mc_utils.MCLabelTranslator method)
MCLabelTranslator (class in imaginaire.model_utils.gancraft.mc_utils)
McVoxel (class in imaginaire.model_utils.gancraft.mc_utils)
Meter (class in imaginaire.utils.meters)
MLP (class in imaginaire.generators.funit)
(class in imaginaire.generators.munit)
ModelAverage (class in imaginaire.utils.model_average)
ModLinear (class in imaginaire.model_utils.gancraft.layers)
modulate() (imaginaire.generators.gancraft_base.RenderCNN method)
ModulatedConv2d (class in imaginaire.layers.conv)
ModulatedConv2dBlock (class in imaginaire.layers.conv)
ModulatedRes2dBlock (class in imaginaire.layers.residual)
move_dont_care() (in module imaginaire.model_utils.label)
msid_descriptor() (in module imaginaire.evaluation.msid)
MultiOutConv2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.conv)
MultiOutRes2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.residual)
MultiPatchDiscriminator (class in imaginaire.discriminators.fs_vid2vid)
MultiResPatchDiscriminator (class in imaginaire.discriminators.multires_patch)
N
NetLinLayer (class in imaginaire.evaluation.lpips)
NLayerPatchDiscriminator (class in imaginaire.discriminators.multires_patch)
NonLocal2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.non_local)
norm() (imaginaire.third_party.flow_net.flow_net.FlowNet method)
norm_type (imaginaire.layers.conv.Embedding2d attribute)
normalize_and_connect_face_keypoints() (in module imaginaire.utils.visualization.face)
normalize_face_keypoints() (in module imaginaire.utils.visualization.face)
normalize_faces() (in module imaginaire.model_utils.fs_vid2vid)
normalize_tensor() (in module imaginaire.evaluation.lpips)
np_euc_cdist() (in module imaginaire.evaluation.msid)
npy_to_tensor() (in module imaginaire.utils.visualization.face)
num_embeddings (imaginaire.layers.conv.Embedding2d attribute)
num_inference_sequences() (imaginaire.datasets.paired_few_shot_videos_native.Dataset method)
(imaginaire.datasets.paired_images.Dataset method)
(imaginaire.datasets.paired_videos.Dataset method)
num_points() (imaginaire.model_utils.wc_vid2vid.render.SplatRenderer method)
O
ObjectStoreDataset (class in imaginaire.datasets.object_store)
one_up_conv_layer() (imaginaire.generators.fs_vid2vid.Generator method)
(imaginaire.generators.vid2vid.Generator method)
openpose_to_npy() (in module imaginaire.utils.visualization.pose)
openpose_to_npy_largest_only() (in module imaginaire.utils.visualization.pose)
out_channels (imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
output_padding (imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
P
padding (imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
padding_idx (imaginaire.layers.conv.Embedding2d attribute)
padding_mode (imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
parse_args() (in module evaluate)
(in module inference)
(in module train)
PartialConv2d (class in imaginaire.layers.conv)
PartialConv2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.conv)
PartialConv3d (class in imaginaire.layers.conv)
PartialConv3dBlock (class in imaginaire.layers)
(class in imaginaire.layers.conv)
PartialRes2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.residual)
PartialRes3dBlock (class in imaginaire.layers)
(class in imaginaire.layers.residual)
PartialSequential (class in imaginaire.layers.misc)
PerceptualLoss (class in imaginaire.losses)
(class in imaginaire.losses.perceptual)
perform_augmentation() (imaginaire.datasets.base.BaseDataset method)
(imaginaire.utils.data.Augmentor method)
perform_individual_video_frame() (imaginaire.datasets.base.BaseDataset method)
pick_image() (in module imaginaire.model_utils.fs_vid2vid)
PixelLayerNorm (class in imaginaire.layers.activation_norm)
PixelNorm (class in imaginaire.layers.activation_norm)
plot_keypoints() (in module imaginaire.utils.visualization.common)
plot_keypoints_on_black() (in module imaginaire.utils.visualization.common)
positional_encoding() (in module imaginaire.model_utils.gancraft.voxlib.positional_encoding)
(in module imaginaire.model_utils.gancraft.voxlib.voxlib)
positional_encoding_backward() (in module imaginaire.model_utils.gancraft.voxlib.voxlib)
positional_encoding_pt() (in module imaginaire.model_utils.gancraft.voxlib.positional_encoding)
PositionalEncodingFunction (class in imaginaire.model_utils.gancraft.voxlib.positional_encoding)
post_process() (imaginaire.trainers.fs_vid2vid.Trainer method)
(imaginaire.trainers.vid2vid.Trainer method)
pre_process() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.fs_vid2vid.Trainer method)
(imaginaire.trainers.pix2pixHD.Trainer method)
(imaginaire.trainers.vid2vid.Trainer method)
pre_process_densepose() (in module imaginaire.model_utils.fs_vid2vid)
R
rand_brightness() (in module imaginaire.utils.diff_aug)
rand_camera_pose_birdseye() (in module imaginaire.model_utils.gancraft.camctl)
rand_camera_pose_firstperson() (in module imaginaire.model_utils.gancraft.camctl)
rand_camera_pose_insideout() (in module imaginaire.model_utils.gancraft.camctl)
rand_camera_pose_thridperson() (in module imaginaire.model_utils.gancraft.camctl)
rand_camera_pose_thridperson2() (in module imaginaire.model_utils.gancraft.camctl)
rand_camera_pose_thridperson3() (in module imaginaire.model_utils.gancraft.camctl)
rand_camera_pose_tour() (in module imaginaire.model_utils.gancraft.camctl)
rand_contrast() (in module imaginaire.utils.diff_aug)
rand_crop() (in module imaginaire.model_utils.gancraft.mc_utils)
rand_cutout() (in module imaginaire.utils.diff_aug)
rand_saturation() (in module imaginaire.utils.diff_aug)
rand_translation() (in module imaginaire.utils.diff_aug)
rand_translation_scale() (in module imaginaire.utils.diff_aug)
random_roll() (in module imaginaire.model_utils.fs_vid2vid)
random_shift() (in module imaginaire.utils.misc)
ray_voxel_intersection_perspective() (in module imaginaire.model_utils.gancraft.voxlib.voxlib)
read() (imaginaire.datasets.cache.Cache method)
recalculate_batch_norm_statistics() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.spade.Trainer method)
recursive_update() (in module imaginaire.config)
ReducedLabelMapper (class in imaginaire.model_utils.gancraft.mc_lbl_reduction)
remove() (imaginaire.layers.weight_norm.ScaledLR method)
remove_equalized_lr() (in module imaginaire.layers.weight_norm)
remove_other_ppl() (in module imaginaire.model_utils.fs_vid2vid)
remove_weight_norms() (in module imaginaire.layers.weight_norm)
rename_inputs() (in module imaginaire.model_utils.rename_inputs)
render_image() (imaginaire.model_utils.wc_vid2vid.render.SplatRenderer method)
RenderCNN (class in imaginaire.generators.gancraft_base)
renderer_update_point_cloud() (imaginaire.generators.wc_vid2vid.Generator method)
RenderMLP (class in imaginaire.generators.gancraft_base)
requires_grad() (in module imaginaire.utils.misc)
Res1dBlock (class in imaginaire.layers)
(class in imaginaire.layers.residual)
Res2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.residual)
Res3dBlock (class in imaginaire.layers)
(class in imaginaire.layers.residual)
resample() (in module imaginaire.model_utils.fs_vid2vid)
ResDiscriminator (class in imaginaire.discriminators.funit)
(class in imaginaire.discriminators.residual)
reset() (imaginaire.generators.fs_vid2vid.Generator method)
(imaginaire.generators.fs_vid2vid.WeightGenerator method)
(imaginaire.model_utils.wc_vid2vid.render.SplatRenderer method)
(imaginaire.trainers.vid2vid.Trainer method)
(imaginaire.trainers.wc_vid2vid.Trainer method)
(imaginaire.utils.meters.Meter method)
reset_batch_norm() (in module imaginaire.utils.model_average)
reset_renderer() (imaginaire.generators.wc_vid2vid.Generator method)
reshape_embed_input() (imaginaire.generators.fs_vid2vid.WeightReshaper method)
reshape_weight() (imaginaire.generators.fs_vid2vid.WeightReshaper method)
ResLinearBlock (class in imaginaire.layers)
(class in imaginaire.layers.residual)
rgetattr() (in module imaginaire.config)
roll() (in module imaginaire.model_utils.fs_vid2vid)
rsetattr() (in module imaginaire.config)
S
sample_camera() (imaginaire.generators.gancraft.Generator method)
sample_depth_batched() (in module imaginaire.model_utils.gancraft.mc_utils)
save_checkpoint() (imaginaire.trainers.base.BaseTrainer method)
save_image() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.fs_vid2vid.Trainer method)
(imaginaire.trainers.vid2vid.Trainer method)
(imaginaire.trainers.wc_vid2vid.Trainer method)
save_intermediate_training_results() (in module imaginaire.utils.io)
save_pilimage_in_jpeg() (in module imaginaire.utils.io)
save_response_content() (in module imaginaire.utils.io)
save_tensor_image() (in module imaginaire.utils.visualization)
(in module imaginaire.utils.visualization.common)
scale_grad_by_freq (imaginaire.layers.conv.Embedding2d attribute)
scaled_lr() (in module imaginaire.layers.weight_norm)
ScaledLeakyReLU (class in imaginaire.layers.nonlinearity)
ScaledLR (class in imaginaire.layers.weight_norm)
ScaleNorm (class in imaginaire.layers.activation_norm)
ScalingLayer (class in imaginaire.evaluation.lpips)
segmask_smooth() (in module imaginaire.model_utils.gancraft.mc_utils)
SegmentationHistModel (class in imaginaire.evaluation.segmentation.common)
select_object() (in module imaginaire.model_utils.fs_vid2vid)
set_affinity() (in module imaginaire.utils.gpu_affinity)
set_color() (in module imaginaire.utils.visualization.face)
set_inference_sequence_idx() (imaginaire.datasets.paired_few_shot_videos.Dataset method)
(imaginaire.datasets.paired_images.Dataset method)
(imaginaire.datasets.paired_videos.Dataset method)
set_random_seed() (in module imaginaire.utils.trainer)
set_sample_class_idx() (imaginaire.datasets.images.Dataset method)
(imaginaire.datasets.unpaired_few_shot_images.Dataset method)
set_sequence_idx() (in module imaginaire.evaluation.common)
set_sequence_length() (imaginaire.datasets.paired_few_shot_videos.Dataset method)
(imaginaire.datasets.paired_images.Dataset method)
(imaginaire.datasets.paired_videos.Dataset method)
set_summary_writer() (in module imaginaire.utils.meters)
setup_filter() (in module imaginaire.third_party.upfirdn2d.upfirdn2d)
SKYMLP (class in imaginaire.generators.gancraft_base)
slice_tensor() (in module imaginaire.utils.misc)
slq_red_var() (in module imaginaire.evaluation.msid)
smooth_face_keypoints() (in module imaginaire.utils.visualization.face)
smooth_interp() (imaginaire.discriminators.gancraft.FPSEDiscriminator static method)
sn_compute_weight() (imaginaire.utils.model_average.ModelAverage method)
sn_reshape_weight_to_matrix() (imaginaire.utils.model_average.ModelAverage method)
(in module imaginaire.utils.meters)
sp_trilinear_worldcoord() (in module imaginaire.model_utils.gancraft.voxlib.voxlib)
sp_trilinear_worldcoord_backward() (in module imaginaire.model_utils.gancraft.voxlib.voxlib)
SPADE_combine() (imaginaire.generators.fs_vid2vid.Generator method)
SPADEGenerator (class in imaginaire.generators.spade)
sparse (imaginaire.layers.conv.Embedding2d attribute)
sparse_trilinear_interp_worldcoord() (in module imaginaire.model_utils.gancraft.voxlib.sp_trilinear)
SparseTrilinearWorldCoordFunction (class in imaginaire.model_utils.gancraft.voxlib.sp_trilinear)
SpatiallyAdaptiveNorm (class in imaginaire.layers.activation_norm)
SplatRenderer (class in imaginaire.model_utils.wc_vid2vid.render)
split_labels() (in module imaginaire.utils.misc)
split_weights() (imaginaire.generators.fs_vid2vid.WeightReshaper method)
SplitMeanStd (class in imaginaire.layers.activation_norm)
start_of_epoch() (imaginaire.trainers.base.BaseTrainer method)
start_of_iteration() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.wc_vid2vid.Trainer method)
step() (imaginaire.optimizers.Fromage method)
(imaginaire.optimizers.fromage.Fromage method)
(imaginaire.optimizers.Madam method)
(imaginaire.optimizers.madam.Madam method)
stride (imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
StyleEncoder (class in imaginaire.generators.funit)
(class in imaginaire.generators.gancraft_base)
(class in imaginaire.generators.munit)
(class in imaginaire.generators.spade)
StyleMLP (class in imaginaire.generators.gancraft_base)
sum() (imaginaire.generators.fs_vid2vid.WeightReshaper method)
sum_mul() (imaginaire.generators.fs_vid2vid.WeightReshaper method)
SwAV (class in imaginaire.evaluation.pretrained)
systemGetDriverVersion() (in module imaginaire.utils.gpu_affinity)
T
tensor2flow() (in module imaginaire.utils.visualization)
(in module imaginaire.utils.visualization.common)
tensor2im() (in module imaginaire.utils.visualization)
(in module imaginaire.utils.visualization.common)
tensor2label() (in module imaginaire.utils.visualization)
(in module imaginaire.utils.visualization.common)
tensor2pilimage() (in module imaginaire.utils.visualization)
(in module imaginaire.utils.visualization.common)
tensor2pose() (in module imaginaire.utils.visualization.pose)
test() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.fs_vid2vid.Trainer method)
(imaginaire.trainers.gancraft.Trainer method)
(imaginaire.trainers.vid2vid.Trainer method)
(imaginaire.trainers.wc_vid2vid.Trainer method)
test_single() (imaginaire.trainers.vid2vid.Trainer method)
(imaginaire.trainers.wc_vid2vid.Trainer method)
TFInceptionV3 (class in imaginaire.evaluation.pretrained)
to_channels_last() (in module imaginaire.utils.misc)
to_cpu() (in module imaginaire.utils.misc)
to_cuda() (in module imaginaire.utils.misc)
to_device() (in module imaginaire.utils.misc)
to_float() (in module imaginaire.utils.misc)
to_half() (in module imaginaire.utils.misc)
to_tensor() (imaginaire.datasets.base.BaseDataset method)
topk_anneal() (imaginaire.losses.gan.GANLoss method)
(imaginaire.losses.GANLoss method)
TourCameraController (class in imaginaire.model_utils.gancraft.camctl)
train (module)
Trainer (class in imaginaire.trainers.fs_vid2vid)
(class in imaginaire.trainers.funit)
(class in imaginaire.trainers.gancraft)
(class in imaginaire.trainers.munit)
(class in imaginaire.trainers.pix2pixHD)
(class in imaginaire.trainers.spade)
(class in imaginaire.trainers.unit)
(class in imaginaire.trainers.vid2vid)
(class in imaginaire.trainers.wc_vid2vid)
training (imaginaire.discriminators.dummy.Discriminator attribute)
(imaginaire.discriminators.fpse.FPSEDiscriminator attribute)
(imaginaire.discriminators.fs_vid2vid.Discriminator attribute)
(imaginaire.discriminators.fs_vid2vid.MultiPatchDiscriminator attribute)
(imaginaire.discriminators.funit.Discriminator attribute)
(imaginaire.discriminators.funit.ResDiscriminator attribute)
(imaginaire.discriminators.gancraft.Discriminator attribute)
(imaginaire.discriminators.gancraft.FPSEDiscriminator attribute)
(imaginaire.discriminators.mlp_multiclass.Discriminator attribute)
(imaginaire.discriminators.multires_patch.Discriminator attribute)
(imaginaire.discriminators.multires_patch.MultiResPatchDiscriminator attribute)
(imaginaire.discriminators.multires_patch.NLayerPatchDiscriminator attribute)
(imaginaire.discriminators.multires_patch.WeightSharedMultiResPatchDiscriminator attribute)
(imaginaire.discriminators.munit.Discriminator attribute)
(imaginaire.discriminators.residual.ResDiscriminator attribute)
(imaginaire.discriminators.spade.Discriminator attribute)
(imaginaire.discriminators.unit.Discriminator attribute)
(imaginaire.evaluation.caption.common.ImageEncoder attribute)
(imaginaire.evaluation.common.CleanInceptionV3 attribute)
(imaginaire.evaluation.lpips.LPIPSNet attribute)
(imaginaire.evaluation.lpips.LPNet attribute)
(imaginaire.evaluation.lpips.NetLinLayer attribute)
(imaginaire.evaluation.lpips.ScalingLayer attribute)
(imaginaire.evaluation.lpips.vgg16 attribute)
(imaginaire.evaluation.pretrained.FIDInceptionA attribute)
(imaginaire.evaluation.pretrained.FIDInceptionC attribute)
(imaginaire.evaluation.pretrained.FIDInceptionE_1 attribute)
(imaginaire.evaluation.pretrained.FIDInceptionE_2 attribute)
(imaginaire.evaluation.pretrained.InceptionV3 attribute)
(imaginaire.evaluation.pretrained.SwAV attribute)
(imaginaire.evaluation.pretrained.TFInceptionV3 attribute)
(imaginaire.evaluation.pretrained.Vgg16 attribute)
(imaginaire.evaluation.segmentation.celebamask_hq.Unet attribute)
(imaginaire.evaluation.segmentation.celebamask_hq.unetConv2 attribute)
(imaginaire.evaluation.segmentation.celebamask_hq.unetUp attribute)
(imaginaire.evaluation.segmentation.cocostuff.DeepLabV2 attribute)
(imaginaire.evaluation.segmentation.common.SegmentationHistModel attribute)
(imaginaire.generators.coco_funit.COCOFUNITTranslator attribute)
(imaginaire.generators.coco_funit.Generator attribute)
(imaginaire.generators.dummy.Generator attribute)
(imaginaire.generators.fs_vid2vid.AttentionModule attribute)
(imaginaire.generators.fs_vid2vid.FlowGenerator attribute)
(imaginaire.generators.fs_vid2vid.Generator attribute)
(imaginaire.generators.fs_vid2vid.LabelEmbedder attribute)
(imaginaire.generators.fs_vid2vid.WeightGenerator attribute)
(imaginaire.generators.funit.ContentEncoder attribute)
(imaginaire.generators.funit.Decoder attribute)
(imaginaire.generators.funit.FUNITTranslator attribute)
(imaginaire.generators.funit.Generator attribute)
(imaginaire.generators.funit.MLP attribute)
(imaginaire.generators.funit.StyleEncoder attribute)
(imaginaire.generators.gancraft.Generator attribute)
(imaginaire.generators.gancraft_base.Base3DGenerator attribute)
(imaginaire.generators.gancraft_base.RenderCNN attribute)
(imaginaire.generators.gancraft_base.RenderMLP attribute)
(imaginaire.generators.gancraft_base.SKYMLP attribute)
(imaginaire.generators.gancraft_base.StyleEncoder attribute)
(imaginaire.generators.gancraft_base.StyleMLP attribute)
(imaginaire.generators.munit.AutoEncoder attribute)
(imaginaire.generators.munit.Decoder attribute)
(imaginaire.generators.munit.Generator attribute)
(imaginaire.generators.munit.MLP attribute)
(imaginaire.generators.munit.StyleEncoder attribute)
(imaginaire.generators.pix2pixHD.Encoder attribute)
(imaginaire.generators.pix2pixHD.Generator attribute)
(imaginaire.generators.pix2pixHD.GlobalGenerator attribute)
(imaginaire.generators.pix2pixHD.LocalEnhancer attribute)
(imaginaire.generators.spade.Generator attribute)
(imaginaire.generators.spade.SPADEGenerator attribute)
(imaginaire.generators.spade.StyleEncoder attribute)
(imaginaire.generators.unit.AutoEncoder attribute)
(imaginaire.generators.unit.ContentEncoder attribute)
(imaginaire.generators.unit.Decoder attribute)
(imaginaire.generators.unit.Generator attribute)
(imaginaire.generators.vid2vid.BaseNetwork attribute)
(imaginaire.generators.vid2vid.FlowGenerator attribute)
(imaginaire.generators.vid2vid.Generator attribute)
(imaginaire.generators.wc_vid2vid.Generator attribute)
(imaginaire.layers.activation_norm.AdaptiveNorm attribute)
(imaginaire.layers.activation_norm.DualAdaptiveNorm attribute)
(imaginaire.layers.activation_norm.HyperSpatiallyAdaptiveNorm attribute)
(imaginaire.layers.activation_norm.LayerNorm2d attribute)
(imaginaire.layers.activation_norm.PixelLayerNorm attribute)
(imaginaire.layers.activation_norm.PixelNorm attribute)
(imaginaire.layers.activation_norm.ScaleNorm attribute)
(imaginaire.layers.activation_norm.SpatiallyAdaptiveNorm attribute)
(imaginaire.layers.activation_norm.SplitMeanStd attribute)
(imaginaire.layers.conv.Conv1dBlock attribute)
(imaginaire.layers.conv.Conv2dBlock attribute)
(imaginaire.layers.conv.Conv3dBlock attribute)
(imaginaire.layers.conv.Embedding2dBlock attribute)
(imaginaire.layers.conv.EmbeddingBlock attribute)
(imaginaire.layers.conv.HyperConv2d attribute)
(imaginaire.layers.conv.HyperConv2dBlock attribute)
(imaginaire.layers.conv.LinearBlock attribute)
(imaginaire.layers.conv.ModulatedConv2d attribute)
(imaginaire.layers.conv.ModulatedConv2dBlock attribute)
(imaginaire.layers.conv.MultiOutConv2dBlock attribute)
(imaginaire.layers.conv.PartialConv2dBlock attribute)
(imaginaire.layers.conv.PartialConv3dBlock attribute)
(imaginaire.layers.Conv1dBlock attribute)
(imaginaire.layers.Conv2dBlock attribute)
(imaginaire.layers.Conv3dBlock attribute)
(imaginaire.layers.DownRes2dBlock attribute)
(imaginaire.layers.HyperConv2dBlock attribute)
(imaginaire.layers.HyperRes2dBlock attribute)
(imaginaire.layers.LinearBlock attribute)
(imaginaire.layers.misc.ApplyNoise attribute)
(imaginaire.layers.misc.ConstantInput attribute)
(imaginaire.layers.misc.PartialSequential attribute)
(imaginaire.layers.MultiOutConv2dBlock attribute)
(imaginaire.layers.MultiOutRes2dBlock attribute)
(imaginaire.layers.non_local.NonLocal2dBlock attribute)
(imaginaire.layers.nonlinearity.ScaledLeakyReLU attribute)
(imaginaire.layers.NonLocal2dBlock attribute)
(imaginaire.layers.PartialConv2dBlock attribute)
(imaginaire.layers.PartialConv3dBlock attribute)
(imaginaire.layers.PartialRes2dBlock attribute)
(imaginaire.layers.PartialRes3dBlock attribute)
(imaginaire.layers.Res1dBlock attribute)
(imaginaire.layers.Res2dBlock attribute)
(imaginaire.layers.Res3dBlock attribute)
(imaginaire.layers.residual.DownRes2dBlock attribute)
(imaginaire.layers.residual.HyperRes2dBlock attribute)
(imaginaire.layers.residual.ModulatedRes2dBlock attribute)
(imaginaire.layers.residual.MultiOutRes2dBlock attribute)
(imaginaire.layers.residual.PartialRes2dBlock attribute)
(imaginaire.layers.residual.PartialRes3dBlock attribute)
(imaginaire.layers.residual.Res1dBlock attribute)
(imaginaire.layers.residual.Res2dBlock attribute)
(imaginaire.layers.residual.Res3dBlock attribute)
(imaginaire.layers.residual.ResLinearBlock attribute)
(imaginaire.layers.residual.UpRes2dBlock attribute)
(imaginaire.layers.residual_deep.DeepRes2dBlock attribute)
(imaginaire.layers.ResLinearBlock attribute)
(imaginaire.layers.UpRes2dBlock attribute)
(imaginaire.layers.vit.ViT2dBlock attribute)
(imaginaire.layers.weight_norm.WeightDemodulation attribute)
(imaginaire.losses.dict.DictLoss attribute)
(imaginaire.losses.DictLoss attribute)
(imaginaire.losses.feature_matching.FeatureMatchingLoss attribute)
(imaginaire.losses.FeatureMatchingLoss attribute)
(imaginaire.losses.flow.FlowLoss attribute)
(imaginaire.losses.flow.MaskedL1Loss attribute)
(imaginaire.losses.FlowLoss attribute)
(imaginaire.losses.gan.GANLoss attribute)
(imaginaire.losses.GANLoss attribute)
(imaginaire.losses.GaussianKLLoss attribute)
(imaginaire.losses.info_nce.InfoNCELoss attribute)
(imaginaire.losses.kl.GaussianKLLoss attribute)
(imaginaire.losses.MaskedL1Loss attribute)
(imaginaire.losses.perceptual.PerceptualLoss attribute)
(imaginaire.losses.PerceptualLoss attribute)
(imaginaire.losses.weighted_mse.WeightedMSELoss attribute)
(imaginaire.losses.WeightedMSELoss attribute)
(imaginaire.model_utils.gancraft.layers.AffineMod attribute)
(imaginaire.model_utils.gancraft.layers.ModLinear attribute)
(imaginaire.model_utils.gancraft.loss.GANLoss attribute)
(imaginaire.model_utils.gancraft.mc_utils.McVoxel attribute)
(imaginaire.third_party.bias_act.bias_act.FusedNonlinearity attribute)
(imaginaire.third_party.bias_act.FusedNonlinearity attribute)
(imaginaire.third_party.flow_net.flow_net.FlowNet attribute)
(imaginaire.third_party.upfirdn2d.Blur attribute)
(imaginaire.third_party.upfirdn2d.BlurDownsample attribute)
(imaginaire.third_party.upfirdn2d.BlurUpsample attribute)
(imaginaire.third_party.upfirdn2d.upfirdn2d.Blur attribute)
(imaginaire.third_party.upfirdn2d.upfirdn2d.BlurDownsample attribute)
(imaginaire.third_party.upfirdn2d.upfirdn2d.BlurUpsample attribute)
(imaginaire.utils.model_average.ModelAverage attribute)
(imaginaire.utils.trainer.WrappedModel attribute)
trans_vec_homo() (in module imaginaire.model_utils.gancraft.mc_utils)
transposed (imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
U
uint32_to_4uint8() (imaginaire.model_utils.gancraft.mc_utils.MCLabelTranslator static method)
Unet (class in imaginaire.evaluation.segmentation.celebamask_hq)
unetConv2 (class in imaginaire.evaluation.segmentation.celebamask_hq)
unetUp (class in imaginaire.evaluation.segmentation.celebamask_hq)
update_average() (imaginaire.utils.model_average.ModelAverage method)
update_point_cloud() (imaginaire.model_utils.wc_vid2vid.render.SplatRenderer method)
upfirdn2d() (in module imaginaire.third_party.upfirdn2d.upfirdn2d)
UpRes2dBlock (class in imaginaire.layers)
(class in imaginaire.layers.residual)
upsample2d() (in module imaginaire.third_party.upfirdn2d.upfirdn2d)
V
vgg16 (class in imaginaire.evaluation.lpips)
Vgg16 (class in imaginaire.evaluation.pretrained)
visualize_label() (imaginaire.trainers.vid2vid.Trainer method)
ViT2dBlock (class in imaginaire.layers.vit)
volum_rendering_relu() (in module imaginaire.model_utils.gancraft.mc_utils)
W
weight (imaginaire.layers.conv.Embedding2d attribute)
(imaginaire.layers.conv.PartialConv2d attribute)
(imaginaire.layers.conv.PartialConv3d attribute)
weight_demod() (in module imaginaire.layers.weight_norm)
WeightDemodulation (class in imaginaire.layers.weight_norm)
WeightedMSELoss (class in imaginaire.losses)
(class in imaginaire.losses.weighted_mse)
WeightGenerator (class in imaginaire.generators.fs_vid2vid)
WeightReshaper (class in imaginaire.generators.fs_vid2vid)
weights_init() (in module imaginaire.utils.init_weight)
weights_rescale() (in module imaginaire.utils.init_weight)
WeightSharedMultiResPatchDiscriminator (class in imaginaire.discriminators.multires_patch)
world2local() (imaginaire.model_utils.gancraft.mc_utils.McVoxel method)
wrap_model_and_optimizer() (in module imaginaire.utils.trainer)
WrappedModel (class in imaginaire.utils.trainer)
write() (imaginaire.datasets.cache.Cache method)
(imaginaire.utils.meters.Meter method)
write_entry() (in module imaginaire.utils.lmdb)
write_image() (imaginaire.utils.meters.Meter method)
write_metrics() (imaginaire.trainers.base.BaseTrainer method)
(imaginaire.trainers.funit.Trainer method)
(imaginaire.trainers.munit.Trainer method)
(imaginaire.trainers.spade.Trainer method)
(imaginaire.trainers.unit.Trainer method)
(imaginaire.trainers.vid2vid.Trainer method)
write_summary() (in module imaginaire.utils.meters)
Y
yaml() (imaginaire.config.AttrDict method)