BrachioLab/chestx_pathols

This model has been pushed to the Hub using the PytorchModelHubMixin integration:

• Library: [More Information Needed]
• Docs: [More Information Needed]

Usage

import torch
from exlib.datasets.chestx import ChestXDataset, ChestXPathologyModel
model = ChestXPathologyModel.from_pretrained("BrachioLab/chestx_pathols")
dataset = ChestXDataset(split="test")
dataloader = torch.utils.data.DataLoader(dataset, batch_size=4, shuffle=True)
item = next(iter(dataloader))
out = model(item["image"])