在此感谢牛小顿同学提供的资源,https://blog.csdn.net/zong596568821xp 与这位博主的博客,我是按着他搭建的环境,亲测好用。
一、使用conda创建一个新的虚拟环境
conda create -n second_test python=3.6 numpy
二、安装tersonflow-gpu版本
2.1安装keras 2.2.4
pip install keras-2.2.4
2.2 安装tensorflow-gpu-1.14.0
conda install tensorflow-gpu==1.14.0 #如果是gpu版本
pip install tensorflow==1.14.0 #如果是cpu版本
效果如图:
三、安装依赖项
在训练的过程中,会有很多依赖,所以提前安装一遍
3.1 conda install Pillow
3.2 conda install matplotlib
3.3 conda install ffmpeg
3.4 pip install opencv-contrib-python
这里有一个疑问:为什么conda install opencv-contrib-python 没那么顺利。
三、下载python——yolo文件
网站:
http://219.217.235.37/gitlab/NewYinbao/iarc2019_keras_yolo3
下载后解压,如图所示:
接下来,我们使用VScode进行编译