Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
yezhy@pcl.ac.cn
8b6645c85e
|
5 days ago | |
|---|---|---|
| networks | 5 days ago | |
| scripts | 2 months ago | |
| utils | 5 days ago | |
| .DS_Store | 2 months ago | |
| README.md | 5 days ago | |
| dataset.py | 2 months ago | |
| inference.py | 5 days ago | |
| loss.py | 2 months ago | |
| main_training.py | 2 months ago | |
| models.py | 2 months ago | |
| requirements.txt | 5 days ago | |
| test.ipynb | 5 days ago | |
| trainer.py | 2 months ago | |
README.md
CardiacSeg
The official implementation of paper "Customizing Pre-Training Volumetric Transformers with Scaling Pyramid for 3D Cardiac Segmentation".
Getting Started
1. Clone the Project & Install Requirements
git clone git@openi.pcl.ac.cn:OpenMedIA/CardiacSeg.git
pip install -r requirements.txt
2. Training
cd CardiacSeg
sh scripts/train_imagechd.sh
3. Inference
python inference.py --ckpt_url {model_path} --in_file {data_path} --out_file {output_path}
Our CardiacSeg model trained on the ImageCHD dataset can be downloaded here.
The official implementation of paper "Customizing Pre-Training Volumetric Transformers with Scaling Pyramid for 3D Cardiac Segmentation".
Python Jupyter Notebook Text Shell other
