本期给大家介绍的是127基于python深度学习识别30种乐器,效果图如下:
代码下载和视频演示地址:
127基于python深度学习识别30种乐器_哔哩哔哩_bilibili
代码整体是非常简便的,主要是data数据集文件夹和3个py文件。
运行01.py能将图片数据保存在txt格式的文本里面,并划分训练集和验证集。
运行02train.py会txt文本里面的图片数据进行读取并训练模型,训练好的模型也保存在logs文件夹下。
在训练时提供了11种卷积神经网络供任意切换。包括resnet\mobilenet\alexnet\vgg\sufflenet\efficientnet等等
训练完成后还会生成评价指标图,f1-score,精确率和召回率。
运行03pyqt.py可以选择自己感兴趣的图片进行识别,通过调用上一步训练好的模型识别图片中的乐器。