Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
Zezhong 83b1f51c1c | 1 year ago | |
---|---|---|
dataset | 1 year ago | |
models | 1 year ago | |
utils | 1 year ago | |
.gitignore | 1 year ago | |
KiUNet.py | 1 year ago | |
README.md | 1 year ago | |
config.py | 1 year ago | |
preprocess_LiTS.py | 1 year ago | |
test.py | 1 year ago | |
train.py | 1 year ago |
The repository is a 3DUNet implemented with pytorch, referring to
this project.
I have redesigned the code structure and used the model to perform liver and tumor segmentation on the lits2017 dataset.
paper: 3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation
pytorch >= 1.1.0
torchvision
SimpleITK
Tensorboard
Scipy
├── dataset # Training and testing dataset
│ ├── dataset_lits_train.py
│ ├── dataset_lits_val.py
│ ├── dataset_lits_test.py
│ └── transforms.py
├── models # Model design
│ ├── nn
│ │ └── module.py
│ │── ResUNet.py # 3DUNet class
│ │── Unet.py # 3DUNet class
│ │── SegNet.py # 3DUNet class
│ └── KiUNet.py # 3DUNet class
├── experiments # Trained model
|── utils # Some related tools
| ├── common.py
| ├── weights_init.py
| ├── logger.py
| ├── metrics.py
| └── loss.py
├── preprocess_LiTS.py # preprocessing for raw data
├── test.py # Test code
├── train.py # Standard training code
└── config.py # Configuration information for training and testing
raw_dataset:
├── test # 20 samples(27~46)
│ ├── ct
│ │ ├── volume-27.nii
│ │ ├── volume-28.nii
| | |—— ...
│ └── label
│ ├── segmentation-27.nii
│ ├── segmentation-28.nii
| |—— ...
│
├── train # 111 samples(0\~26 and 47\~131)
│ ├── ct
│ │ ├── volume-0.nii
│ │ ├── volume-1.nii
| | |—— ...
│ └── label
│ ├── segmentation-0.nii
│ ├── segmentation-1.nii
| |—— ...
./preprocess_LiTS.py
, such as: row_dataset_path = './raw_dataset/train/' # path of origin dataset
fixed_dataset_path = './fixed_data/' # path of fixed(preprocessed) dataset
python ./preprocess_LiTS.py
./fixed_data
│—— train_path_list.txt
│—— val_path_list.txt
│
|—— ct
│ volume-0.nii
│ volume-1.nii
│ volume-2.nii
│ ...
└─ label
segmentation-0.nii
segmentation-1.nii
segmentation-2.nii
...
config.py
,especially, please set --dataset_path
to ./fixed_data
config.py
.python train.py --save model_name
tensorboard --logdir ./output/model_name
.run test.py
Please pay attention to path of trained model in test.py
.
(Since the calculation of the 3D convolution operation is too large,
I use a sliding window to block the input tensor before prediction, and then stitch the results to get the final result.
The size of the sliding window can be set by yourself in config.py
)
After the test, you can get the test results in the corresponding folder:./experiments/model_name/result
You can also read my Chinese introduction about this 3DUNet project here. However, I no longer update the blog, I will try my best to update the github code.
If you have any suggestions or questions,
welcome to open an issue to communicate with me.
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》