笔记链接:代码结构分析
以下为笔记内容:
目标检测一般代码结构
以陈云—<深度学习框架PyTorch入门与实践>中猫狗大战实例进行,有兴趣可以看博客中pytorch入门推荐,有相关资源推荐!
陈云的链接------资源链接
程序功能:
一般文件组织结构:
├── checkpoints/ # 用于保持训练的模型
├── data/ # 数据相关操作,数据预处理,dataset实现等
│ ├── init.py
│ ├── dataset.py
│ └── get_data.sh
├── models/ # 模型定义存放多个模型
│ ├── init.py
│ ├── AlexNet.py
│ ├── BasicModule.py
│ └── ResNet34.py
└── utils/ # 可能用到的工具函数
│ ├── init.py
│ └── visualize.py
├── config.py # 配置文件,所有可配置变量都集中在此,并且提供默认值
├── main.py # 主文件,训练和测试程序的入口
├── requirements.txt # 程序依赖的第三方库
├── README.md 提供程序的必要说明
主函数主要大致结构