- 博客(8)
- 收藏
- 关注
RSS订阅原创 tensorflow恢复restore官网ckpt出错,Tensor name not found
搭建好vgg,resnet等网络,加载tensorflow官网(https://github.com/tensorflow/models/tree/master/research/slim)提供的预训练ckpt模型.使用restore运行出错,一些节点(conv/biases)在ckpt中不存在,出错无法运行.正确的做法,slim.assign_from_checkpoint_fn(’./data/pre_trained/resnet_v1_50.ckpt’,slim.get_trainable_va
2020-05-22 11:43:22
322
原创 python -- numpy或list列表与其元素间修改数据的相互联系
普通的赋值>>>a = [1,2]>>>b = [a, a]>>>b[1] = 100>>>b[[1, 2], 100]>>>a[1, 2] # a是b的元素，如果只是修改了b【1】，只是把此处a元素换成了其他，不会改变a本身>>&...
2020-03-31 17:05:10
168
原创 numpy中的broadcast，以maximum和加法为例
用numpy对array进行maximum或加法操作，当两个参数的维度不同时，首先以加法为例。>>>a = np.arange(5).reshape(-1,1)>>>a # a shape is (3,1)array([[0],...
2020-03-31 09:13:59
211
原创 目标检测训练pascal，batch为1却会OOM！
直接上图，pascal数据中存在这么一张图片，尺寸为500X71，pascal图片预处理时会用到short_size_resize,如下：def short_side_resize(img_tensor, gtboxes_and_label, target_shortside_len):‘’’:param img_tensor:[h, w, c], gtboxes_and_label:[-1...
2020-03-18 17:26:35
567
原创 tensorflow中内存挤爆，训练慢，可能时sess.run了图中没有构建的新节点！
tensorflow训练FPN+Faster-rcnn，每次训练到1000步左右就会cpu杀死进程，每一步训练的时间在10-15s之间，比较慢。原因：之前在debug时，输出了图中所示的信息，这在训练中不断创建了新节点，导致graph不断增大以至于内存挤爆。在sess run中，特别是有循环的情况，要保证run的节点是图中创建好的。...
2020-03-11 11:11:19
297
原创 FPN+Faster_Rcnn调试，fast_rcnn_location_loss一直为0？
如题，调试FPN+fastrcnn源码，tensorflow版本，过程中fast-rcnn_location_loss，一直为0。经过分析，初始过程，rpn网络推荐的box在fast rcnn阶段跟gtbox对比时没有positive样本，这时候，rcnn_location_loss为0是合理的。但网络训练一直进行，无论怎么调整，依然一直为0，就难以理解来。无奈之中，一步步调试，取出fast-r...
2020-03-04 13:59:55
836
1
原创 faster-rcnn训练pascal-voc2012出现‘Tensor had NaN values’
使用faster-rcnn训练pascal2012时发现出错，总是会出现张量异常。减小学习率，改变损失函数毫无作用。总是发生某些数据一传入网络，马上就挂掉。不得不从数据源入手，通过逐一排查，发现voc2012中存在2008和2010不同版本的数据集，两种数据集定义标志框坐标顺序不同，导致按照2008格式数据解析2010格式数据会出现，出现尺寸为负的现象，这种情况网络loss不崩溃才怪！！查错除了...
2020-02-27 17:05:23
335
3
原创 faster-rcnn中对fast-rcnn loss部分的代码解析
fast-rcnn部分对于fast_rcnn_loss的实现，首先通过minibatch_indices,minibatch_reference_boxes_mattached_gtboxes, minibatch_object_mask, minibatch_label_one_hot = self.fast_rcnn_minibatch(self.fast_rcnn_all_level_p...
2020-02-26 14:35:18
670
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人