diff --git a/readme.md b/readme.md index b5af9cafe8bd8f776c2aa52b2bfc0d4e691f0404..e0596d0d7f5e3288f8b0b420730ba2e2f438277d 100644 --- a/readme.md +++ b/readme.md @@ -4,16 +4,19 @@ 在数据处理阶段,根据不同的需求对影像组学数据和深度学习数据进行预处理,并从原始数据库中查询标注的数据,读取DICOM数据并保存处理后的结果 -- **影像组学数据**:`data/db_data_to_feat.py` -- **深度学习数据**:`data/db_data_to_npy.py` +- **影像组学数据(csv数据)**:`data/db_data_to_feat.py` +- **深度学习数据(npy数据)**:`data/db_data_to_npy.py` ## 2. 分类训练 分类训练模块涵盖了不同类型的模型训练,包括2d3d分类、预训练分类以及基于影像组学特征的分类 -- **2d3d分类**:`pytorch_train/classification_model_2d3d.py` -- **预训练分类**:`pytorch_train/encoder_cls.py` +- **2d3d分类网络**:`pytorch_train/classification_model_2d3d.py` +- **预训练分类网络**:`pytorch_train/encoder_cls.py` - **影像组学分类**:`pytorch_train/feat_cls.py` +- **深度学习分类**:`pytorch_train/train_2d3d.py` +- **多卡启动**:`torchrun --nproc_per_node=4 train_2d3d.py` +- **单卡启动**:`python train_2d3d.py` ## 3. 评测指标