基于yolov3和pythorch框架的火焰识别检测算法
这是本人第一次写博客,就当是自己实现算法的一个记录吧,有什么不好的地方也请多多指教。我会详细的从环境的配置到算法实现都说明一下,希望对大家能有帮助。 本火焰识别算法采用的是pytorch版本的yolov3检测,yolov3基于Darknet-53网络结构,在图像识别领域应用特别广。 本算法识别的效果如下: 下面开始实际操作啦 一、配置环境 算法所需环境如下: Python: 3.7.4 Tensorflow-GPU 1.14.0 Keras: 2.2.4 numpy:1.17.4 我的操作系统是Ubuntu16.04,windows系统同样可以实现此算法这里建议使用anaconda来快速搭建一个虚拟环境。 如果是ubuntu系统可以参考这篇博客安装anaconda,Anaconda 是 Python 的一个出色的集成开发工具集,对于搞深度学习的人来说应该是必备的。 ubuntu安装anaconda链接 有了anaconda后就可以用它创建虚拟环境啦。具体可以参考这篇博客。 anaconda创建虚拟环境 之后安装Tensorflow,由于我的显卡是AMD的,所以安装的CPU版的,大家可以根据自己的显卡安装相应版本。运行下面的命令安装Tensorflow。 pip install tensorflow==1.14.0 下一步安装相应版本的Keras,运行下面的命令。 pip