MCPcopy Create free account

hub / github.com/coperception/star / types & classes

Types & classes34 in github.com/coperception/star

↓ 8 callersClassAmortizedIndivMMAEViT
Temporal amorized version
star/models/multiagent_mae.py:684
↓ 5 callersClassFusionMultiAgentMAEViT
joint reconstruction Serves as the DetModel, handles fusion/communication between encoder and decoder
star/models/multiagent_mae.py:24
↓ 4 callersClassAmortizedFusionMMAEViT
Temporal amorized version
star/models/multiagent_mae.py:347
↓ 4 callersClassVQVAENet
wrapper of vqvae, handles fusion
star/models/VQVAE.py:263
↓ 3 callersClassCoModule
star/utils/CoModule.py:13
↓ 3 callersClassMultiTempV2XSimDet
star/datasets/MultiTempDet.py:10
↓ 3 callersClassiouEval
star/utils/metrics.py:14
↓ 2 callersClassCNNNet
Modified from: https://arxiv.org/pdf/2012.12395.pdf Args: config (object): The Config object. layer (int, optional): Col
star/models/CNNNet.py:283
↓ 2 callersClassConv3D
3D cnn used in the encoder.
star/models/CNNNet.py:451
↓ 2 callersClassConvPred
a upsampling and deconve module in repalce of the original final linear projection, to obtain the prediction of the original image. for p
star/models/mae_base.py:16
↓ 2 callersClassConvPred16
a upsampling and deconve module in repalce of the original final linear projection, to obtain the prediction of the original image. for p
star/models/mae_base.py:57
↓ 2 callersClassResidualStack
star/models/VQVAE.py:25
↓ 2 callersClassSmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
star/utils/misc.py:24
↓ 1 callersClassIndivMultiAgentMAEViT
individual reconstruction Serves as the DetModel, handles fusion/communication after encoder and decoder
star/models/multiagent_mae.py:261
↓ 1 callersClassLossesTrackEpoch
star/utils/metrics.py:84
↓ 1 callersClassMetrics
star/utils/metrics.py:128
↓ 1 callersClassMultiAgentMaskedAutoencoderViT
Masked Autoencoder with VisionTransformer backbone
star/models/mae_base.py:257
↓ 1 callersClassMultiTempV2XSimSeg
star/datasets/MultiTempSeg.py:47
↓ 1 callersClassResidual
star/models/VQVAE.py:7
↓ 1 callersClassSTARVectorQuantizer
star/models/mae_base.py:107
↓ 1 callersClassSTARVectorQuantizerEMA
star/models/mae_base.py:176
↓ 1 callersClassSTPN_KD
Used by non-intermediate models. Pass the output from encoder directly to decoder.
star/models/CNNNet.py:267
↓ 1 callersClassSegModule
star/utils/SegModule.py:7
↓ 1 callersClassVQSTARViT
STAR + Vector Quantizer
star/models/VQSTAR.py:14
↓ 1 callersClassVQVAEDecoder
star/models/VQVAE.py:183
↓ 1 callersClassVQVAEEncoder
star/models/VQVAE.py:151
↓ 1 callersClassVQVAEModel
Use vqvae to reconstruct the scene
star/models/VQVAE.py:227
↓ 1 callersClassVectorQuantizer
star/models/VQVAE.py:38
↓ 1 callersClassVectorQuantizerEMA
star/models/VQVAE.py:83
ClassBackbone
The backbone class that contains encode and decode function
star/models/CNNNet.py:5
ClassMetricLogger
star/utils/misc.py:86
ClassNativeScalerWithGradNormCount
star/utils/misc.py:251
ClassSoftmaxFocalLoss
star/utils/softmax_focal_loss.py:5
ClassTransform
star/datasets/MultiTempSeg.py:13