深度学习工具
文章平均质量分 55
张叫张大卫
放弃从来都是一件很容易的事情
展开
-
Tensorflow(0)基础概念
tensorflow基础概念一、第一个简单计算样例tensorflow是基于计算图的,因此,需要先建立图中的连接,然后运行图图的连接可以想象成做流程图,例子中a,b,result就是建立流程图中的各个节点以及连接方式import tensorflow as tfa=tf.constant([1.0,2.0],name='a')b=tf.constant([2.0,3....原创 2018-08-19 21:11:46 · 358 阅读 · 0 评论 -
使用anaconda创建虚拟环境安装不同深度学习框架
使用anaconda安装tensorflow等框架,可以先通过anaconda创建虚拟环境,然后在每一个虚拟环境下装上不同的版本的框架(例如在python2上安装一个tensorflow,在oython3上安装一个tensorflow)。并且如果需要删除也比较方便,而不影响这个系统。一、在虚拟环境中安装tensorflow和keras主要翻译自tensorflow的安装教程,这里添加了对虚...原创 2018-08-27 17:19:54 · 12319 阅读 · 1 评论 -
Tensorflow(1)进行多维矩阵的拆分与拼接
最近在使用tensorflow进行网络训练的时候,需要提取出别人训练好的卷积核的部分层的数据。由于tensorflow中的tensor和python中的list不同,无法直接使用加法进行拼接,后来发现一个函数可以完成tensor的拼接。函数形式如下: tf.concat(concat_dim,values,name='concat')其中,第一个参数表示需要拼接的多维tensor,并且可以...原创 2018-09-26 19:38:21 · 14191 阅读 · 1 评论 -
Tensorflow(2)保存模型与恢复
###一、数据模型的保存使用saver类,自动保存tensorflow的图结构(***.ckpt.meta),参数取值(***.ckpt.data),以及目录下的文件列表(***.ckpt.index),还有一个checkpoint文件。定义变量变量操作变量初始化构建saver类使用保存模型参数到文件import tensorflow as tfv1=tf.Variable(...原创 2018-09-26 22:28:58 · 1438 阅读 · 0 评论 -
Tensorflow+SSD使用原始权重并修改分类网络进行单目标检测
本文的源码地址是https://github.com/balancap/SSD-Tensorflow由于项目需要,需要对场景中的人体进行检测,但是原始的SSD网络是20种类别的网络,而只需要获取人的分类即可,当我按照其说明在具有两块1080Ti的服务器上训练8个小时,损失值降低到10左右,但是,其效果不如原始的权重参数的效果,因此,想修改网络连接,只保留最后一层的分类网络中对人体的分类。步骤如...原创 2018-09-27 18:05:27 · 2660 阅读 · 10 评论 -
Pytorch(1) pytorch中的BN层的注意事项
最近修改一个代码的时候,当使用网络进行推理的时候,发现每次更改测试集的batch size大小竟然会导致推理结果不同,甚至产生错误结果,后来发现在网络中定义了BN层,BN层在训练过程中,会将一个Batch的中的数据转变成正太分布,在推理过程中使用训练过程中的参数对数据进行处理,然而网络并不知道你是在训练还是测试阶段,因此,需要手动的加上,需要在测试和训练阶段使用如下函数。model.train(...原创 2018-11-22 21:44:03 · 10858 阅读 · 0 评论