版权声明:原创作品,欢迎转发!转发附上链接 https://blog.csdn.net/qq_26369907/article/details/90300760
一、文件结构
checkpoint
data/
__init__.py
dataset .py
get_data.py
models/
__init__.py
shuffleNetV2 .py
ResNet .py
utils/
__init__.py
config .py
main .py
requirement.txt
README .md
二、作用
1. __inti__.py作用
这个文件在每个目录是必须存在的,如果目录有了该文件,目录相当于一个包package。
在main.py中,应用 dataset.py中的函数 loader
from data.dataset import loader
注意:如__init__.py文件中添加有引入
from .dataset import loader
在main.py中,dataset.py中的函数 loader
from data import loader
2.目录作用
- checkpoint 目录: 存放训练好的模型
- data 目录:存放数据相关操作,数据预处理等;
- models 目录: 定义模型
- utils 目录: 存放工具函数
- config.py文件:所有可用到的静态变量存放于此