作为一个比较喜欢玩花活的程序员,涉猎的技术比较多,各种语言都会了解来扩展知识面。
比如,最近发现Python的pytorch很好玩,打算玩一手机器学习,然后就有了如下这篇文章。
(当然不是摸鱼!已经拿到4家公司的offer啦)
效果演示
运行效果演示
效果演示比较简单啦,就是你可以放入一个图片或者一个视频,然后我的代码就会检测出图片中是什么物品。
并且为了提高程序的可用性,我专门做了代码的封装,只需要你放入不同的数据集,就可以直接生成对应的不同类型的模板。
方法如上,你只需要设定要训练的模型的名称,然后放入模型的各种图片即可。
之后运行train代码即可进行模型的训练,当然,模型的训练最好你的电脑的GPU还不错,不然,你的训练速度嗷嗷慢哦。
上图的代码使用PyQt5来制作UI。
卷积神经网络使用的是CNN。
论文
同时,我也编写了比较完整的论文模板,保证你可以理解所有的代码并帮助你良好的交差。