This repository is written for the "CVPR 2025: Foundation Models for Interactive 3D Biomedical Image Segmentation"(link) challenge. It is based on MONAI 1.4. Many of the functions in the main VISTA3D repository are moved to MONAI 1.4 and this simplified folder will directly use components from MONAI.
It is simplified to train interactive segmentation models across different modalities. The sophisticated transforms and recipes used for VISTA3D are removed. The finetuned VISTA3D checkpoint on the challenge subsets is available here
pip install -r requirements.txt
Download the challenge subsets finetuned checkpoint or VISTA3D original checkpoint. Generate a json list that contains your traning data and update the json file path in the script.
torchrun --nnodes=1 --nproc_per_node=8 train_cvpr.py
You can directly download the docker file for the challenge baseline. We provide a Dockerfile to satisfy the challenge format. For more details, refer to the challenge website.
docker build -t vista3d:latest .
docker save -o vista3d.tar.gz vista3d:latest