Tensorflow
Tsingzao-于廷照
这个作者很懒,什么都没留下…
展开
-
Tensorflow由于版本问题出现的几种错误及解决办法
1、AttributeError: 'module' object has no attribute 'rnn_cell' S:将tf.nn.rnn_cell替换为tf.contrib.rnn 2、TypeError: Expected int32, got list containing Tensors of type '_Message' instead. S:由于tf.conca...原创 2017-12-28 20:34:43 · 47382 阅读 · 6 评论 -
Tensorflow + Faster RCNN 遇到的问题及解决方案
代码fork自 https://github.com/endernewton/tf-faster-rcnn 1、数据准备过程,参见pytorch版本,见这里。 2、遇到问题:InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. 解决方案:删除...原创 2018-07-24 20:32:21 · 1477 阅读 · 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 · 348 阅读 · 0 评论 -
用TensorFlow搭建网络训练、验证并测试
本文总结tensorflow使用的相关方法,包括: 0、定义网络输入 1、如何利用tensorflow在已有网络入resnet基础上搭建自己的网络结构 2、如何添加自己的网络层 3、如何导入已有模块入resnet全连接层之前部分的参数 4、定义网络损失 5、定义优化算子以及衰减优化算子 6、预测网络输出 7、保存网络模型 8、自定义生成训练batch 9、训练网络 10、利用...原创 2018-08-15 19:35:16 · 7236 阅读 · 1 评论