基于模块化脚本的深度学习模型构建与应用
1. 模块化脚本概述
整个脚本可拆分为四个模块,分别为 train.py 、 test.py 、 plotting.py 和 crop_resize_transform.py 。你可以在相应文件夹中找到这些脚本。运行脚本时,将该文件夹设为项目文件夹,然后运行 train.py 文件,此文件会在执行过程中根据需要从其他模块导入函数。
1.1 模块功能概述
| 模块名称 | 主要功能 |
|---|---|
crop_resize_transform.py |
包含图像裁剪、调整大小以及生成训练和测试数据的函数 |
plotting.py |
包含绘制关节和肢体的函数 |
test.py |
包含测试 VGG16 模型性能的函数 |
train.py |
包含训练 VGG16 模型所需的函数以及训练和测试模型的调用代码 |
超级会员免费看
订阅专栏 解锁全文

被折叠的 条评论
为什么被折叠?



