MCPcopy Create free account

hub / github.com/alipay/Ant-Multi-Modal-Framework / functions

Functions4,004 in github.com/alipay/Ant-Multi-Modal-Framework

↓ 3 callersMethodforward_text_encoder
(self, caption_input=None, ocr_input=None)
prj/snps3_vtp/roi_univl/univl/model/univl_base.py:301
↓ 3 callersMethodforward_video
(self, inputs, mixed5c=False)
antmmf/models/s3dg.py:366
↓ 3 callersMethodfrom_pretrained
(cls, pretrained_model_name_or_path, *inputs, **kwargs)
prj/M2_Encoder/vlmo/tokenizer/tokenization_glm.py:299
↓ 3 callersMethodget_1d_position_embedding
(self, input_ids=None, position_ids=None)
antmmf/modules/embeddings/layout_lm_embeddings.py:95
↓ 3 callersMethodget_2d_position_embedding
(self, bbox)
antmmf/modules/embeddings/layout_lm_embeddings.py:74
↓ 3 callersFunctionget_bert_configured_parameters
(module, lr=None)
antmmf/utils/general.py:343
↓ 3 callersMethodget_input_embeddings
(self)
prj/M2_omni/models/modeling_m2omni.py:280
↓ 3 callersMethodget_mil_nce_loss
:param sim_matrix: #text(bsz) x #video(bsz*n_pair) :return:
prj/dmae_vtp/roi_univl/univl/model/univl_video_ret.py:156
↓ 3 callersMethodget_mil_nce_loss
:param sim_matrix: #text(bsz) x #video(bsz*n_pair) :return:
prj/base_vtp/roi_univl/univl/model/univl_video_ret.py:146
↓ 3 callersMethodget_mil_nce_loss
:param sim_matrix: #text(bsz) x #video(bsz*n_pair) :return:
prj/cnvid_vtp/roi_univl/univl/model/univl_video_ret.py:155
↓ 3 callersMethodget_model_class
(cls, name)
antmmf/common/registry.py:633
↓ 3 callersMethodget_optimizer_class
(cls, name)
antmmf/common/registry.py:649
↓ 3 callersMethodget_rope_index
Calculate the 3D rope index based on image and video's temporal, height and width in LLM. Explanation: Each embedding se
prj/M2_omni/models/modeling_m2omni.py:327
↓ 3 callersMethodget_simi_logits
:return: bsz_text, bsz_video, num_clips: clip-level score for training bsz_text, num_clips: clip-level score for val/
prj/cnvid_vtp/roi_univl/univl/model/univl_video_ret.py:260
↓ 3 callersMethodinfer_image
( self, batch, mask_image=False, image_token_type_idx=1,
prj/M2_Encoder/vlmo/modules/vlmo_module.py:359
↓ 3 callersMethodinfer_text
( self, batch, mask_text=False, )
prj/M2_Encoder/vlmo/modules/vlmo_module.py:323
↓ 3 callersMethodinit_processors
(self)
antmmf/datasets/base_dataset.py:48
↓ 3 callersMethodinit_weights
(self)
antmmf/models/visual_bert.py:207
↓ 3 callersMethodinspect
(self, func: Callable, pop_first: bool = False)
antmmf/utils/inspector.py:106
↓ 3 callersMethodis_frozen
Whether the configuration is frozen.
antmmf/common/configuration.py:493
↓ 3 callersMethodis_leaf
(self)
antmmf/modules/utils.py:308
↓ 3 callersFunctionjsonl_dump
Print a list of json dictionary in a file The file then contains information such as {xxx:yyy} {zzz:qqq} Parameters:
antmmf/utils/general.py:481
↓ 3 callersFunctionload_audio
(audio_file, sample_rate=16000)
prj/M2_omni/models/m2omni_utils.py:81
↓ 3 callersMethodload_optimizer
(self)
antmmf/trainers/base_trainer.py:373
↓ 3 callersMethodload_task
(self)
antmmf/trainers/base_trainer.py:220
↓ 3 callersFunctionmake_non_local
(net, n_segment)
antmmf/modules/vision/non_local.py:189
↓ 3 callersFunctionmake_temporal_shift
( net, n_segment=8, n_div=4, place="blockres", temporal_pool=False )
antmmf/modules/vision/temporal_shift.py:102
↓ 3 callersFunctionno_sync_module_apply
FSDP module .apply will use _unshard_params_recurse which will sync params across ranks. using this function when apply fn is unnecessary to sync
prj/M2_Encoder/vlmo/modules/vlmo_utils.py:6
↓ 3 callersMethodnormal_sample
(num_frames, num_segments, clip_length, stage)
antmmf/datasets/database/video_database.py:142
↓ 3 callersFunctionnullcontext
(enter_result=None)
antmmf/utils/general.py:522
↓ 3 callersMethodprepare_cross_text
(self, input_ids, input_mask)
prj/cnvid_vtp/roi_univl/univl/model/univl_video_base.py:168
↓ 3 callersFunctionprocessDigitArticle
(inText)
prj/Pink/pink/eval/eval_gqa.py:67
↓ 3 callersFunctionprocessPunctuation
(inText)
prj/Pink/pink/eval/eval_gqa.py:55
↓ 3 callersMethodprocess_text_embedding
( self, sample_list, embedding_attr="text_embeddings", info=None )
antmmf/models/ant_mmf.py:183
↓ 3 callersMethodprofile
(self, text)
antmmf/models/layoutlm.py:662
↓ 3 callersMethodprofile
(self, text)
antmmf/predictors/base_predictor.py:270
↓ 3 callersMethodreduce_clips
(self, simi_logits, level)
prj/snps3_vtp/roi_univl/univl/model/univl_video_ret.py:409
↓ 3 callersMethodreduce_clips
(self, simi_logits, level)
prj/dmae_vtp/roi_univl/univl/model/univl_video_ret.py:357
↓ 3 callersMethodresidual_connection
(self, x, residual)
prj/M2_Encoder/vlmo/torchscale/architecture/decoder.py:130
↓ 3 callersFunctionrotate_half
Rotates half the hidden dims of the input.
prj/M2_omni/models/qwen2_vit.py:55
↓ 3 callersFunctionround_by_factor
Returns the closest integer to 'number' that is divisible by 'factor'.
prj/M2_omni/models/m2omni_utils.py:46
↓ 3 callersFunctionsalt_pepper_noise
Add salt noise, pepper noise, or, s&p noise to img. @param img: ndarray Input image data on range[0,255]. @param amount : floa
antmmf/utils/dataset_utils.py:247
↓ 3 callersMethodsample_frames
( self, num_clips, vlen, sample="rand", num_frame=1, fix_start
antmmf/utils/video_utils.py:25
↓ 3 callersMethodsave_one
(self, image_path, feature, feature_info=None)
antmmf/datasets/features/vision/feature_saver.py:65
↓ 3 callersFunctionset_split_position
(position)
prj/M2_Encoder/vlmo/torchscale/component/multiway_network.py:16
↓ 3 callersFunctionsplit_encoder_output
:param encoder_embeds: (b, sq_length, hidden) :param position_range: :return:
prj/snps3_vtp/roi_univl/univl/model/univl_base.py:17
↓ 3 callersFunctionsplit_encoder_output
:param encoder_embeds: (b, sq_length, hidden) :param position_range: :return:
prj/dmae_vtp/roi_univl/univl/model/univl_base.py:17
↓ 3 callersFunctionsplit_encoder_output
:param encoder_embeds: (b, sq_length, hidden) :param position_range: :return:
prj/base_vtp/roi_univl/univl/model/univl_base.py:17
↓ 3 callersFunctionsplit_encoder_output
:param encoder_embeds: (b, sq_length, hidden) :param position_range: :return:
prj/cnvid_vtp/roi_univl/univl/model/univl_base.py:17
↓ 3 callersFunctiontokenize
(sentence, regex=SENTENCE_SPLIT_REGEX, keep=["'s"], remove=[",", "?"])
antmmf/utils/text_utils.py:285
↓ 3 callersMethodtrain
(self)
antmmf/trainers/base_trainer.py:473
↓ 3 callersMethodtrain
Override the default train() to allow freezing image encoder, text encoder, and classifier :return:
antmmf/models/concat_bert.py:169
↓ 3 callersMethodtranspose_for_scores
(self, x)
antmmf/modules/vision/backbone/clip/modeling_bert.py:126
↓ 3 callersMethodtranspose_for_scores
(self, x)
antmmf/models/vilbert.py:47
↓ 3 callersMethodtranspose_for_scores
(self, x)
antmmf/models/vilbert.py:151
↓ 3 callersFunctiontrunc_normal_
(tensor, mean=0.0, std=1.0, a=-2.0, b=2.0)
antmmf/modules/vision/backbone/cctt.py:22
↓ 3 callersMethodupdate
Update the config in an inplace way. Usage: .. code-block:: python config: Configuration = ... upd
antmmf/common/configuration.py:367
↓ 3 callersMethodupdate_registry_for_model
Use this if there is some specific configuration required by model which must be inferred at runtime.
antmmf/tasks/base_task.py:177
↓ 3 callersFunctionwhitespace_tokenize
Runs basic whitespace cleaning and splitting on a piece of text.
antmmf/modules/vision/backbone/clip/cn_tokenizer.py:172
↓ 3 callersFunctionwrap_to_configuration
(val)
antmmf/common/configuration.py:143
↓ 3 callersFunctionxavier_normal
(tensor, gain=1.0)
antmmf/utils/init.py:46
↓ 2 callersMethodVWM_feature_rearrange
(self, raw_cap_embed, IW_word_idx_list, bsz)
prj/snps3_vtp/roi_univl/univl/model/univl_video_ret.py:588
↓ 2 callersMethodVWM_forward_stage
(self, word_embed_list, video_embed, output_dict, num_clips=1, cal_cross=True, stage="after")
prj/snps3_vtp/roi_univl/univl/model/univl_video_ret.py:51
↓ 2 callersMethod__init__
( self, input_size: int, output_size: int = 256, attention_heads: int = 4,
prj/M2_omni/models/sanm_audio.py:222
↓ 2 callersMethod__init__
( self, imgdir, annotation_db, transform=None, loader=ExifImageLoader.
prj/snps3_vtp/roi_univl/roi/dataset.py:80
↓ 2 callersMethod__init__
(self, config)
prj/snps3_vtp/roi_univl/univl/model/clip_text_encoder.py:64
↓ 2 callersMethod__init__
(self, annotation_path, anno_type="normal", **kwargs)
prj/snps3_vtp/roi_univl/univl/pretrain_video_text/dataset.py:106
↓ 2 callersMethod__init__
( self, datasets = None, probabilities: Optional[List[float]] = None,
prj/Pink/pink/datasets/ProbMergeDataset.py:11
↓ 2 callersMethod__init__
( self, imgdir, annotation_db, transform=None, loader=ExifImageLoader.
prj/dmae_vtp/roi_univl/roi/dataset.py:80
↓ 2 callersMethod__init__
(self, config)
prj/dmae_vtp/roi_univl/univl/model/clip_text_encoder.py:64
↓ 2 callersMethod__init__
(self, num_frames: int, num_tokens: int, embed_dim: int, qk_bias: bool = False, qdim: int = N
prj/dmae_vtp/roi_univl/univl/model/tpmcl_utils.py:9
↓ 2 callersMethod__init__
(self, annotation_path, anno_type="normal", **kwargs)
prj/dmae_vtp/roi_univl/univl/pretrain_video_text/dataset.py:106
↓ 2 callersMethod__init__
( self, imgdir, annotation_db, transform=None, loader=ExifImageLoader.
prj/base_vtp/roi_univl/roi/dataset.py:80
↓ 2 callersMethod__init__
(self, config)
prj/base_vtp/roi_univl/univl/model/clip_text_encoder.py:64
↓ 2 callersMethod__init__
(self, annotation_path, anno_type="normal", **kwargs)
prj/base_vtp/roi_univl/univl/pretrain_video_text/dataset.py:106
↓ 2 callersMethod__init__
( self, imgdir, annotation_db, transform=None, loader=ExifImageLoader.
prj/cnvid_vtp/roi_univl/roi/dataset.py:80
↓ 2 callersMethod__init__
(self, annotation_path, anno_type="normal", **kwargs)
prj/cnvid_vtp/roi_univl/univl/pretrain_video_text/dataset.py:106
↓ 2 callersMethod__init__
(self, config)
antmmf/modules/fusions.py:78
↓ 2 callersMethod__init__
(self, *args, **kwargs)
antmmf/modules/metrics/roc_auc.py:23
↓ 2 callersMethod__init__
( self, in_dim: int, out_dim: int, weight_norm_dim: int = -1, )
antmmf/modules/layers/linear.py:11
↓ 2 callersMethod__init__
( self, embed_dim: int, # vision image_resolution: int, vision_layers:
antmmf/modules/vision/backbone/clip/cn_model.py:126
↓ 2 callersMethod__init__
(self, vocab_file, data_root_dir=None)
antmmf/utils/text_utils.py:324
↓ 2 callersMethod__init__
( self, name: str, dataset_type: str, config: Union[Configuration, Dict] )
antmmf/datasets/base_dataset.py:29
↓ 2 callersMethod__init__
(self, config)
antmmf/models/ant_mmf.py:21
↓ 2 callersMethod__init__
(self, config, model)
antmmf/models/nlp_adversarial.py:30
↓ 2 callersMethod__set_size__
(self, size: List[Optional[int]], dim: int, src: Tensor)
antmmf/modules/message_passing/message_passing.py:101
↓ 2 callersMethod__setattr__
(self, key, value)
antmmf/common/configuration.py:580
↓ 2 callersMethod__setitem__
(self, key, value)
antmmf/utils/general.py:322
↓ 2 callersMethod_adv_build_noisy_sample
(self, current_sample_list)
antmmf/models/mm_adversarial.py:212
↓ 2 callersMethod_adversarial_data_generation
(self, dataset_type)
antmmf/trainers/adv_trainer.py:179
↓ 2 callersMethod_agg_visual_feat
(self, visual_output, video_mask, sim_header="meanP")
prj/dmae_vtp/roi_univl/univl/model/dmae_utils.py:186
↓ 2 callersMethod_aggregate_token_embeddings
( embeddings_list: List[torch.Tensor], )
antmmf/modules/interpret/saliency_interpreter.py:92
↓ 2 callersMethod_build
(self)
antmmf/utils/text_utils.py:336
↓ 2 callersMethod_build_loader_list
(self)
antmmf/trainers/base_trainer.py:141
↓ 2 callersMethod_build_sample
Args: data: image bytes json_obj: one json object include multimodal info Returns: sample(antmm
antmmf/predictors/base_predictor.py:211
↓ 2 callersMethod_cal_recall
(self, ind)
antmmf/modules/metrics/mm_retrieval_recall.py:74
↓ 2 callersMethod_calculate
(self, output, expected)
antmmf/modules/metrics/roc_auc.py:32
↓ 2 callersMethod_calculate
(self, sample_list, model_output, *args, **kwargs)
antmmf/modules/metrics/span_f1.py:30
↓ 2 callersMethod_calculate
(self, ranks)
antmmf/modules/metrics/recall_at_k.py:80
↓ 2 callersMethod_calculate
(self, output, expected)
antmmf/modules/metrics/ks.py:27
← previousnext →301–400 of 4,004, ranked by callers