Tensorflow
Tsingzao-于廷照
这个作者很懒,什么都没留下…
展开
-
Tensorflow由于版本问题出现的几种错误及解决办法
1、AttributeError: 'module' object has no attribute 'rnn_cell'S:将tf.nn.rnn_cell替换为tf.contrib.rnn2、TypeError: Expected int32, got list containing Tensors of type '_Message' instead.S:由于tf.conca...原创 2017-12-28 20:34:43 · 47440 阅读 · 6 评论 -
Tensorflow + Faster RCNN 遇到的问题及解决方案
代码fork自 https://github.com/endernewton/tf-faster-rcnn1、数据准备过程,参见pytorch版本,见这里。2、遇到问题:InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. 解决方案:删除...原创 2018-07-24 20:32:21 · 1488 阅读 · 1 评论 -
Tensorflow Faster RCNN 如何增加额外输出
假定我们的需求是额外的增加一个多类别信息,并且与之前已有的类别共用同一个bounding box,修改如下:1、数据读取层: (1) lib/datasets/pascal_voc.py ① __init__函数 需要在41行之后增加需求类别及index ② _load_pascal_annotation函数 需要在...原创 2018-08-05 17:03:06 · 358 阅读 · 0 评论 -
用TensorFlow搭建网络训练、验证并测试
本文总结tensorflow使用的相关方法,包括:0、定义网络输入1、如何利用tensorflow在已有网络入resnet基础上搭建自己的网络结构2、如何添加自己的网络层3、如何导入已有模块入resnet全连接层之前部分的参数4、定义网络损失5、定义优化算子以及衰减优化算子6、预测网络输出7、保存网络模型8、自定义生成训练batch9、训练网络10、利用...原创 2018-08-15 19:35:16 · 7282 阅读 · 1 评论