我的“工具”库

csdn推荐

本代码是基于python pytorch环境安装的。下载本代码后,有个requirement.txt文本,里面介绍了如何安装环境,环境需要自行配置。或可直接参考下面博文进行环境安装。如果实在不会安装的,可以直接下载免安装环境包,有偿的哦安装好环境之后,代码需要依次运行 01数据集文本生成制作.py02深度学习模型训练.py和03pyqt_ui界面.py数据集文件夹存放了本次识别的各个类别图片。本代码对数据集进行了预处理,包括通过在较短边增加灰边,使得图片变为正方形(如果图片原本就是正方形则不会增加灰边),和旋转角度,来扩增增强数据集,运行01数据集文本制作.py文件,会就读取数据集下每个类别文件中的图片路径和对应的标签运行02深度学习模型训练.py就会将txt文本中记录的训练集和验证集进行读取训练,训练好后会保存模型在本地

文章来源:https://blog.csdn.net/weixin_44741835/article/details/139631567



微信扫描下方的二维码阅读本文

© 版权声明
THE END
喜欢就支持一下吧
点赞12 分享
评论 抢沙发
头像
欢迎您留下宝贵的见解!
提交
头像

昵称

取消
昵称表情代码图片

    暂无评论内容