深度学习之目标检测(二)配置篇win+python3.5

图像检测环境配置

安装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,显存用尽任务失败。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值