本代码对数据集进行了预处理,包括通过在较短边增加灰边,使得图片变为正方形(如果图片原本就是正方形则不会增加灰边),和旋转角度,来扩增增强数据集,运行02深度学习模型训练.py就会将txt文本中记录的训练集和验证集进行读取训练,训练好后会保存模型在本地。运行03pyqt_ui界面.py就可以有个可视化的ui界面,通过点击按钮可以加载自己感兴趣的图片识别。下载本代码后,有个requirement.txt文本,里面介绍了如何安装环境,环境需要自行配置。本代码是基于python pytorch环境安装的。