Views
No views yet
pip install histolytics
pip install albumentations1from histolytics.models.hovernet_panoptic import HoverNetPanoptic
2
3model = HoverNetPanoptic.from_pretrained("cin2_v1_efficientnet_b5")1from albumentations import Resize, Compose
2from histolytics.utils import FileHandler
3from histolytics.transforms import MinMaxNormalization
4
5model.set_inference_mode()
6
7# Resize to multiple of 32 of your own choosing
8transform = Compose([Resize(1024, 1024), MinMaxNormalization()])
9
10im = FileHandler.read_img(IMG_PATH)
11im = transform(image=im)["image"]
12
13prob = model.predict(im)
14out = model.post_process(prob)
15# out = {"nuc": [(nuc instances (H, W), nuc types (H, W))], "tissue": [tissues (H, W)], "cyto": None}1import torch
2from histolytics.utils import FileHandler
3
4model.set_inference_mode()
5
6# dont use random matrices IRL
7batch = torch.rand(8, 3, 1024, 1024)
8
9prob = model.predict(im)
10out = model.post_process(prob)
11# out = {
12# "nuc": [
13# (nuc instances (H, W), nuc types (H, W)),
14# (nuc instances (H, W), nuc types (H, W)),
15# .
16# .
17# .
18# (nuc instances (H, W), nuc types (H, W))
19# ],
20# "tissue": [
21# (nuc instances (H, W), nuc types (H, W)),
22# (nuc instances (H, W), nuc types (H, W)),
23# .
24# .
25# .
26# (nuc instances (H, W), nuc types (H, W))
27# ],
28# "cyto": None,
29#}1from matplotlib import pyplot as plt
2from skimage.color import label2rgb
3
4fig, ax = plt.subplots(1, 4, figsize=(24, 6))
5ax[0].imshow(im)
6ax[1].imshow(label2rgb(out["nuc"][0][0], bg_label=0)) # inst_map
7ax[2].imshow(label2rgb(out["nuc"][0][1], bg_label=0)) # type_map
8ax[3].imshow(label2rgb(out["tissue"][0], bg_label=0)) # tissue_map
nuc_classes = {
0: "background",
1: "neoplastic",
2: "inflammatory",
3: "connective",
4: "dead",
5: "glandular_epithelial",
6: "squamous_epithelial",
}
tissue_classes = {
0: "background",
1: "stroma",
2: "cin",
3: "squamous_epithelium",
4: "glandular_epithelium",
5: "slime",
6: "blood",
}@article{
}@article{Graham2019,
title = {Hover-Net: Simultaneous segmentation and classification of nuclei in multi-tissue histology images},
volume = {58},
ISSN = {1361-8415},
url = {http://dx.doi.org/10.1016/j.media.2019.101563},
DOI = {10.1016/j.media.2019.101563},
journal = {Medical Image Analysis},
publisher = {Elsevier BV},
author = {Graham, Simon and Vu, Quoc Dang and Raza, Shan E Ahmed and Azam, Ayesha and Tsang, Yee Wah and Kwak, Jin Tae and Rajpoot, Nasir},
year = {2019},
month = dec,
pages = {101563}
}