TensorFlow
lsjweiyi
足够折腾
展开
-
TensorFlow下如何通过文件名/文件夹名获取标签,并加入队列
TensorFlow官网给的cifar-10教程,是卷积神经网络入门的好例子,有时想直接拿这个模型来跑自己的数据,却发现他的数据类型不是常见的,我们一般获取的数据(图片)以文件夹分类存好,或者直接在文件名上注明了类别,这时就要通过文件名的来获取标签,显然直接用cifar-10的方式是不行的。 这里当然可以吧数据转化成cifar-10那种类型,不过个人不喜欢这种方式吧。 另一种就还是老办法,对文...原创 2018-04-04 18:51:32 · 3542 阅读 · 3 评论 -
win10+vs2017+python3.6+tensorflow1.8-gpu+cudnn 安装及使用遇到的问题
win10下安装tensorflow-gpu版总是容易出问题,gpu为1050ti,记录下: 第一步:安装VS2017,去官网下载安装就行了https://www.visualstudio.com/zh-hans/thank-you-downloading-visual-studio/?sku=Community&rel=15,安装时选择c++通用开发和Python开发,并在右边选择Py...原创 2018-05-16 18:46:23 · 2702 阅读 · 0 评论 -
windows下tensorflow的CNN框架
环境:windows+tensorflow-GPU-1.8+python3.6 我在cifar10代码的基础上将代码修改,排版,使功能模块更清晰,也更容易更换网络模型,代码里加上大量中文注释。有点基础的应该都能看懂了,主要目的是为了方便像我这样的新入门tensorflow的人写CNN网络,使用的时候要注意修改数据的路径,模型保存路径,图片的SIZE,类别NUM_CLASSES等一些参数,这些参数...原创 2018-06-08 13:27:42 · 583 阅读 · 0 评论