MCPcopy Create free account

hub / github.com/coperception/star / functions

Functions236 in github.com/coperception/star

Method__init__
(self, n_classes, ignore=None)
star/utils/metrics.py:15
Method__init__
(self, num_iterations)
star/utils/metrics.py:85
Method__init__
(self, nbr_classes, num_iterations_epoch)
star/utils/metrics.py:130
Method__init__
(self, model, teacher, config, optimizer, kd_flag)
star/utils/SegModule.py:8
Method__init__
Inherited from the V2XSimSeg dataset, now support multi timestamp loading output padded_voxel_points_next are for other timestamp
star/datasets/MultiTempSeg.py:48
Method__init__
Inherited from the V2XSimDet dataset, now support multi timestamp training output padded_voxel_points_next are for other timestamp
star/datasets/MultiTempDet.py:11
Method__init__
(self, in_channels, num_hiddens, num_residual_layers, num_residual_hiddens)
star/models/VQVAE.py:26
Method__init__
(self, num_embeddings, embedding_dim, commitment_cost)
star/models/VQVAE.py:39
Method__init__
(self, num_embeddings, embedding_dim, commitment_cost, decay, epsilon=1e-5)
star/models/VQVAE.py:84
Method__init__
(self, in_channels, num_hiddens, num_residual_layers, num_residual_hiddens)
star/models/VQVAE.py:152
Method__init__
(self, in_channels, num_hiddens, num_residual_layers, num_residual_hiddens)
star/models/VQVAE.py:184
Method__init__
(self, num_hiddens, num_residual_layers, num_residual_hiddens, num_embeddings, embedding_dim
star/models/VQVAE.py:229
Method__init__
(self, num_hiddens, num_residual_layers, num_residual_hiddens, num_embeddings, embedding_dim
star/models/VQVAE.py:265
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=1024, depth=24, num_heads=16,
star/models/multiagent_mae.py:29
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=1024, depth=24, num_heads=16,
star/models/multiagent_mae.py:351
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=1024, depth=24, num_heads=16,
star/models/multiagent_mae.py:688
Method__init__
(self, input_size=16, output_size=256, input_chans=512, output_chans=13)
star/models/mae_base.py:63
Method__init__
(self, num_embeddings, embedding_dim, commitment_cost)
star/models/mae_base.py:108
Method__init__
(self, num_embeddings, embedding_dim, commitment_cost, decay, epsilon=1e-5)
star/models/mae_base.py:177
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=1024, depth=24, num_heads=16,
star/models/mae_base.py:260
Method__init__
(self, height_feat_size, compress_level=0, train_completion=False)
star/models/CNNNet.py:8
Method__init__
(self, height_feat_size=13, compress_level=0, train_completion=False)
star/models/CNNNet.py:270
Method__init__
(self, in_channel, out_channel, kernel_size, stride, padding)
star/models/CNNNet.py:454
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=1024, depth=24, num_heads=16,
star/models/VQSTAR.py:18
Method__str__
(self)
star/utils/misc.py:77
Method__str__
(self)
star/utils/misc.py:108
Method_init_weights
(self, m)
star/models/mae_base.py:367
Methodadd_meter
(self, name, meter)
star/utils/misc.py:120
Functionadjust_learning_rate
Decay the learning rate with half-cycle cosine after warmup
star/utils/lr_sched.py:9
Methodagents_to_batch
(feats)
star/models/mae_base.py:638
Functionall_reduce_mean
(x)
star/utils/misc.py:332
Functionamo_fusion_bev_multi_mae_vit_base_patch16_dec512d6b
(**kwargs)
star/models/multiagent_mae.py:1193
Functionamo_fusion_bev_multi_mae_vit_base_patch16_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1207
Functionamo_fusion_bev_multi_mae_vit_base_patch8_dec512d6b
(**kwargs)
star/models/multiagent_mae.py:1186
Functionamo_fusion_bev_multi_mae_vit_base_patch8_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1200
Functionamo_individual_bev_multi_mae_vit_base_patch16_dec256d4b
(**kwargs)
star/models/multiagent_mae.py:1143
Functionamo_individual_bev_multi_mae_vit_base_patch16_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1165
Functionamo_individual_bev_multi_mae_vit_base_patch32_dec256d4b
(**kwargs)
star/models/multiagent_mae.py:1150
Functionamo_individual_bev_multi_mae_vit_base_patch32_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1172
Functionamo_individual_bev_multi_mae_vit_base_patch4_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1179
Functionamo_individual_bev_multi_mae_vit_base_patch8_dec256d4b
(**kwargs)
star/models/multiagent_mae.py:1136
Functionamo_individual_bev_multi_mae_vit_base_patch8_dec512d4b
(**kwargs)
star/models/multiagent_mae.py:1129
Functionamo_individual_bev_multi_mae_vit_base_patch8_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1158
Methodamortized_complement_masking
x_ts: [bxa x time_stamp, L, D] NOTE: the returned mask and ids_restore has a different dimensionality from the random masking
star/models/multiagent_mae.py:440
Methodamortized_complement_masking
x_ts: [bxa x time_stamp, L, D] NOTE: the returned mask and ids_restore has a different dimensionality from the random masking
star/models/multiagent_mae.py:800
Methodamortized_complement_masking
x_ts: [bxa x time_stamp, L, D] NOTE: the returned mask and ids_restore has a different dimensionality from the random masking
star/models/VQSTAR.py:141
Methodamortized_complement_unmasking
x_ts: [BxAxts, L, D] mask: [BxA, ts, L] -> this is real L ids_restore: [BxA, L] -> this is real L restore, add tempor
star/models/multiagent_mae.py:478
Methodamortized_complement_unmasking
x_ts: [BxAxts, L, D] mask: [BxA, ts, L] -> this is real L ids_restore: [BxA, L] -> this is real L restore, add tempor
star/models/multiagent_mae.py:836
Methodamortized_complement_unmasking
x_ts: [BxAxts, L, D] mask: [BxA, ts, L] -> this is real L ids_restore: [BxA, L] -> this is real L restore, add tempor
star/models/VQSTAR.py:177
Methodamortized_random_masking
Random masking for each data of each timestamp. Maintain indices for decoder to fuse and possibly fill in mask tokens x_ts: [
star/models/multiagent_mae.py:390
Methodamortized_random_masking
Random masking for each data of each timestamp. Maintain indices for decoder to fuse and possibly fill in mask tokens x_ts: [
star/models/multiagent_mae.py:743
Methodamortized_random_masking
Random masking for each data of each timestamp. Maintain indices for decoder to fuse and possibly fill in mask tokens x_ts: [
star/models/VQSTAR.py:84
Methodamortized_random_unmasking
Reverse masking with fillings from other timestamp, using mask and torch where x_ts: [BxAxts, L, D] mask: [BxAxts, L]
star/models/multiagent_mae.py:399
Methodamortized_random_unmasking
Reverse masking with fillings from other timestamp, using mask and torch where x_ts: [BxAxts, L, D] mask: [BxAxts, L] -> full
star/models/multiagent_mae.py:752
Methodamortized_random_unmasking
Reverse masking with fillings from other timestamp, using mask and torch where x_ts: [BxAxts, L, D] mask: [BxAxts, L] -> full
star/models/VQSTAR.py:93
Methodavg
(self)
star/utils/misc.py:61
Functionbev_multi_mae_vit_base_patch16_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1250
Methodbuild_feature_list
(self, batch_size: int, feat_maps: dict)
star/models/mae_base.py:572
Methodbuild_local_communication_matrix
(feature_list: list)
star/models/mae_base.py:586
Methodbuild_neighbors_feature_list
(self, b, agent_idx, all_warp, num_agent, local_com_mat, device, size)
star/models/mae_base.py:617
Functioncheck_folder
(folder_path)
completion/test_coseg.py:30
Functioncheck_folder
(folder_path)
completion/test_completion.py:35
Methodcomplement_masking
mask x1 with mask_ratio mask x2 with 1 - mask ratio so that they are complementary x1, x2: same size [N, D, L]
star/models/mae_base.py:454
Methodforward
(self, inputs, target)
star/utils/softmax_focal_loss.py:12
Methodforward
(self, x)
star/models/VQVAE.py:21
Methodforward
(self, x)
star/models/VQVAE.py:32
Methodforward
(self, inputs)
star/models/VQVAE.py:49
Methodforward
(self, inputs)
star/models/VQVAE.py:101
Methodforward
(self, inputs)
star/models/VQVAE.py:172
Methodforward
(self, inputs)
star/models/VQVAE.py:212
Methodforward
(self, x)
star/models/VQVAE.py:251
Methodforward
(self, bevs, trans_matrices=None, num_agent_tensor=None, batch_size=None)
star/models/VQVAE.py:394
Methodforward
Modified from the original forward, make fusion happen
star/models/multiagent_mae.py:235
Methodforward
(self, imgs1, imgs_next, teacher, trans_matrices, num_agent_tensor, batch_size, mask_ratio=0.7)
star/models/multiagent_mae.py:335
Methodforward
Encoder encodes each timestamp alone Decoder fuses multi timestamp together
star/models/multiagent_mae.py:660
Methodforward
Encoder encodes each timestamp alone Decoder fuses multi timestamp together
star/models/multiagent_mae.py:1084
Methodforward
(self, x)
star/models/mae_base.py:39
Methodforward
(self, x)
star/models/mae_base.py:81
Methodforward
(self, inputs)
star/models/mae_base.py:122
Methodforward
(self, inputs)
star/models/mae_base.py:197
Methodforward
(self, imgs1, imgs2, mask_ratio=0.75)
star/models/mae_base.py:742
Methodforward
(self, x)
star/models/CNNNet.py:273
Methodforward
(self, bevs, trans_matrices=None, num_agent_tensor=None, batch_size=None)
star/models/CNNNet.py:426
Methodforward
(self, x)
star/models/CNNNet.py:465
Methodforward
Encoder encodes each timestamp alone Decoder fuses multi timestamp together
star/models/VQSTAR.py:421
Methodforward_encoder_all
x1: [bxa, C, H, W] x_next: [bxa, ts-1, C, H, W] beq_next_frames
star/models/multiagent_mae.py:946
Methodforward_encoder_all
x1: [bxa, C, H, W] x_next: [bxa, ts-1, C, H, W] beq_next_frames
star/models/VQSTAR.py:287
Methodforward_encoder_partial
x1: [bxa, C, H, W] x_next: [bxa, ts-1, C, H, W] beq_next_frames
star/models/multiagent_mae.py:986
Methodforward_encoder_partial
x1: [bxa, C, H, W] x_next: [bxa, ts-1, C, H, W] beq_next_frames
star/models/VQSTAR.py:317
Methodforward_focal_loss
(self, target, pred)
star/models/multiagent_mae.py:1081
Methodforward_focal_loss
(self, target, pred)
star/models/VQSTAR.py:418
Methodforward_loss
overwrite the original forward_loss, now calculate loss on the entire image
star/models/multiagent_mae.py:634
Methodforward_loss
overwrite the original forward_loss, now calculate loss on the entire image
star/models/VQSTAR.py:394
Methodfusion
(self, mode="sum")
star/models/mae_base.py:624
Functionfusion_bev_multi_mae_vit_base_patch16_dec1024d8b
(**kwargs)
star/models/multiagent_mae.py:1257
Functionfusion_bev_multi_mae_vit_base_patch16_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1222
Functionfusion_bev_multi_mae_vit_base_patch1_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1236
Functionfusion_bev_multi_mae_vit_base_patch32_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1243
Functionfusion_bev_multi_mae_vit_base_patch8_dec512d8b
(**kwargs)
star/models/multiagent_mae.py:1229
Methodget_eval_mask_Lidar
eval_mask_lidar is only to ingore unknown voxels in groundtruth
star/utils/metrics.py:192
← previousnext →101–200 of 236, ranked by callers