MCPcopy Create free account

hub / github.com/ZJUDataIntelligence/Ultrasound-CLIP / functions

Functions28 in github.com/ZJUDataIntelligence/Ultrasound-CLIP

↓ 2 callersMethodcalculate_label_similarity
(self, tags1, tags2, task_id)
similarity_processor.py:96
↓ 2 callersMethodencode_text
(self, texts, normalize=True)
enhanced_clip_model.py:115
↓ 2 callersMethodget_tags_for_image
(self, image_key)
similarity_processor.py:76
↓ 1 callersMethod__init__
(self, base_clip_model, graph_encoder, freeze_graph_encoder=False)
enhanced_clip_model.py:53
↓ 1 callersMethod_ensure_convs
(self, g: dgl.DGLHeteroGraph, device)
graph_encoder.py:39
↓ 1 callersMethod_ensure_type_embeddings
(self, g: dgl.DGLHeteroGraph, device)
graph_encoder.py:31
↓ 1 callersMethod_infer_batch_size
(self, g: dgl.DGLHeteroGraph)
graph_encoder.py:49
↓ 1 callersMethod_safe_mean_nodes
(self, g: dgl.DGLHeteroGraph, ntype: str)
graph_encoder.py:60
↓ 1 callersFunctionbuild_single_sample_graph
(full_data_dict, image_key)
graph_builder.py:5
↓ 1 callersFunctionbuild_vocabs
()
tag_vocab.py:22
↓ 1 callersMethodcompute_predicted_similarity_matrix
(self, image_features, text_features)
semantic_loss.py:17
↓ 1 callersMethodencode_image
(self, images, normalize=True)
enhanced_clip_model.py:112
↓ 1 callersFunctionload_matrix_from_upper_triangle
(filename)
similarity_processor.py:8
↓ 1 callersMethodsemantic_matching_loss
(self, predicted_sim, target_sim)
semantic_loss.py:29
Method__init__
(self, out_dim: int, args=None, hidden: int = 128, n_layers: int = 2, dropout: float = 0.1)
graph_encoder.py:12
Method__init__
(self, args, similarity_weight=1.0, temperature=0.07)
semantic_loss.py:9
Method__init__
(self, similarity_matrices_dir, full_data_file=None)
similarity_processor.py:17
Method__init__
(self, text_dim, graph_dim, hidden_dim=None)
enhanced_clip_model.py:9
Functionbuild_hetero_graph_from_data
(full_data, image_keys)
graph_builder.py:103
Methodcalculate_batch_similarity_matrix
(self, batch_indices, dataset)
similarity_processor.py:121
Methodcalculate_batch_similarity_matrix_from_paths
(self, image_keys, batch_size)
similarity_processor.py:162
Methodencode_text_with_graph
(self, texts,graphs, normalize=True)
enhanced_clip_model.py:118
Methodforward
(self, g: dgl.DGLHeteroGraph)
graph_encoder.py:69
Methodforward
(self, prediction, similarity_matrix)
semantic_loss.py:52
Methodforward
(self, text_features, graph_features)
enhanced_clip_model.py:24
Methodforward
(self, images, texts, graphs=None)
enhanced_clip_model.py:82
Methodlogit_scale
(self)
enhanced_clip_model.py:79
Methodminmax_normalize
(self, matrix)
similarity_processor.py:202