tensorflow
文章平均质量分 74
Android_chunhui
人不一定能什么都擅长,有自己的特点就好!
展开
-
服务器用户修改libcudnn.so链接
在安装tensorflow的时候,每个版本要求的cuda和cudnn’版本都不相同,在更新tensorflow后很容易遇到与cuda,cudnn版本不匹配的情况,如找不到libcudnn.so.xx的,可以通过连接的方式,一般来说高版本兼容低版本。 但是作为服务器用户没有权限直接在usr目录下ln -s,那么可以在anaconda/envs/tensorflow/lib链接就好原创 2018-03-25 15:54:35 · 902 阅读 · 0 评论 -
跑通faster-rcnn --游戏体验极差
程序使用基于tensorflow的faster rcnn实现, https://github.com/smallcorgi/Faster-RCNN_TF 但按照readme上跑pascal voc数据集时遇到了诸多问题,记录如下: 1.需要指定g++参数,-D_GLIBCXX_USE_CXX11_ABI=0,直接在lib/make.sh赋值CXXFLAGS='-D_GLIBCXX_USE_CX原创 2018-05-01 11:03:52 · 413 阅读 · 5 评论 -
faster-rcnn 笔记
anchor-based方法faster rcnn是基于anchor的方法,包括后面的yolo-9000也是。数据集中bounding box数目是不确定的,使用anchor解决了这个问题。 从bounding box角度来说, 开始设置的9个anchor是RPN预测的bbox的先验,而RPN输出的bbox又作为了fast rcnn预测的bbox先验。frcnn获取anchor的方式是直接...原创 2018-04-27 16:44:11 · 403 阅读 · 0 评论 -
计算CNN感受野
定义调整输入尺寸,当有input size使网络feature map的尺寸为 1x1时,这个input size为当前feature map的感受野大小。整个网络的感受野就是最后一层feature map的感受野大小。如VGG16感受野时196x196,指的时conv5_3(pool5之前)的感受野大小。如何计算 RFl=(RFl−1−1)∗stride+fsizeRFl=(RF...原创 2018-07-04 19:40:38 · 343 阅读 · 0 评论 -
pytorch 模型加载部分权重与部分模型加载权重
模型加载部分权重model = ...model_dict = model.state_dict()pretrained_dict = torch.load(load_name)# 1. filter out unnecessary keyspretrained_dict = {k: v for k, v in model_dict.items() if k in pretraine...原创 2019-06-20 21:20:38 · 5925 阅读 · 7 评论 -
tensorflow笔记
维度第一个括号里面的每一项是第一维的元素,再进一层括号中的项是第二维的元素,最后一维的元素是最后一层括号中单个的数字。后面的维度作为前面维度的项, 当前面维度的元素修改或复制后,后面维度的元素也是跟着变得。a = np.arange(2*4*4)b = a.reshape(1,4,4,2) #应该这样按反序来理解:最后一个2是一个只有2个元素的向量,最后的4,2代表4...原创 2019-07-24 11:21:36 · 150 阅读 · 0 评论 -
记录tensorflow c++接口编译
tensorflow版本:r1.9cuda: 9.0cudnn: 7.3.0bazel: 0.15protobuf: 3.5.0 (版本要与tf对应)eigen:使用tensorflow脚本生成。nccl: 2.4.8步骤:(基本参考:https://blog.csdn.net/qq_37541097/article/details/86232687)安装protobufsu...原创 2019-07-28 02:11:54 · 509 阅读 · 1 评论