深度学习项目框架

发布时间 2023-11-15 11:00:15作者: seekwhale13

常见项目架构如下:

|--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 # 项目说明