![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
DL
Array03
这个作者很懒,什么都没留下…
展开
-
关于TensorFlow使用的一些笔记
1. Tensorflow切换使用CPU运算config = tf.ConfigProto(device_count = {'GPU': 0})with tf.Session(config=config) as sess:2. can’t open CUDA library libcupti.so.8.0.locate libcupti* 找到其所在位置然后在环境变量里加上:export LD原创 2017-07-04 20:07:24 · 22633 阅读 · 10 评论 -
caffe通过CMakeLists.txt编译适用于自定义GPU ARCH的动态库
1.修改caffe目录下cmake文件夹下Cuda.cmake文件,修改下面这一行:set(Caffe_known_gpu_archs "30 35 50 52 61")2.编译的时候设置编译选项:cmake -DCUDA_ARCH_NAME=All ..原创 2017-12-03 21:14:39 · 1887 阅读 · 0 评论 -
关于caffe模型训练与处理的一些小结
1. BN层的参数 use_global_statususe_global_stats: true 表示使用预训练模型内的BatchNorm层均值与方差参数,而非基于当前batch内计算均值和方差,若是从头训练则必须设置为false(caffe默认是设置false,可不手动设置), 若是finetuning则根据情况设置。2. LOSS层参数 loss_weightloss_weight表示当前l...原创 2018-04-14 17:21:32 · 573 阅读 · 0 评论 -
SSD(Single Shot MultiBox Detector)及基于SSD结构的目标检测网络总结
SSD是目标检测的主要框架之一,本博客对基于SSD结构设计的各种目标检测网络做一个总结。目前主要有以下这些: original SSD : https://github.com/weiliu89/caffe/tree/ssd https://arxiv.org/abs/1512.02325 DSSD: https://github.com/chengyangf...原创 2018-06-30 14:12:44 · 3241 阅读 · 0 评论