forked from TripleCoenzyme/ResNet50-Unet
zengchan/ResNet50-Unet
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
read_img.py用于将数据集中的nii.gz文件存储为tif图片格式 cut_img.py用于裁剪tif格式图片生成训练集 save_img.py用于将测试集的结果存储为nii.gz文件格式 utils.py中包含网络训练、测试中所需要的一些辅助函数 model.py为网络结构程序 dataset.py为数据集构建程序 main.py为网络训练用程序,验证过程也包含在其中 test.py为网络测试用程序 final_test.py为比赛现场测试脚本,从给定的nii文件直接生成结果nii文件 The u-net uses the first 4 layers of ResNet50 for the downsampling part and replace the transposed convolution with Pixel Shuffle in the upsampling part.
About
Segmentation for vertebra in MR images
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 100.0%