Functions — github.com/WanderRainy/MaRS

hub / github.com/WanderRainy/MaRS / functions

Functions61 in github.com/WanderRainy/MaRS

⨍ Functions 61 ◇ Types & classes 8

↓ 27 callersFunctionprint

(*args, **kwargs)

utils/misc.py:179

↓ 3 callersFunctionis_dist_avail_and_initialized

()

utils/misc.py:190

↓ 3 callersMethodload_state_dict

(self, state_dict)

utils/misc.py:279

↓ 3 callersMethodstate_dict

(self)

utils/misc.py:276

↓ 3 callersMethodupdate

(self, **kwargs)

utils/misc.py:94

↓ 2 callersMethod__init__

(self, dim, num_layers, num_heads, mlp_ratio=4.0, drop=0.0, aa_order=('modality', 'global'))

models/MaRS.py:12

↓ 2 callersFunctionbuild_mars

(model_type='base', contrast_type='clip')

models/MaRS.py:171

↓ 2 callersFunctionconvert_state_dict

(state_dict, prefix)

utils/convert_mars_checkpoints_to_swin.py:16

↓ 2 callersFunctionget_1d_sincos_pos_embed_from_grid

embed_dim: output dimension for each position pos: a list of positions to be encoded: size (M,) out: (M, D)

utils/pos_embed.py:49

↓ 2 callersFunctionget_patch_start_positions

(length, tile_size)

data/split_patch.py:19

↓ 2 callersFunctionget_world_size

()

utils/misc.py:198

↓ 2 callersFunctionsetup_for_distributed

This function disables printing when not in master process

utils/misc.py:173

↓ 2 callersMethodupdate

(self, value, n=1)

utils/misc.py:40

↓ 1 callersMethod_load_or_build_cache

(self)

mars_dataset.py:65

↓ 1 callersMethodadd_meter

(self, name, meter)

utils/misc.py:123

↓ 1 callersFunctionclip_contrastive

(rgb_feats, sar_feats, t)

models/clips.py:5

↓ 1 callersFunctioncross_contrastive

多层次跨模态对比损失函数 Args: rgb_feats: RGB特征 [B, N, D] sar_feats: SAR特征 [B, N, D] temp_*: 各层级的温度系数 alpha, beta, gamma:

models/clips.py:30

↓ 1 callersFunctionget_2d_sincos_pos_embed_from_grid

(embed_dim, grid)

utils/pos_embed.py:38

↓ 1 callersFunctionget_args_parser

()

main_pretrain.py:28

↓ 1 callersFunctionget_grad_norm_

(parameters, norm_type: float = 2.0)

utils/misc.py:283

↓ 1 callersFunctionget_model

(model_str, norm_pix_loss=True)

models/model_factory.py:4

↓ 1 callersFunctionget_rank

()

utils/misc.py:204

↓ 1 callersFunctionis_main_process

()

utils/misc.py:210

↓ 1 callersMethodlog_every

(self, iterable, print_freq, header=None)

utils/misc.py:126

↓ 1 callersFunctionmain

(args)

main_pretrain.py:90

↓ 1 callersFunctionmain

()

data/split_patch.py:80

↓ 1 callersMethodmax

(self)

utils/misc.py:73

↓ 1 callersFunctionsave_on_master

(*args, **kwargs)

utils/misc.py:214

↓ 1 callersMethodsynchronize_between_processes

(self)

utils/misc.py:119

↓ 1 callersFunctiontrain_one_epoch

(model: torch.nn.Module, data_loader: Iterable, optimizer: torch.optim.Optimizer,

engine_pretrain.py:14

(self, loss, optimizer, clip_grad=None, parameters=None, create_graph=False, update_grad=True)

(self, input_size=512, mask_patch_size=32, model_patch_size=4, mask_ratio=0.6)

mars_dataset.py:21

Method__init__

( self, dataset_root="./data", img_size=256, # 512,32 mask_patch_size=16, #32,

mars_dataset.py:38

Method__init__

(self, window_size=20, fmt=None)

utils/misc.py:32

Method__init__

(self, delimiter="\t")

(self, model_name='swinv2_base_window8_256', img_size=512, in_chans=3, pretrained=True)

models/MaRS.py:77

Method__init__

(self, encoder_rgb, encoder_sar, teacher_encoder, encoder_stride=32, contrast_type='none')

Functionadjust_learning_rate

Decay the learning rate with half-cycle cosine after warmup

utils/lr_sched.py:9

Functionall_reduce_mean

(self, rgb_feat, sar_feat)

(self, samples, mask_ratio)

models/MaRS.py:135

Functionget_2d_sincos_pos_embed

grid_size: int of the grid height and width return: pos_embed: [grid_size*grid_size, embed_dim] or [1+grid_size*grid_size, embed_dim] (w/

utils/pos_embed.py:20

Methodglobal_avg

(self)

utils/misc.py:69

Functioninit_distributed_mode

(args)

utils/misc.py:219

Functioninterpolate_pos_embed

(model, checkpoint_model)

utils/pos_embed.py:75

Functionload_model

(args, model_without_ddp, optimizer, loss_scaler)

Methodno_weight_decay

(self)

models/MaRS.py:101

Functionsave_model

(args, epoch, model, model_without_ddp, optimizer, loss_scaler)

utils/misc.py:298

Functionsplit_and_save

(sar_path, rgb_path, base_name)

data/split_patch.py:25

Methodsynchronize_between_processes

Warning: does not synchronize the deque!