本期给大家介绍的是138基于yolov4目标检测的蔬菜检测含数据集。效果图如下:
代码下载和视频演示地址:
138基于yolov4目标检测的蔬菜检测含数据集_哔哩哔哩_bilibili
代码是十分的简介,主要是4个py文件
运行01数据集制作.py会将datasets文件夹下的JPEGImages(jpg或png图片)和Annotations(xml标签)进行读取生成txt文本,里面存放了图片路径和对应的标签。
运行02train.py能够读取txt文本内容结合图片进行训练模型,训练好的模型保存在weights文件夹下。
运行03predict.py可以对单张图片进行检测。
运行04pyqt.py会有个pyqt的可视化界面,通过点击按钮可以加载感兴趣的图片进行识别。