MCPcopy Create free account

hub / github.com/andrewsonga/generative_view_stitching / functions

Functions993 in github.com/andrewsonga/generative_view_stitching

⨍ Functions 993◇ Types & classes 210
Method__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
algorithms/vae/common/losses/discriminator.py:117
Method__init__
Construct a 3D PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input volumes
algorithms/vae/common/losses/discriminator.py:189
Method__init__
(self, cause, keys=None, visited=None)
algorithms/vae/common/losses/utils.py:42
Method__init__
(self)
algorithms/vae/common/losses/lpips.py:54
Method__init__
(self, chn_in, chn_out=1, use_dropout=False)
algorithms/vae/common/losses/lpips.py:70
Method__init__
(self, requires_grad=False, pretrained=True)
algorithms/vae/common/losses/lpips.py:86
Method__init__
( self, disc_start, kl_weight=1.0, perceptual_weight=1.0, # --- Discri
algorithms/vae/common/losses/perceptual_loss.py:196
Method__init__
( self, cfg: DictConfig, )
algorithms/vae/video_vae/trainer.py:20
Method__init__
( self, z_channels: int, hidden_size: int, hidden_size_mult: Tuple[int] = (1,
algorithms/vae/video_vae/model.py:154
Method__init__
( self, hidden_size: int = 128, z_channels: int = 4, hidden_size_mult: Tuple[i
algorithms/vae/video_vae/model.py:282
Method__init__
(self, cfg: DictConfig)
algorithms/common/base_pytorch_algo.py:23
Method__init__
(self, cfg: DictConfig)
algorithms/common/base_algo.py:12
Method__init__
( self, img_size: int = 256, distance: str = "cosine", # Default to feature similarit
algorithms/common/metrics/video/met3r.py:24
Method__init__
(self, reset_real_features: bool = True, **kwargs: Any)
algorithms/common/metrics/video/fvd.py:13
Method__init__
(self, num_classes: int = 400, **kwargs: Any)
algorithms/common/metrics/video/inception_score.py:19
Method__init__
Initialize DepthCollisionMetric for DFOT system. Args: registry: Shared video metric model registry (required for DFOT c
algorithms/common/metrics/video/depth_collision.py:31
Method__init__
( self, registry: SharedVideoMetricModelRegistry, metric_types: List[str] | List[Video
algorithms/common/metrics/video/video_metric.py:88
Method__init__
( self, registry: SharedVideoMetricModelRegistry, similarity_metric: str = "lpips",
algorithms/common/metrics/video/loop_consistency.py:20
Method__init__
(self)
algorithms/common/metrics/video/shared_registry.py:62
Method__init__
( self, registry: SharedVideoMetricModelRegistry, reset_real_features: bool = True,
algorithms/common/metrics/video/fvmd.py:16
Method__init__
( self, registry: SharedVideoMetricModelRegistry, similarity_metric: str = 'lpips',
algorithms/common/metrics/video/temporal_consistency.py:17
Method__init__
( self, registry: SharedVideoMetricModelRegistry, reset_real_features=True, no
algorithms/common/metrics/video/fid.py:14
Method__init__
( self, registry: Optional[SharedVideoMetricModelRegistry], features: int, res
algorithms/common/metrics/video/base_fid.py:18
Method__init__
( self, registry: SharedVideoMetricModelRegistry, reduction: Literal["sum", "mean"] =
algorithms/common/metrics/video/lpips.py:20
Method__init__
( self, registry: SharedVideoMetricModelRegistry, resolution: int = 224, **kwa
algorithms/common/metrics/video/vbench/dynamic_degree.py:19
Method__init__
(self, registry: SharedVideoMetricModelRegistry, **kwargs: Any)
algorithms/common/metrics/video/vbench/dimension.py:15
Method__init__
( self, registry: SharedVideoMetricModelRegistry, resolution: int = 224, **kwa
algorithms/common/metrics/video/vbench/motion_smoothness.py:17
Method__init__
Args: registry: The shared video metric model registry. resolution: The resolution to resize the videos to before com
algorithms/common/metrics/video/vbench/vbench.py:65
Method__init__
(self)
algorithms/common/metrics/video/models/laion.py:8
Method__init__
(self, name: str)
algorithms/common/metrics/video/models/clip.py:14
Method__init__
(self)
algorithms/common/metrics/video/models/musiq.py:11
Method__init__
(self)
algorithms/common/metrics/video/models/dino.py:9
Method__init__
(self)
algorithms/common/metrics/video/models/i3d.py:24
Method__init__
(self, in_channels, out_channels, kernel_size, stride, groups=1)
algorithms/common/metrics/video/models/pips2.py:17
Method__init__
( self, in_channels, out_channels, kernel_size, stride, groups
algorithms/common/metrics/video/models/pips2.py:45
Method__init__
(self, in_planes, planes, norm_fn="group", stride=1)
algorithms/common/metrics/video/models/pips2.py:142
Method__init__
( self, input_dim=3, output_dim=128, stride=8, norm_fn="batch", dropout=0.0 )
algorithms/common/metrics/video/models/pips2.py:204
Method__init__
(self, latent_dim=128, hidden_dim=128, corr_levels=4, corr_radius=3)
algorithms/common/metrics/video/models/pips2.py:309
Method__init__
(self, fmaps, num_levels=4, radius=4)
algorithms/common/metrics/video/models/pips2.py:417
Method__init__
( self, resolution: int = 256, segment_length: int = 16, num_points: int = 400
algorithms/common/metrics/video/models/motion_extractor.py:21
Method__init__
(self, in_planes, planes, norm_fn="group", stride=1)
algorithms/common/metrics/video/models/amt/feat_enc.py:6
Method__init__
(self, in_planes, planes, norm_fn="group", stride=1)
algorithms/common/metrics/video/models/amt/feat_enc.py:67
Method__init__
(self, output_dim=128, norm_fn="batch", dropout=0.0)
algorithms/common/metrics/video/models/amt/feat_enc.py:196
Method__init__
(self, output_dim=128, norm_fn="batch", dropout=0.0)
algorithms/common/metrics/video/models/amt/feat_enc.py:272
Method__init__
(self, dims, divisor=16)
algorithms/common/metrics/video/models/amt/utils.py:32
Method__init__
(self, in_channels, side_channels, bias=True)
algorithms/common/metrics/video/models/amt/ifrnet.py:39
Method__init__
(self, in_ch, out_ch, skip_ch)
algorithms/common/metrics/video/models/amt/ifrnet.py:124
Method__init__
(self, in_ch, out_ch, skip_ch)
algorithms/common/metrics/video/models/amt/ifrnet.py:142
Method__init__
(self)
algorithms/common/metrics/video/models/amt/__init__.py:13
Method__init__
(self, in_ch, skip_ch, num_flows=3)
algorithms/common/metrics/video/models/amt/multi_flow.py:58
Method__init__
( self, corr_radius=3, corr_lvls=4, num_flows=3, channels=[20, 32, 44,
algorithms/common/metrics/video/models/amt/amt_s.py:22
Method__init__
( self, cdim, hidden_dim, flow_dim, corr_dim, fc_dim,
algorithms/common/metrics/video/models/amt/raft.py:38
Method__init__
( self, cdim, hidden_dim, flow_dim, corr_dim, corr_dim2,
algorithms/common/metrics/video/models/amt/raft.py:103
Method__init__
(self, in_planes, planes, norm_fn="group", stride=1)
algorithms/common/metrics/video/models/raft/extractor.py:6
Method__init__
(self, in_planes, planes, norm_fn="group", stride=1)
algorithms/common/metrics/video/models/raft/extractor.py:61
Method__init__
(self, output_dim=128, norm_fn="batch", dropout=0.0)
algorithms/common/metrics/video/models/raft/extractor.py:198
Method__init__
(self, dims, mode="sintel")
algorithms/common/metrics/video/models/raft/utils.py:12
Method__init__
(self)
algorithms/common/metrics/video/models/raft/__init__.py:12
Method__init__
(self, fmap1, fmap2, num_levels=4, radius=4)
algorithms/common/metrics/video/models/raft/corr.py:7
Method__init__
(self, input_dim=128, hidden_dim=256)
algorithms/common/metrics/video/models/raft/update.py:7
Method__init__
(self, hidden_dim=128, input_dim=192 + 128)
algorithms/common/metrics/video/models/raft/update.py:36
Method__init__
(self, args)
algorithms/common/metrics/video/models/raft/update.py:77
Method__init__
(self, args)
algorithms/common/metrics/video/models/raft/update.py:95
Method__init__
(self, args, hidden_dim=96)
algorithms/common/metrics/video/models/raft/update.py:116
Method__init__
(self, args, hidden_dim=128, input_dim=128)
algorithms/common/metrics/video/models/raft/update.py:132
Method__init__
(self, args)
algorithms/common/metrics/video/models/raft/raft.py:12
Method__init__
(self, cfg: DictConfig)
algorithms/dfot/gvs_video_pose.py:18
Method__init__
(self, cfg: DictConfig)
algorithms/dfot/dfot_video_pose.py:17
Method__init__
Initialize the history segment, see the class docstring for more details.
algorithms/dfot/history_guidance.py:57
Method__init__
(self, num_hist_segments: int, disabled: bool = False)
algorithms/dfot/history_guidance.py:185
Method__init__
Initialize the history guidance manager. Args ---- history_guidance: HistoryGuidance The history guidanc
algorithms/dfot/history_guidance.py:336
Method__init__
Initialize the history guidance.
algorithms/dfot/history_guidance.py:602
Method__init__
(self, history_guidance: "HistoryGuidance", mask: torch.Tensor)
algorithms/dfot/history_guidance.py:952
Method__init__
(self, history_guidance: "HistoryGuidance", mask: torch.Tensor)
algorithms/dfot/history_guidance.py:1041
Method__init__
(self, cfg: DictConfig)
algorithms/dfot/stochsync_video_pose.py:18
Method__init__
(self, cfg: DictConfig)
algorithms/dfot/gvs_no_omniguide_video_pose.py:17
Method__init__
(self, cfg: DictConfig)
algorithms/dfot/dfot_rag_video_pose.py:20
Method__init__
(self, cfg: DictConfig)
algorithms/dfot/gvs_scalable_video_pose.py:17
Method__init__
(self, cfg: DictConfig)
algorithms/dfot/dfot_video.py:37
Method__init__
( self, cfg: DictConfig, backbone_cfg: DictConfig, x_shape: torch.Size,
algorithms/dfot/diffusion/discrete_diffusion.py:35
Method__init__
( self, cfg: DictConfig, backbone_cfg: DictConfig, x_shape: torch.Size,
algorithms/dfot/diffusion/continuous_diffusion.py:95
Method__init__
( self, cfg: DictConfig, x_shape: torch.Size, max_tokens: int, externa
algorithms/dfot/backbones/base_backbone.py:13
Method__init__
( self, num_channels: int, p: float = 1.0, )
algorithms/dfot/backbones/modules/embeddings.py:35
Method__init__
( self, dim: int, time_embed_dim: int, use_fourier: bool = False, p: f
algorithms/dfot/backbones/modules/embeddings.py:68
Method__init__
(self, num_channels, bandwidth=1)
algorithms/dfot/backbones/modules/embeddings.py:99
Method__init__
Args: dims: the number of dimensions for each axis. sizes: the maximum length for each axis.
algorithms/dfot/backbones/modules/embeddings.py:161
Method__init__
( self, dim: int, seq_len: int, theta: float = 10000.0, flatten: bool
algorithms/dfot/backbones/modules/embeddings.py:230
Method__init__
( self, dim: int, sizes: Tuple[int, int], theta: float = 10000.0, flat
algorithms/dfot/backbones/modules/embeddings.py:247
Method__init__
( self, dim: int, sizes: Tuple[int, int, int], theta: float = 10000.0,
algorithms/dfot/backbones/modules/embeddings.py:265
Method__init__
(self, p: float = 0.0)
algorithms/dfot/backbones/modules/embeddings.py:294
Method__init__
( self, cond_dim: int, cond_emb_dim: int, dropout_prob: float = 0.0, )
algorithms/dfot/backbones/modules/embeddings.py:323
Method__init__
( self, dropout_prob: float = 0.1, img_size: Optional[int] = 224, patch_size:
algorithms/dfot/backbones/modules/embeddings.py:344
Method__init__
Initialize the RMSNorm normalization layer. Args: dim (int): The dimension of the input tensor. eps (float,
algorithms/dfot/backbones/modules/normalization.py:11
Method__init__
( self, cfg: DictConfig, x_shape: torch.Size, max_tokens: int, externa
algorithms/dfot/backbones/u_vit/u_vit3d_pose.py:18
Method__init__
(self, in_channels: int, dim: int, patch_size: int)
algorithms/dfot/backbones/u_vit/u_vit_blocks.py:22
Method__init__
(self, dim: int, out_channels: int, patch_size: int)
algorithms/dfot/backbones/u_vit/u_vit_blocks.py:39
Method__init__
(self, dim: int, emb_dim: int)
algorithms/dfot/backbones/u_vit/u_vit_blocks.py:101
Method__init__
( self, dim: int, heads: int, emb_dim: int, rope: Optional[RotaryEmbed
algorithms/dfot/backbones/u_vit/u_vit_blocks.py:124
Method__init__
If len(sizes) == 2, each axis corresponds to each dimension. If len(sizes) == 3, the first dimension corresponds to the first axis, a
algorithms/dfot/backbones/u_vit/u_vit_blocks.py:171
Method__init__
( self, dim: int, heads: int, emb_dim: int, dropout: float, us
algorithms/dfot/backbones/u_vit/u_vit_blocks.py:201
← previousnext →501–600 of 993, ranked by callers