tensorflow
ShanLanX
这个作者很懒,什么都没留下…
展开
-
tensorflow gpu版本读取cifar10-binary出现类似卡死状态
今天使用自己写的卷积去调用tensorflow cifar10模组中的cifar10_input.py去读取cifar10数据集,但是每次运行到distorted_input函数就一直停在那,并且cpu全部占满,想了很久都找不到原因,因为tensorflow没有报错,最后实在没办法我就去翻tensorflow的cifar源码,果然官方有说明这个问题。 找到cifar_train.py这个文件,在其原创 2017-12-23 01:32:00 · 2586 阅读 · 6 评论 -
TensorFlow resize_images函数导致TFRecord产生形状不匹配
一般我们在遇到训练图像大小不统一时,都需要将其统一大小,而TensorFlow提供resize_images函数,然而按一般能找到的资料的使用方式一般是先调用resize_images调整图像大小,然后将其存入TFRecord。但是我们知道像素值是没有小数的,每个通道的值都是0-255,神奇的是在使用resize_images后,进行插值计算后,数据矩阵中的值会被强行变成小数也就是浮点型,本来是无...原创 2018-02-09 16:23:25 · 732 阅读 · 2 评论 -
尝试使用TensorFlow实现yolo2遇到的问题
以前就发现TensorFlow的模型组件中一直没有yolo,最近比较闲,想看看能不能TensorFlow试试。根据paper和linux上运行的测试结果,网络主体还是比较容易实现的。但是主要问题是损失函数的实现,根据darknet的GitHub工程源码来看,基本思路是遍历每一个box使用预定好的规则处理不同情况的损失。但是TensorFlow的话,如果按照这种逻辑写损失,计算图的加载都慢到极点de...原创 2018-03-07 20:07:54 · 1726 阅读 · 2 评论 -
TensorFlow实现inception-resnet
论文地址inception-v1,inception-v2,inception-v3,inception-v4事前准备我这边使用的是Windows平台TensorFlow1.4 gpu版本,当然如果是cpu版本也玩不了,太慢了,这个网络可是用来跑ImageNet的。另外你可以下载前面的inception-v4论文,因为接下来要对照论文进行实现。当然这里只是尝试模拟,可能会有错误,欢迎指...原创 2018-02-16 17:05:13 · 12465 阅读 · 9 评论