@@ -97,12 +97,17 @@ We transferred the above pre-trained models to the multi-class segmentation task
97
97
```
98
98
MedicalNet is used to transfer the pre-trained model to other datasets (here the MRBrainS18 dataset is used as an example).
99
99
MedicalNet/
100
- |--datasets/:Data preprocessing module
101
- | |--brains18.py:MRBrainS18 data preprocessing script
102
- | |--models/:Model construction module
103
- | |--resnet.py:3D-ResNet network build script
104
- |--utils/:tools
105
- | |--logger.py:Logging script
100
+ |--med3d
101
+ | |--datasets/:Data preprocessing module
102
+ | | |--brains18.py:MRBrainS18 data preprocessing script
103
+ | |--models/:Model construction module
104
+ | | |--resnet.py:3D-ResNet network build script
105
+ | |--utils/:tools
106
+ | | |--logger.py:Logging script
107
+ | |--model.py: Network processing script
108
+ | |--setting.py: Parameter setting script
109
+ | |--train.py: MRBrainS18 training demo script
110
+ | |--test.py: MRBrainS18 testing demo script
106
111
|--toy_data/:For CI test
107
112
|--data/:Data storage module
108
113
| |--MRBrainS18/:MRBrainS18 dataset
@@ -111,10 +116,6 @@ MedicalNet/
111
116
| |--train.txt: training data lists
112
117
| |--val.txt: validation data lists
113
118
|--pretrain/:Pre-trained models storage module
114
- |--model.py: Network processing script
115
- |--setting.py: Parameter setting script
116
- |--train.py: MRBrainS18 training demo script
117
- |--test.py: MRBrainS18 testing demo script
118
119
|--requirement.txt: Dependent library list
119
120
|--README.md
120
121
```
0 commit comments