深度学习项目框架

常见项目架构如下:

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


个人认为这个框架比较规范

posted @ 2023-11-15 10:54  seekwhale13  阅读(62)  评论(0)    收藏  举报