本代码是基于python pytorch环境安装的。下载本代码后,有个环境安装的requirement.txt文本,环境需要自行配置。或可直接参考下面博文进行环境安装。
安装好环境之后,
代码如需重新训练的话,需要依次运行
01、02、03py文件。
,如果只是调用已经训练好的模型,去做识别的话,直接运行03pyqt.py即可
以下关于每个py文件的介绍:
输入指令python 01划分数据集.py
就会将我们的数据集转成yolo格式的txt,同时生成train.txt和val.txt,和配置文件data.yaml
运行02train.py即可开始训练模型。
最后运行03pyqt.py文件就有pyqt的可视化界面。
通过点击加载图片按钮,来选择我们要识别的图片,再点击检测按钮就可以完成识别了。