关注
分享
张叫张大卫
放弃从来都是一件很容易的事情
展开
-
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 · 343 阅读 · 0 评论 -
使用anaconda创建虚拟环境安装不同深度学习框架
使用anaconda安装tensorflow等框架，可以先通过anaconda创建虚拟环境，然后在每一个虚拟环境下装上不同的版本的框架（例如在python2上安装一个tensorflow，在oython3上安装一个tensorflow）。并且如果需要删除也比较方便，而不影响这个系统。一、在虚拟环境中安装tensorflow和keras主要翻译自tensorflow的安装教程，这里添加了对虚...原创 2018-08-27 17:19:54 · 12162 阅读 · 1 评论 -
Tensorflow（1）进行多维矩阵的拆分与拼接
最近在使用tensorflow进行网络训练的时候，需要提取出别人训练好的卷积核的部分层的数据。由于tensorflow中的tensor和python中的list不同，无法直接使用加法进行拼接，后来发现一个函数可以完成tensor的拼接。函数形式如下： tf.concat(concat_dim,values,name='concat')其中，第一个参数表示需要拼接的多维tensor，并且可以...原创 2018-09-26 19:38:21 · 14115 阅读 · 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 · 1420 阅读 · 0 评论 -
Tensorflow+SSD使用原始权重并修改分类网络进行单目标检测
本文的源码地址是https://github.com/balancap/SSD-Tensorflow由于项目需要，需要对场景中的人体进行检测，但是原始的SSD网络是20种类别的网络，而只需要获取人的分类即可，当我按照其说明在具有两块1080Ti的服务器上训练8个小时，损失值降低到10左右，但是，其效果不如原始的权重参数的效果，因此，想修改网络连接，只保留最后一层的分类网络中对人体的分类。步骤如...原创 2018-09-27 18:05:27 · 2622 阅读 · 10 评论 -
Pytorch(1) pytorch中的BN层的注意事项
最近修改一个代码的时候，当使用网络进行推理的时候，发现每次更改测试集的batch size大小竟然会导致推理结果不同，甚至产生错误结果，后来发现在网络中定义了BN层，BN层在训练过程中，会将一个Batch的中的数据转变成正太分布，在推理过程中使用训练过程中的参数对数据进行处理，然而网络并不知道你是在训练还是测试阶段，因此，需要手动的加上，需要在测试和训练阶段使用如下函数。model.train(...原创 2018-11-22 21:44:03 · 10809 阅读 · 0 评论