【Python】Pytorch（CNN）实现垃圾，宝石，表情，车牌，鸟类，食物等各种分类

作为一个比较喜欢玩花活的程序员，涉猎的技术比较多，各种语言都会了解来扩展知识面。
比如，最近发现Python的pytorch很好玩，打算玩一手机器学习，然后就有了如下这篇文章。
（当然不是摸鱼！已经拿到4家公司的offer啦）

效果演示

明星/人脸识别

鸟类识别

垃圾分类

运行效果演示

效果演示比较简单啦，就是你可以放入一个图片或者一个视频，然后我的代码就会检测出图片中是什么物品。
并且为了提高程序的可用性，我专门做了代码的封装，只需要你放入不同的数据集，就可以直接生成对应的不同类型的模板。

方法如上，你只需要设定要训练的模型的名称，然后放入模型的各种图片即可。
之后运行train代码即可进行模型的训练，当然，模型的训练最好你的电脑的GPU还不错，不然，你的训练速度嗷嗷慢哦。

上图的代码使用PyQt5来制作UI。
卷积神经网络使用的是CNN。

论文

同时，我也编写了比较完整的论文模板，保证你可以理解所有的代码并帮助你良好的交差。

代码下载

代码下载链接