|
--- |
|
tags: |
|
- pytorch_model_hub_mixin |
|
- model_hub_mixin |
|
datasets: |
|
- BrachioLab/cholecystectomy |
|
--- |
|
|
|
This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.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_gonogo") |
|
dataset = CholecDataset(split="test") |
|
dataloader = torch.utils.data.DataLoader(dataset, batch_size=4, shuffle=True) |
|
item = next(iter(dataloader)) |
|
out = model(item["image"]) |
|
``` |
