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
This is a PyTorch Lightning .ckpt
checkpoint for a ResNeXt101 model trained on chest CT images.
A quickstart script is below.
import radfusion3
dm = radfusion3.data.DataModule(config, test_split=config.test_split)
model = radfusion3.builder.build_lightning_model(config, ckpt=ckpt)
trainer.fit(model=model, datamodule=dm, ckpt_path=config.ckpt)
trainer.test(datamodule=dm, ckpt_path="best")
For detailed instructions please follow the README in Github repo.