metadata
tags:
- pytorch_model_hub_mixin
- model_hub_mixin
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.cholec import CholecDataset, CholecModel
model = CholecModel.from_pretrained("BrachioLab/cholecystectomy_organs")
dataset = CholecDataset(split="test")
dataloader = torch.utils.data.DataLoader(dataset, batch_size=4, shuffle=True)
item = next(iter(dataloader))
out = model(item["image"])