Views
No views yet
1from PIL import Image
2import torch
3from transformers import AutoImageProcessor, AutoModel
4
5
6# Load an image
7image = Image.open(
8 requests.get(
9 "https://github.com/owkin/HistoSSLscaling/blob/main/assets/example.tif?raw=true",
10 stream=True
11 ).raw
12)
13
14# Load phikon-v2
15processor = AutoImageProcessor.from_pretrained("owkin/phikon-v2")
16model = AutoModel.from_pretrained("owkin/phikon-v2")
17model.eval()
18
19# Process the image
20inputs = processor(image, return_tensors="pt")
21
22# Get the features
23with torch.inference_mode():
24 outputs = model(**inputs)
25 features = outputs.last_hidden_state[:, 0, :] # (1, 1024) shape
26
27assert features.shape == (1, 1024)alexandre.filiot@owkin.com).1@misc{filiot2024phikonv2largepublicfeature,
2 title={Phikon-v2, A large and public feature extractor for biomarker prediction},
3 author={Alexandre Filiot and Paul Jacob and Alice Mac Kain and Charlie Saillard},
4 year={2024},
5 eprint={2409.09173},
6 archivePrefix={arXiv},
7 primaryClass={eess.IV},
8 url={https://arxiv.org/abs/2409.09173},
9}