We used CosMos 2022 Challenge Dataset

please download the dataset through this

Train dataset: https://zenodo.org/record/6481870#.Y7aqaXZBzdk

Test dataset: https://zenodo.org/record/6843257#.Y7aqSHZBzdk

1. Download the data set from the link above and preprocess the dataset

2. Extract the center point from the given label

3. Make a learning plan using Totalloss at loss.py.

python main.py --mode total --model Unet3D --fold 0 

If you find this repo useful for your research, please consider citing our paper:

@article{
  title={},
  author={},
  journal={},
  pages={},
  year={2024},
  publisher={}
}