本项目是毕业论文的代码。欢迎fork或者comment!
- data
用于图像的预处理和PyTorch dataset类的定义。
- model
包含论文所用到的所有模型的网络构建代码。
- datasets
论文用到的训练集和测试集,具体的下载和引用方式详见该目录下的README文件。
- test
整个流程用到的shell脚本。执行教程详见该目录下的README文件。
- dataset_generate.py
生成数据集的代码。
- train.py
训练和测试代码。
- config.py
项目的配置参数。