|
|
--- |
|
|
license: apache-2.0 |
|
|
base_model: |
|
|
- paige-ai/Virchow2 |
|
|
--- |
|
|
|
|
|
### Complete Codebase |
|
|
Code for loading PulmoFoundation can be found in https://github.com/dddavid4real/PulmoFoundation. |
|
|
|
|
|
### Download Model Checkpoint |
|
|
Download PulmoFoundation-E2.pth from *Files and versions* to models/ckpts/ |
|
|
|
|
|
### Basic Usage |
|
|
|
|
|
```python |
|
|
from models import get_model, get_transform |
|
|
from PIL import Image |
|
|
|
|
|
# Load model and preprocessing pipeline |
|
|
model = get_model('cuda', 'models/ckpts/PulmoFoundation-E2.pth') |
|
|
transform = get_transform() |
|
|
|
|
|
# Load and preprocess image |
|
|
img = Image.open('path/to/your/image.jpg') # Prefer 512x512 patches at 40X |
|
|
img_tensor = transform(img) |
|
|
img_tensor = img_tensor.unsqueeze(0) # Add batch dimension [1, 3, H, W] |
|
|
img_tensor = img_tensor.cuda() # Move to GPU |
|
|
|
|
|
# Extract features |
|
|
features = model(img_tensor) # Shape: [1, 2560] |
|
|
print(f"Feature shape: {features.shape}") |
|
|
print(f"Feature vector: {features}") |
|
|
``` |
|
|
|
