Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/RyanWangZf/MedCLIP
/ types & classes
Types & classes
29 in github.com/RyanWangZf/MedCLIP
⨍
Functions
110
◇
Types & classes
29
↓ 1 callers
Class
BasicLayer
A basic Swin Transformer layer for one stage. Args: dim (int): Number of input channels. input_resolution (tuple[int]): Input res
medclip/vision_model.py:335
↓ 1 callers
Class
Evaluator
do evaluation on chexpert5x200 zero-shot classification
medclip/evaluator.py:15
↓ 1 callers
Class
ImageTextContrastiveCollator
medclip/dataset.py:320
↓ 1 callers
Class
ImageTextContrastiveDataset
medclip/dataset.py:156
↓ 1 callers
Class
ImageTextContrastiveLoss
medclip/losses.py:7
↓ 1 callers
Class
MedCLIPFeatureExtractor
medclip/dataset.py:30
↓ 1 callers
Class
MedCLIPModel
medclip/modeling_medclip.py:129
↓ 1 callers
Class
MedCLIPTextModel
medclip/modeling_medclip.py:15
↓ 1 callers
Class
Mlp
medclip/vision_model.py:14
↓ 1 callers
Class
PartiallyFixedEmbedding
medclip/modeling_medclip.py:335
↓ 1 callers
Class
PatchEmbed
r""" Image to Patch Embedding Args: img_size (int): Image size. Default: 256. patch_size (int): Patch token size. Default: 4.
medclip/vision_model.py:405
↓ 1 callers
Class
PromptClassifier
take MedCLIP model with prompts for zero-shot classification
medclip/modeling_medclip.py:242
↓ 1 callers
Class
SwinTransformerBlock
r""" Swin Transformer Block. Args: dim (int): Number of input channels. input_resolution (tuple[int]): Input resulotion. n
medclip/vision_model.py:160
↓ 1 callers
Class
Trainer
trainer for single-gpu training.
medclip/trainer.py:20
↓ 1 callers
Class
Uwinformer
r""" UWin Transformer A PyTorch impl of : `Swin Transformer: Hierarchical Vision Transformer using Shifted Windows` - https://arxiv
medclip/vision_model.py:451
↓ 1 callers
Class
WindowAttention
r""" Window based multi-head self attention (W-MSA) module with relative position bias. It supports both of shifted and non-shifted window. Ar
medclip/vision_model.py:63
↓ 1 callers
Class
ZeroShotImageCollator
medclip/dataset.py:418
↓ 1 callers
Class
ZeroShotImageDataset
medclip/dataset.py:365
Class
ImageSuperviseLoss
medclip/losses.py:90
Class
MedCLIPProcessor
A processor that takes input images and texts and provides inputs for `MedCLIPModel`.
medclip/dataset.py:143
Class
MedCLIPVisionModel
take resnet50 as backbone.
medclip/modeling_medclip.py:45
Class
MedCLIPVisionModelViT
take an VIT model as the backbone.
medclip/modeling_medclip.py:84
Class
PatchMerging
r""" Patch Merging Layer. Args: input_resolution (tuple[int]): Resolution of input feature. dim (int): Number of input channels.
medclip/vision_model.py:286
Class
PromptTuningClassifier
take MedCLIP model with prompt tuning
medclip/modeling_medclip.py:360
Class
PromptTuningImageCollator
medclip/dataset.py:587
Class
PromptTuningImageDataset
medclip/dataset.py:532
Class
SuperviseClassifier
take MedCLIP model with linear heads for supervised classification on images.
medclip/modeling_medclip.py:282
Class
SuperviseImageCollator
medclip/dataset.py:507
Class
SuperviseImageDataset
medclip/dataset.py:454