深度学习项目框架
常见项目架构如下:
|--project_name/
| |--data/ # 数据
| |--datasets/ # 生成数据集,加载数据集
| | |--data_loader.py
| |--models/ # 模型
| | |--model.py
| |--configs/ # 配置文件
| | |--config.py
| |--model_hub/ # 预训练模型权重
| |--utils/ # 辅助模块,可以是日志、评价指标等
| | |--utils.py
| | |--metrics.py
| |--tools/ # 训练,评估,预测的脚本
| | |--train.py
| | |--eval.py
| | |--predict.py
| |--outputs/ # 输出的结果
| | |--checkpoints/ # 训练好的模型文件
| | |--logs/ # 日志
| | |--images/ # 可视化的结果图片
| |--requirements.py # 依赖包
| |--README.md # 项目说明
个人认为这个框架比较规范

浙公网安备 33010602011771号