修改README，暂时去除不需要的模型定义

README.md

@@ -103,4 +103,11 @@
     └── watermarking_data_process.py #对数据集进行处理，嵌入黑盒水印
 
 ```
-
+### 训练步骤
+1. 将数据集目录结构组织好
+2. 使用脚本生成类型标签文本
+```shell
+python ./tool/generate_txt.py --txt_path './dataset/CIFAR-10_ori'  --specific_data 'train_cifar10_JPG' --txt_name 'train'
+python ./tool/generate_txt.py --txt_path './dataset/CIFAR-10_ori'  --specific_data 'test_cifar10_JPG'  --txt_name 'test'
+```
+3. 调用bash_run.sh脚本进行训练

model/__init__.py

@@ -1,3 +1,3 @@
-from .timm_model import timm_model
-from .yolov7_cls import yolov7_cls
+# from .timm_model import timm_model
+# from .yolov7_cls import yolov7_cls
 from .layer import cbs, elan, mp, sppcspc, linear_head