Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ZJUDataIntelligence/Ultrasound-CLIP
/ functions
Functions
28 in github.com/ZJUDataIntelligence/Ultrasound-CLIP
⨍
Functions
28
◇
Types & classes
5
↓ 2 callers
Method
calculate_label_similarity
(self, tags1, tags2, task_id)
similarity_processor.py:96
↓ 2 callers
Method
encode_text
(self, texts, normalize=True)
enhanced_clip_model.py:115
↓ 2 callers
Method
get_tags_for_image
(self, image_key)
similarity_processor.py:76
↓ 1 callers
Method
__init__
(self, base_clip_model, graph_encoder, freeze_graph_encoder=False)
enhanced_clip_model.py:53
↓ 1 callers
Method
_ensure_convs
(self, g: dgl.DGLHeteroGraph, device)
graph_encoder.py:39
↓ 1 callers
Method
_ensure_type_embeddings
(self, g: dgl.DGLHeteroGraph, device)
graph_encoder.py:31
↓ 1 callers
Method
_infer_batch_size
(self, g: dgl.DGLHeteroGraph)
graph_encoder.py:49
↓ 1 callers
Method
_safe_mean_nodes
(self, g: dgl.DGLHeteroGraph, ntype: str)
graph_encoder.py:60
↓ 1 callers
Function
build_single_sample_graph
(full_data_dict, image_key)
graph_builder.py:5
↓ 1 callers
Function
build_vocabs
()
tag_vocab.py:22
↓ 1 callers
Method
compute_predicted_similarity_matrix
(self, image_features, text_features)
semantic_loss.py:17
↓ 1 callers
Method
encode_image
(self, images, normalize=True)
enhanced_clip_model.py:112
↓ 1 callers
Function
load_matrix_from_upper_triangle
(filename)
similarity_processor.py:8
↓ 1 callers
Method
semantic_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
Function
build_hetero_graph_from_data
(full_data, image_keys)
graph_builder.py:103
Method
calculate_batch_similarity_matrix
(self, batch_indices, dataset)
similarity_processor.py:121
Method
calculate_batch_similarity_matrix_from_paths
(self, image_keys, batch_size)
similarity_processor.py:162
Method
encode_text_with_graph
(self, texts,graphs, normalize=True)
enhanced_clip_model.py:118
Method
forward
(self, g: dgl.DGLHeteroGraph)
graph_encoder.py:69
Method
forward
(self, prediction, similarity_matrix)
semantic_loss.py:52
Method
forward
(self, text_features, graph_features)
enhanced_clip_model.py:24
Method
forward
(self, images, texts, graphs=None)
enhanced_clip_model.py:82
Method
logit_scale
(self)
enhanced_clip_model.py:79
Method
minmax_normalize
(self, matrix)
similarity_processor.py:202