Image Models
Collection
This repo contains medical imaging models for tte-pretraining paper (https://arxiv.org/abs/2411.09361) and INSPECT (https://arxiv.org/abs/2311.10798)
•
7 items
•
Updated
DenseNet Checkpoint
This is a PyTorch Lightning .ckpt checkpoint for a DenseNet model trained on chest CT images with TTE objective.
Usage
A quickstart script is below.
import torch
from src.networks import DenseNet121
model = DenseNet121(spatial_dims=3, in_channels=1, out_channels=2).to(device)
state_dict = torch.load(
loadmodel_path, map_location=f"cuda:{torch.cuda.current_device()}"
)
model.load_state_dict(state_dict)
For detailed instructions please follow the README in Github repo.
- Downloads last month
- 3
Inference Providers
NEW
This model isn't deployed by any Inference Provider.
🙋
Ask for provider support