Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/3DTopia/DynamicCity
/ types & classes
Types & classes
41 in github.com/3DTopia/DynamicCity
⨍
Functions
220
◇
Types & classes
41
↓ 2 callers
Class
Attention
dynamic_city/utils/attention_utils.py:6
↓ 2 callers
Class
DiT
Diffusion model with a Transformer backbone.
dynamic_city/diffusion/models.py:76
↓ 2 callers
Class
VAETrainer
dynamic_city/trainer/vae_trainer.py:28
↓ 2 callers
Class
VoxelEncoder
dynamic_city/vae/encoder_blocks.py:8
↓ 1 callers
Class
CmdCondEmbedder
dynamic_city/diffusion/embedders.py:137
↓ 1 callers
Class
CrossAttention
dynamic_city/utils/attention_utils.py:32
↓ 1 callers
Class
DiTBlock
A DiT block with adaptive layer norm zero (adaLN-Zero) conditioning.
dynamic_city/diffusion/models.py:24
↓ 1 callers
Class
DiTTrainer
dynamic_city/trainer/dit_trainer.py:23
↓ 1 callers
Class
DynamicCityAE
Main class for the DynamicCity autoencoder. Enables easy switching between different encoders and decoders and optional VAE.
dynamic_city/vae/vae.py:9
↓ 1 callers
Class
FinalLayer
The final layer of DiT.
dynamic_city/diffusion/models.py:55
↓ 1 callers
Class
GaussianDiffusion
Utilities for training and sampling diffusion models. Original ported from this codebase: https://github.com/hojonathanho/diffusion/blob/
dynamic_city/diffusion/gaussian_diffusion.py:144
↓ 1 callers
Class
HexCondEmbedder
dynamic_city/diffusion/embedders.py:106
↓ 1 callers
Class
HexPlaneVAE
dynamic_city/vae/vae.py:52
↓ 1 callers
Class
HexplaneConvBlock
dynamic_city/vae/decoder_blocks.py:9
↓ 1 callers
Class
LayoutCondEmbedder
dynamic_city/diffusion/embedders.py:112
↓ 1 callers
Class
LossSecondMomentResampler
dynamic_city/diffusion/timestep_sampler.py:120
↓ 1 callers
Class
Metrics
dynamic_city/utils/metrics.py:8
↓ 1 callers
Class
PlaneDownsampler
dynamic_city/vae/encoder_blocks.py:52
↓ 1 callers
Class
PlaneUpsampler
dynamic_city/vae/decoder_blocks.py:43
↓ 1 callers
Class
SpacedDiffusion
A diffusion process which can skip steps in a base diffusion process. :param use_timesteps: a collection (sequence or set) of timesteps from
dynamic_city/diffusion/respace.py:65
↓ 1 callers
Class
TimestepEmbedder
Embeds scalar timesteps into vector representations.
dynamic_city/diffusion/embedders.py:8
↓ 1 callers
Class
TrajCondEmbedder
dynamic_city/diffusion/embedders.py:126
↓ 1 callers
Class
Transformer
Simple Transformer block with flash attention.
dynamic_city/vae/encoder_blocks.py:72
↓ 1 callers
Class
UniformSampler
dynamic_city/diffusion/timestep_sampler.py:62
↓ 1 callers
Class
VoxelDecoderBlock
dynamic_city/vae/decoder_blocks.py:60
↓ 1 callers
Class
_WrappedModel
dynamic_city/diffusion/respace.py:117
Class
CarlaSCHexPlaneDataset
dynamic_city/dataset/carlasc.py:40
Class
CarlaSCOccSequenceDataset
dynamic_city/dataset/carlasc.py:10
Class
Command
dynamic_city/utils/data_utils.py:57
Class
ConvDecoder
dynamic_city/vae/decoder.py:36
Class
DecoderBase
dynamic_city/vae/decoder.py:9
Class
Embedder
dynamic_city/diffusion/embedders.py:49
Class
EncoderBase
dynamic_city/vae/encoder.py:12
Class
HexPlaneDataset
dynamic_city/dataset/hexplane_dataset.py:11
Class
LossAwareSampler
dynamic_city/diffusion/timestep_sampler.py:71
Class
LossType
dynamic_city/diffusion/gaussian_diffusion.py:46
Class
ModelMeanType
Which type of output the model predicts.
dynamic_city/diffusion/gaussian_diffusion.py:23
Class
ModelVarType
What is used as the model's output variance. The LEARNED_RANGE option has been added to allow the model to predict values between FIXED_S
dynamic_city/diffusion/gaussian_diffusion.py:33
Class
OccSequenceDataset
dynamic_city/dataset/occ_sequence_dataset.py:10
Class
ScheduleSampler
A distribution over timesteps in the diffusion process, intended to reduce variance of the objective. By default, samplers perform unbias
dynamic_city/diffusion/timestep_sampler.py:27
Class
TrEncoder
dynamic_city/vae/encoder.py:72