图像检测环境配置
安装anaconda
网上下载最新版本的anaconda,新建一个python3.5的环境
安装keras
pip install keras==2.1.5
安装tensorflow
若没有安装gpu,因此没有安装gpu版本的tensorflow
pip install tensorflow==1.13.1
若安装了gpu,则安装gpu版本的tensorflow
pip install tensorflow_gpu==1.13.1
安装其他的模块
PIL模块
pip install pillow
opencv模块
pip install opencv-python
下载失败则下载whl文件直接安装
opencv-python 4.4.0.42
matplotlib模块
pip install matplotlib
tqdm模块,显示进度条的,训练时需要
pip install tqdm
安装cuda和cudnn
参考教程
版本问题
版本问题2
tf-1.13 对应的CUDA是10,cuDNN是7.4
cudnn-9.0-windows10-x64-v7.4.1.5
cuda_10.0.130_411.31_win10
运行GPU训练出现的问题
utils下默认gpu需要修改,0->1。
GTX 1050是2G显存,将输入图片数据改为[50, 50, 3],实际是[960,540,3],一次训练所选取的样本数batch_size = 1。训练速度上去了,但是图片精度下降,每次只训练一张图片,梯度难以收敛。运行到部分,容易OOM,显存用尽任务失败。