Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/coperception/star
/ types & classes
Types & classes
34 in github.com/coperception/star
⨍
Functions
236
◇
Types & classes
34
↓ 8 callers
Class
AmortizedIndivMMAEViT
Temporal amorized version
star/models/multiagent_mae.py:684
↓ 5 callers
Class
FusionMultiAgentMAEViT
joint reconstruction Serves as the DetModel, handles fusion/communication between encoder and decoder
star/models/multiagent_mae.py:24
↓ 4 callers
Class
AmortizedFusionMMAEViT
Temporal amorized version
star/models/multiagent_mae.py:347
↓ 4 callers
Class
VQVAENet
wrapper of vqvae, handles fusion
star/models/VQVAE.py:263
↓ 3 callers
Class
CoModule
star/utils/CoModule.py:13
↓ 3 callers
Class
MultiTempV2XSimDet
star/datasets/MultiTempDet.py:10
↓ 3 callers
Class
iouEval
star/utils/metrics.py:14
↓ 2 callers
Class
CNNNet
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 callers
Class
Conv3D
3D cnn used in the encoder.
star/models/CNNNet.py:451
↓ 2 callers
Class
ConvPred
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 callers
Class
ConvPred16
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 callers
Class
ResidualStack
star/models/VQVAE.py:25
↓ 2 callers
Class
SmoothedValue
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 callers
Class
IndivMultiAgentMAEViT
individual reconstruction Serves as the DetModel, handles fusion/communication after encoder and decoder
star/models/multiagent_mae.py:261
↓ 1 callers
Class
LossesTrackEpoch
star/utils/metrics.py:84
↓ 1 callers
Class
Metrics
star/utils/metrics.py:128
↓ 1 callers
Class
MultiAgentMaskedAutoencoderViT
Masked Autoencoder with VisionTransformer backbone
star/models/mae_base.py:257
↓ 1 callers
Class
MultiTempV2XSimSeg
star/datasets/MultiTempSeg.py:47
↓ 1 callers
Class
Residual
star/models/VQVAE.py:7
↓ 1 callers
Class
STARVectorQuantizer
star/models/mae_base.py:107
↓ 1 callers
Class
STARVectorQuantizerEMA
star/models/mae_base.py:176
↓ 1 callers
Class
STPN_KD
Used by non-intermediate models. Pass the output from encoder directly to decoder.
star/models/CNNNet.py:267
↓ 1 callers
Class
SegModule
star/utils/SegModule.py:7
↓ 1 callers
Class
VQSTARViT
STAR + Vector Quantizer
star/models/VQSTAR.py:14
↓ 1 callers
Class
VQVAEDecoder
star/models/VQVAE.py:183
↓ 1 callers
Class
VQVAEEncoder
star/models/VQVAE.py:151
↓ 1 callers
Class
VQVAEModel
Use vqvae to reconstruct the scene
star/models/VQVAE.py:227
↓ 1 callers
Class
VectorQuantizer
star/models/VQVAE.py:38
↓ 1 callers
Class
VectorQuantizerEMA
star/models/VQVAE.py:83
Class
Backbone
The backbone class that contains encode and decode function
star/models/CNNNet.py:5
Class
MetricLogger
star/utils/misc.py:86
Class
NativeScalerWithGradNormCount
star/utils/misc.py:251
Class
SoftmaxFocalLoss
star/utils/softmax_focal_loss.py:5
Class
Transform
star/datasets/MultiTempSeg.py:13