Functions — github.com/OmkarThawakar/composed-video-retrieval

hub / github.com/OmkarThawakar/composed-video-retrieval / functions

Functions232 in github.com/OmkarThawakar/composed-video-retrieval

⨍ Functions 232◇ Types & classes 52

↓ 26 callersFunction_n2p

(w, t=True)

src/model/vit.py:288

↓ 13 callersMethod__init__

(self, config, layer_num)

src/model/med.py:372

↓ 10 callersFunctionprint_dist

Function to print a message only on device 0 in a distributed training setup. Args: message (str): The message to be printed.

src/tools/utils.py:22

↓ 9 callersFunctionjson_dump

(data, json_pth: Union[Path, str])

src/tools/files.py:14

↓ 8 callersFunctionid2int

(data, sub="")

src/data/utils.py:29

↓ 7 callersMethodtranspose_for_scores

(self, x)

src/model/med.py:156

↓ 6 callersFunctioncreate_vit

( vit, image_size, use_grad_checkpointing=False, ckpt_layer=0, drop_path_rate=0 )

src/model/blip.py:246

↓ 6 callersFunctioninit_tokenizer

()

src/model/blip.py:233

↓ 6 callersFunctionload_checkpoint

(model, url_or_filename)

src/model/blip.py:284

↓ 5 callersFunctionpre_caption

(caption, max_words=50)

src/data/utils.py:7

↓ 4 callersFunctionenhance_level_to_args

(MAX_LEVEL)

src/data/randaugment.py:217

↓ 4 callersFunctionwrite_txt

(data, txt_pth: Union[Path, str])

src/tools/files.py:27

↓ 3 callersMethod__init__

