003基于python深度学习的水果或其他物体识别小程序_ai识别水果 python

作者：Gausst松鼠会 | 2024-02-25 09:59:39

踩

ai识别水果 python

直接上效果图如下：

代码演示和demo仓库看b站视频：

代码展示：

数据集图片放置在data文件夹下，大家可以根据自己需要比如识别其他物体，只需要模仿data文件夹下的文件命名放入图片即可运行训练模型了。

编辑

运行01数据集文本生成制作.py可以将data文件夹下的图片保存成txt格式

运行02train.py可以将txt中记录的数据读取并训练模型

训练可以使用的卷积神经网络包括： efficientnet、Alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNext、ShuffleNet、Swin_transformer、VGG等。

训练结束后同时会生成评价指标图

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/140211