代码下载和视频演示地址:
效果演示图如下:
完整的文件展示如下:
算法部分
其中dataset文件夹下存放的图像数据集。
运行01create_txt.py会将数据集图片路径带上标签保存在txt文本中。
运行02train.py会将txt文本中的图像数据读取进行模型的训练,最后保存在weights文件夹下。
运行03predict.py会对单张图片进行检测。
运行04ui_pyqt5.py是一个可视化的操作界面,通过点击按钮检测自己感兴趣的视频。
代码下载和视频演示地址:
效果演示图如下:
完整的文件展示如下:
算法部分
其中dataset文件夹下存放的图像数据集。
运行01create_txt.py会将数据集图片路径带上标签保存在txt文本中。
运行02train.py会将txt文本中的图像数据读取进行模型的训练,最后保存在weights文件夹下。
运行03predict.py会对单张图片进行检测。
运行04ui_pyqt5.py是一个可视化的操作界面,通过点击按钮检测自己感兴趣的视频。