( self, in_features, hidden_features=None, out_features=None, act_laye

src/model/vit.py:24

↓ 3 callersFunctionblip_embs

(pretrained="", **kwargs)

src/model/blip_embs.py:47

↓ 3 callersFunctioneval_recall

(scores_q2t)

src/test/webvid_covr.py:116

↓ 3 callersMethodtest_dataloader

(self)

src/data/cirr.py:107

↓ 2 callersMethod__init__

( self, transform, annotation: str, vid_dir: str, emb_dir: str,

src/data/webvid_covr.py:157

↓ 2 callersMethod__init__

( self, transform, annotation: str, img_dir: str, emb_dir: str,

src/data/cirr.py:119

↓ 2 callersMethod__init__

( self, transform, annotation: str, targets: str, img_dir: str,

src/data/fashioniq.py:119

↓ 2 callersFunctionconcat_h_imgs

(im_list, resample=Image.Resampling.BICUBIC)

src/tools/visualize.py:66

↓ 2 callersFunctioninterpolate_pos_embed

(pos_embed_checkpoint, visual_encoder)

src/model/vit.py:397

↓ 2 callersFunctionjson_dump

(data, json_pth: Union[Path, str])

tools/scripts/merge_fiq_annotations.py:16

↓ 2 callersFunctionjson_load

(json_pth: Union[Path, str])

tools/scripts/merge_fiq_annotations.py:8

↓ 2 callersFunctionmain

(args)

tools/embs/save_blip_embs_imgs.py:53

↓ 2 callersFunctionremove_non_digits

(string, sub: str = "")

src/data/utils.py:36

↓ 2 callersFunctionsample_frames

(vlen, n_frames=15)

src/tools/visualize.py:54

↓ 2 callersFunctionshear_level_to_args

(MAX_LEVEL, replace_value)

src/data/randaugment.py:224

↓ 2 callersFunctiontranslate_level_to_args

(translate_const, MAX_LEVEL, replace_value)

src/data/randaugment.py:234

↓ 2 callersFunctiontune_channel

(ch)

src/data/randaugment.py:18

↓ 1 callersMethod__init__

Args: alpha: rescaling factor for positiver terms beta: concentration parameter Note: alpha = 1

src/model/loss.py:33

↓ 1 callersMethod__init__

Args: med_config (str): path for the mixture of encoder-decoder model's configuration file image_size (int): input im

src/model/blip.py:26

↓ 1 callersMethod__init__

( self, transform, annotation: str, vid_dir: str, emb_dir: str,

src/data/webvid_covr_rulebased.py:101

↓ 1 callersFunction_load_weights

Load weights from .npz checkpoints for official Google Brain Flax implementation

src/model/vit.py:284

↓ 1 callersFunctionadd_different_words

(df)

src/data/webvid_covr_rulebased.py:287

↓ 1 callersFunctioncalculate_model_params

(model)

src/tools/utils.py:5

↓ 1 callersFunctioneval_recall

(scores_q2t)

src/test/utils.py:109

↓ 1 callersFunctionextract_frames

(url, n_frames=10)

src/tools/visualize.py:83

↓ 1 callersMethodgenerate_rule_based_edit

(txt1, txt2)

src/data/webvid_covr_rulebased.py:257

↓ 1 callersFunctionget_blip_config

(model="base")

tools/embs/save_blip_embs_imgs.py:20

↓ 1 callersFunctionget_blip_config

(model="base")

tools/embs/save_blip_embs_vids.py:20

↓ 1 callersFunctionget_blip_config

(model="base")

tools/embs/save_blip_embs_vids_multimodal.py:22

↓ 1 callersFunctionget_different_word_in_each_sentence

(sentence1, sentence2)

src/data/webvid_covr_rulebased.py:274

↓ 1 callersMethodget_extended_attention_mask

Makes broadcastable attention and causal masks so that future and masked tokens are ignored. Arguments: attention_mask (

src/model/med.py:672

↓ 1 callersMethodget_random_ops

(self)

src/data/randaugment.py:330

↓ 1 callersFunctionget_recalls_labels

( sims, query_lbls, target_lbls, ks: List[int] = [1, 5, 10, 50] )

src/test/fashioniq.py:140

↓ 1 callersFunctionget_urls

(split)

tools/scripts/download_covr.py:21

↓ 1 callersFunctionget_video_frames

(video_pth, frames_video=15)

src/tools/visualize.py:8

↓ 1 callersFunctionget_video_frames

(video_pth, frames_video=15)

src/data/embs.py:147

↓ 1 callersMethodget_video_frames

( self, video_pth: str, start_time: float = 0.0, end_time: Union[float, None]

src/data/utils.py:160

↓ 1 callersFunctionis_url

(url_or_filename)

src/tools/visualize.py:136

↓ 1 callersFunctionis_url

(url_or_filename)

src/model/blip.py:279

↓ 1 callersFunctionjson_load

(json_pth: Union[Path, str])

src/tools/files.py:6

↓ 1 callersFunctionmain

(cfg: DictConfig)

train.py:16

↓ 1 callersFunctionmain

(cfg: DictConfig)

test.py:11

↓ 1 callersFunctionmain

(args)

tools/embs/save_blip_embs_vids.py:55

↓ 1 callersFunctionmain

(args)

tools/embs/save_blip_embs_vids_multimodal.py:57

↓ 1 callersFunctionmain

(args)

tools/scripts/download_covr.py:41

↓ 1 callersFunctionmain

(ann_dir: str)

tools/scripts/merge_fiq_annotations.py:23

↓ 1 callersFunctionmean_results

(dir=".", fabric=None, save=True)

src/test/fashioniq.py:146

↓ 1 callersFunctionparse_args

()

tools/scripts/webvid_descriptions_2m.py:44

↓ 1 callersFunctionparse_args

()

tools/scripts/webvid_descriptions_8m.py:44

↓ 1 callersFunctionposterize_level_to_args

(MAX_LEVEL)

src/data/randaugment.py:264

↓ 1 callersMethodprune_heads

(self, heads)

src/model/med.py:293

↓ 1 callersFunctionread_txt

(txt_pth: Union[Path, str])

src/tools/files.py:21

↓ 1 callersFunctionrecall_at_k_labels

(sim, query_lbls, target_lbls, k=10)

src/test/fashioniq.py:124

↓ 1 callersFunctionremove_module

(state_dict)

src/model/blip.py:315

↓ 1 callersFunctionrotate_level_to_args

(MAX_LEVEL, replace_value)

src/data/randaugment.py:272

↓ 1 callersFunctionsample_frames

(frames_videos, vlen)

src/data/utils.py:122

↓ 1 callersFunctionsample_frames

(vlen, frames_per_video=15)

src/data/embs.py:190

↓ 1 callersMethodsave_attention_map

(self, attention_map)

src/model/vit.py:77

↓ 1 callersMethodsave_attention_map

(self, attention_map)

src/model/med.py:150

↓ 1 callersFunctionsolarize_level_to_args

(MAX_LEVEL)

src/data/randaugment.py:252

↓ 1 callersFunctiontrain

(model, train_loader, optimizer, fabric, epoch, cfg)

train.py:99

↓ 1 callersMethodtrain_dataloader

(self)

src/data/cirr.py:57

↓ 1 callersMethodval_dataloader

(self)

src/data/cirr.py:67

↓ 1 callersFunctionvisualize_pth_video

(video_pth, n_frames=10)

src/tools/visualize.py:123

↓ 1 callersFunctionvisualize_url_video

(url, n_frames=10)

src/tools/visualize.py:116

Method__call__

Decay the learning rate

src/tools/scheduler.py:11

Method__call__

(self, optimizer, epoch)

src/tools/scheduler.py:26

Method__call__

(self, img)

src/data/transforms.py:43

Method__call__

(self, img)

src/data/transforms.py:60

Method__call__

(self, video_pth: str)

src/data/utils.py:153

Method__call__

(self, img)

src/data/randaugment.py:334

Method__call__

(self, model, data_loader, fabric)

src/test/webvid_covr_exp.py:18

Method__call__

(model, data_loader, fabric)

src/test/webvid_covr_exp.py:113

Method__call__

(self, model, data_loader, fabric)

src/test/webvid_covr.py:18

Method__call__

(model, data_loader, fabric)

src/test/cirr.py:20

Method__call__

(self, model, data_loader, fabric)

src/test/fashioniq.py:22

Method__getitem__

(self, index)

src/data/webvid_covr.py:300

Method__getitem__

(self, index)

src/data/webvid_covr_rulebased.py:227

src/data/fashioniq.py:184

Method__init__

(self, min_lr, init_lr, decay_rate, max_epochs)

src/tools/scheduler.py:5

Method__init__

(self, min_lr, init_lr, decay_rate)

src/tools/scheduler.py:21

( self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_

src/model/vit.py:50

Method__init__

( self, dim, num_heads, mlp_ratio=4.0, qkv_bias=False, qk_scal

src/model/vit.py:111

next →1–100 of 232, ranked by callers