Index A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W A adam_beta1 (transmotion.configs.OptimizerConfig attribute) adam_beta2 (transmotion.configs.OptimizerConfig attribute) AntiAliasInterpolation2d (class in transmotion.blocks) B background (transmotion.configs.LossConfig attribute) background_motion (transmotion.network_bundle.NetworksBundle attribute) background_param (transmotion.network_bundle.NetworkBundleResult attribute) background_start_epoch (transmotion.configs.TrainConfig attribute) BackgroundLoss (class in transmotion.configs) base_dim (transmotion.configs.DenseMotionConf attribute) (transmotion.configs.InpaintingConfig attribute) batch_size (transmotion.configs.DataLoadingConfig attribute) (transmotion.kp_detection.KPResult attribute) bg_params (transmotion.dense_motion.BGMotionParam attribute) BGMotionParam (class in transmotion.dense_motion) BGMotionPredictor (class in transmotion.dense_motion) build_nets() (in module transmotion.network_bundle) C chain_param() (in module transmotion.network_bundle) clear_grads() (transmotion.network_bundle.NetworksBundle method) collate_fn() (in module transmotion.datasets) comp() (transmotion.debug.MMapDebug method) conform_to_type_and_device() (in module transmotion.cord_warp) contribution_maps (transmotion.dense_motion.DenseMotionResult attribute) control_params (transmotion.cord_warp.ThinPlateSpline attribute) control_points (transmotion.cord_warp.ThinPlateSpline attribute) D data_loading (transmotion.configs.TrainConfig attribute) DataLoadingConfig (class in transmotion.configs) Decoder (class in transmotion.nn_blocks) deform_frame() (transmotion.dense_motion.WarppedCords method) deform_with_4d_deformation() (in module transmotion.cord_warp) deform_with_5d_deformation() (in module transmotion.cord_warp) deformed_source (transmotion.dense_motion.DenseMotionResult attribute) (transmotion.inpainting.InpaintingResult attribute) deformed_src_fmaps (transmotion.inpainting.InpaintingResult attribute) dense_motion (transmotion.configs.TrainConfig attribute) (transmotion.network_bundle.NetworkBundleResult attribute) (transmotion.network_bundle.NetworksBundle attribute) DenseMotionConf (class in transmotion.configs) DenseMotionNetwork (class in transmotion.dense_motion) DenseMotionResult (class in transmotion.dense_motion) detach_to_cpu() (transmotion.kp_detection.KPResult method) DownBlock2d (class in transmotion.nn_blocks) draw_points_on_tensors() (in module transmotion.viz) driving (transmotion.datasets.SourceDrivingBatch attribute) (transmotion.datasets.SourceDrivingSample attribute) driving_keypoints (transmotion.network_bundle.NetworkBundleResult attribute) dropout (transmotion.configs.TrainConfig attribute) DropoutConfig (class in transmotion.configs) dummy_conf() (in module transmotion.configs) E Encoder (class in transmotion.nn_blocks) equivariance (transmotion.configs.LossConfig attribute) EquivarianceLoss (class in transmotion.configs) eval() (transmotion.network_bundle.NetworksBundle method) F fit() (transmotion.cord_warp.ThinPlateSpline class method) foreground_kp (transmotion.kp_detection.KPResult attribute) forward() (transmotion.blocks.AntiAliasInterpolation2d method) (transmotion.blocks.ImagePyramid method) (transmotion.dense_motion.BGMotionPredictor method) (transmotion.dense_motion.DenseMotionNetwork method) (transmotion.dense_motion.LearnedUpsample method) (transmotion.dense_motion.OcclusionMasks method) (transmotion.inpainting.InpaintingNetwork method) (transmotion.kp_detection.KPDetector method) (transmotion.network_bundle.NetworksBundle method) (transmotion.nn_blocks.Decoder method) (transmotion.nn_blocks.DownBlock2d method) (transmotion.nn_blocks.Encoder method) (transmotion.nn_blocks.Hourglass method) (transmotion.nn_blocks.ResBlock2d method) (transmotion.nn_blocks.SameBlock2d method) (transmotion.nn_blocks.UpBlock2d method) (transmotion.nn_blocks.Vgg19 method) fwd_decoder() (transmotion.inpainting.InpaintingNetwork method) fwd_encoder() (transmotion.inpainting.InpaintingNetwork method) G generated_image (transmotion.network_bundle.NetworkBundleResult property) get_encode_no_grad() (transmotion.inpainting.InpaintingNetwork method) get_kp_normalized_forward_func() (in module transmotion.network_bundle) get_pretrained_vgg19() (in module transmotion.nn_blocks) grid_like_featuremap() (in module transmotion.cord_warp) H hg_to_num_mappings (transmotion.dense_motion.DenseMotionNetwork attribute) Hourglass (class in transmotion.nn_blocks) I image_size (transmotion.configs.TrainConfig attribute) image_to_tensor() (in module transmotion.data_loading) ImagePyramid (class in transmotion.blocks) import_state_dict() (in module transmotion.weights) in_dim (transmotion.configs.PerceptualLoss attribute) in_features (transmotion.configs.DenseMotionConf attribute) (transmotion.configs.InpaintingConfig attribute) infer_optical_flow() (transmotion.dense_motion.DenseMotionNetwork method) init_prob (transmotion.configs.DropoutConfig attribute) initial_lr (transmotion.configs.OptimizerConfig attribute) inpaint (transmotion.network_bundle.NetworksBundle attribute) inpainted_img (transmotion.inpainting.InpaintingResult attribute) inpainting (transmotion.configs.TrainConfig attribute) (transmotion.network_bundle.NetworkBundleResult attribute) InpaintingConfig (class in transmotion.configs) InpaintingNetwork (class in transmotion.inpainting) InpaintingResult (class in transmotion.inpainting) K key_points (transmotion.network_bundle.NetworksBundle attribute) keypoints_to_heatmap_representation() (in module transmotion.dense_motion) kp_variance (transmotion.configs.DenseMotionConf attribute) KPDetector (class in transmotion.kp_detection) KPResult (class in transmotion.kp_detection) L last_out_dim (transmotion.nn_blocks.Hourglass attribute) LearnedUpsample (class in transmotion.dense_motion) load_driving_video() (in module transmotion.data_loading) load_original_weights() (in module transmotion.network_bundle) (transmotion.network_bundle.NetworksBundle method) loss (transmotion.configs.TrainConfig attribute) loss_weight (transmotion.configs.BackgroundLoss attribute) (transmotion.configs.EquivarianceLoss attribute) (transmotion.configs.WarpLoss attribute) loss_weights (transmotion.configs.PerceptualLoss attribute) LossConfig (class in transmotion.configs) lr_decay_epoch_sched (transmotion.configs.OptimizerConfig attribute) lr_decay_gamma (transmotion.configs.OptimizerConfig attribute) M make_coordinate_grid() (in module transmotion.cord_warp) map_numpy() (in module transmotion.data_loading) max_features (transmotion.configs.DenseMotionConf attribute) (transmotion.configs.InpaintingConfig attribute) max_prob (transmotion.configs.DropoutConfig attribute) maybe_upsample_infer_occlusion_masks() (transmotion.dense_motion.DenseMotionNetwork method) MMapDebug (class in transmotion.debug) module transmotion transmotion.blocks transmotion.configs transmotion.cord_warp transmotion.data_loading transmotion.datasets transmotion.debug transmotion.dense_motion transmotion.inpainting transmotion.kp_detection transmotion.network_bundle transmotion.nn_blocks transmotion.viz transmotion.weights N nearest_scale() (transmotion.blocks.ImagePyramid method) NetworkBundleResult (class in transmotion.network_bundle) NetworksBundle (class in transmotion.network_bundle) no_background() (transmotion.network_bundle.NetworksBundle method) normalize_relative_to() (transmotion.kp_detection.KPResult method) num_blocks (transmotion.configs.DenseMotionConf attribute) num_down_blocks (transmotion.configs.InpaintingConfig attribute) num_epochs (transmotion.configs.TrainConfig attribute) num_occlusion_masks (transmotion.configs.DenseMotionConf attribute) (transmotion.configs.InpaintingConfig attribute) num_tps (transmotion.configs.TPSConfig attribute) (transmotion.dense_motion.WarppedCords attribute) (transmotion.kp_detection.KPResult attribute) num_workers (transmotion.configs.DataLoadingConfig attribute) O occlusion_masks (transmotion.dense_motion.DenseMotionResult attribute) OcclusionMasks (class in transmotion.dense_motion) optical_flow (transmotion.dense_motion.DenseMotionResult attribute) optimizer (transmotion.configs.TrainConfig attribute) OptimizerConfig (class in transmotion.configs) out_channels (transmotion.nn_blocks.Decoder attribute) (transmotion.nn_blocks.Hourglass attribute) out_dim (transmotion.nn_blocks.Decoder attribute) P param_name_iou() (in module transmotion.weights) parameters() (transmotion.network_bundle.NetworksBundle method) perceptual (transmotion.configs.LossConfig attribute) PerceptualLoss (class in transmotion.configs) points_per_tps (transmotion.configs.EquivarianceLoss attribute) (transmotion.configs.TPSConfig attribute) prob_inc_epochs (transmotion.configs.DropoutConfig attribute) pts_per_tps (transmotion.kp_detection.KPResult attribute) push() (transmotion.debug.MMapDebug method) R random() (transmotion.cord_warp.ThinPlateSpline class method) reconcile_weights() (in module transmotion.weights) ResBlock2d (class in transmotion.nn_blocks) resize_deform() (in module transmotion.inpainting) resize_occlude() (in module transmotion.inpainting) S SameBlock2d (class in transmotion.nn_blocks) scale_factor (transmotion.configs.DenseMotionConf attribute) scales (transmotion.configs.PerceptualLoss attribute) show_on_gird() (in module transmotion.viz) show_points_on_grid() (in module transmotion.viz) sigma_affine (transmotion.configs.EquivarianceLoss attribute) sigma_tps (transmotion.configs.EquivarianceLoss attribute) source (transmotion.datasets.SourceDrivingBatch attribute) (transmotion.datasets.SourceDrivingSample attribute) source_keypoints (transmotion.network_bundle.NetworkBundleResult attribute) SourceDrivingBatch (class in transmotion.datasets) SourceDrivingSample (class in transmotion.datasets) spatial_dim (transmotion.kp_detection.KPDetector attribute) start_epoch (transmotion.configs.DropoutConfig attribute) T theta (transmotion.cord_warp.ThinPlateSpline attribute) ThinPlateSpline (class in transmotion.cord_warp) to_gaussians() (transmotion.kp_detection.KPResult method) tps (transmotion.configs.DenseMotionConf attribute) (transmotion.configs.TrainConfig attribute) tps_dropout_softmax() (in module transmotion.dense_motion) tps_warp_to_keypoints_with_background() (in module transmotion.dense_motion) TPSConfig (class in transmotion.configs) TrainConfig (class in transmotion.configs) transmotion module transmotion.blocks module transmotion.configs module transmotion.cord_warp module transmotion.data_loading module transmotion.datasets module transmotion.debug module transmotion.dense_motion module transmotion.inpainting module transmotion.kp_detection module transmotion.network_bundle module transmotion.nn_blocks module transmotion.viz module transmotion.weights module U UpBlock2d (class in transmotion.nn_blocks) V Vgg19 (class in transmotion.nn_blocks) video_iterator() (in module transmotion.data_loading) video_to_tensor() (in module transmotion.data_loading) W warp (transmotion.configs.LossConfig attribute) warp_cords() (transmotion.cord_warp.ThinPlateSpline method) WarpLoss (class in transmotion.configs) warpped_cords (transmotion.dense_motion.WarppedCords attribute) WarppedCords (class in transmotion.dense_motion) weight_decay (transmotion.configs.OptimizerConfig attribute)