demo仓库和视频演示:
到此一游7758258的个人空间_哔哩哔哩_bilibili
效果展示图如下:
代码文件展示如下:
运行01create_txt.py可以将dataset文件下的图片路径保存在txt文本中,
运行02train.py能对图片进行路径和对应的标签进行加载训练模型保存在weights文件夹下,
运行03predict.py可以调用模型对单张图片进行预测,
运行04pyqt界面.py可以生成一个可视化的界面,通过点击按钮加载图片识别。
科普下卷积神经网络相关知识:
CNN是卷积神经网络(Convolutional Neural Network)的缩写。它是一种前馈神经网络,特别适用于处理具有网格状结构的数据,例如图像、视频和声音等。CNN由多个卷积层、池化层和全连接层组成。
CNN的核心思想是通过卷积操作和池化操作来提取输入数据中的特征。卷积层使用卷积核对输入数据进行滑动窗口操作,以捕捉不同位置的局部特征。通过堆叠多个卷积层,网络可以学