From afdb8e43268dcb9fcbe6bc28438cdfc420202a80 Mon Sep 17 00:00:00 2001
From: sunrui <icefoxd@163.com>
Date: Thu, 19 Dec 2024 01:45:18 +0000
Subject: [PATCH] =?UTF-8?q?=E5=88=86=E7=B1=BB=E8=AE=AD=E7=BB=83.md?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 readme.md | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/readme.md b/readme.md
index b5af9ca..e0596d0 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. 评测指标
 
-- 
2.22.0