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 

model weight: google drive