
深度学习
文章平均质量分 78
Muzi_Water
这个作者很懒,什么都没留下…
展开
-
keras_retinanet 目标检测——自定义图片数据集的模型训练步骤
最近在学习 keras_retinanet ,下面就记录下用自己的数据集进行的模型训练。大致分为以下几步:自定义训练数据 图片目标标注 生成用于训练的图片名称、目标标注位置及目标类别的.csv文件 开始训练模型(注意参数调整) 转换训练好的模型 用转换后的模型进行目标检测下面就一步一步介绍吧:目录1.下载包,安装环境。2.准备数据集3.训练模型4.目标检测...原创 2019-03-13 18:14:34 · 5896 阅读 · 6 评论 -
ImageAI 学习
最近在学习ImageAI对图片的处理,在此记录下学习过程。参考:ImageAI 的 Github 链接目标:用自己的图片库实现液晶屏异常预测。1. 将自己的图片整理成224×224(模型默认)像素的大小,新建一个tf-train目录(名字随便起),在该目录下分为train和test两个目录,目录下将正常的图片放在normal文件夹下,将存在异常的图片放在abnormal文件夹下,文件夹...原创 2019-03-13 16:10:57 · 1078 阅读 · 0 评论 -
tensorflow错误解决:“tensorflow.python.framework.errors_impl.ResourceExhaustedError”
tensorflow错误:“tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[32,256,55,55]”解决方法:减少批处理Batch 的大小 降低全连接层的维度 增加池化 层 缩小输入图片大小总之呢,就是显卡内存不能满足此次训...原创 2019-03-11 11:37:54 · 19421 阅读 · 5 评论 -
Win10安装CUDA 9.0+cuDNN v7.0+tensorflow-gpu 1.11.0和opencv的教程
鉴于我自己踩过的坑,提醒大家千万不要直接pip install tensorflow-gpu而不指定版本!!!我一开始用最新的1.13.1版本总是报“ImportError: DLL load failed: 找不到指定的模块。”的错误,果断的卸载后安装了1.11.0版本,直接就OK啦!注:如果需要matplotlib包,请在安装tensorflow-gpu之前安装,因为conda insta...原创 2019-03-05 13:24:58 · 2744 阅读 · 0 评论 -
神经网络最优化方法总结比较
参考:深度学习最全优化方法总结比较(SGD,Adagrad,Adadelta,Adam,Adamax,Nadam)学习率 方法 缺点 优点 经验 手动设置学习率 SGD(mini-batch gradient descent) 选择合适的learning rate比较困难 - 对所有的参数更新使用同样的learning rate。对于稀疏数据或者特征,有...原创 2018-08-06 14:17:53 · 3787 阅读 · 0 评论