分类训练.md

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. 评测指标