本期给大家介绍的是149使用python基于CNN的150种动物识别。效果图如下:
代码仓库和视频演示地址:
149使用python基于CNN的150种动物识别_哔哩哔哩_bilibili
代码主要是data文件夹存放数据集,和3个py文件,分别用于运行01数据集制作,02训练模型,03pyqt的可视化ui界面。
01数据集文本生成制作.py
运行之后就会在logs文件夹下生成txt文本。
里面存放的是图片路径和对应的标签。
运行02train.py即可训练模型。训练好的模型也是保存在logs文件夹下。
最后运行03pyqt.py就是可视化的界面,在ui界面上点击加载自己的感兴趣的图片识别即可。