Skip to content

Commit dcece0d

Browse files
authored
Update readme.md
1 parent c29d1ca commit dcece0d

File tree

1 file changed

+16
-11
lines changed

1 file changed

+16
-11
lines changed

examples/CNN_UTS/readme.md

Lines changed: 16 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3,17 +3,22 @@
33
本项目基于 Paddle框架复现的材料微观结构强度预测项目,通过 X 射线 CT 图像预测聚合物 - 陶瓷复合材料的极限抗拉强度(UTS),实现了论文《Predicting the Strength of Composites with Computer Vision Using Small Experimental Datasets》中的核心方法,实现了图像的回归预测,支持ResNet18、五折交叉验证、超参数配置、自动训练与评估、可视化等功能
44

55
## 目录结构
6-
.
7-
├── main.py # 主入口,支持训练/评估
8-
├── data_utils.py # 数据集与数据处理
9-
├── model_utils.py # 模型、可视化、指标等工具
10-
├── conf/
11-
│ └── resnet.yaml # 配置文件(可自定义参数)
12-
├── Dataset/ # 数据集目录(见下方数据格式说明)
13-
├── resnet18-v5-finetune/ # 保存每折模型权重
14-
├── Saved_Output/ # 保存预测结果与可视化
15-
├── requirements.txt # 依赖库
16-
└── readme.md # 项目说明文档
6+
```
7+
CNN_UTS/
8+
9+
├─ conf/ # 配置文件
10+
│ └─ resnet.yaml
11+
├─ data_utils.py # 数据集加载与处理工具
12+
├─ model_utils.py # 模型相关工具(如随机种子设置)
13+
├─ main.py # 主程序,包含训练与评估流程
14+
├─ requirements.txt # 依赖包列表
15+
├─ readme.md # 项目说明文档
16+
├─ resnet18-v5-finetune/ # 各折训练得到的模型参数
17+
├─ outputs/ # 日志与输出目录
18+
├─ Saved_Output/ # 保存的预测结果与可视化图片
19+
└─ Dataset/ # 数据集目录
20+
├─ Train_val/ # 训练/验证集
21+
└─ Test/ # 测试集
1722
```
1823

1924
## 环境依赖

0 commit comments

Comments
 (0)