2017年7月31日

tensorflow中使用指定的GPU及GPU显存

摘要: ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 本文目录 1 终端执行程序时设置使用的GPU 2 python代码中设置使用的GPU 3 设置tensorflow使用的显存大小 3.1 定量设置显存 3.2 按需设置显存 ~~~~~~~~~~~~~~~~~~~~~~~~ 阅读全文

posted @ 2017-07-31 16:16 蚂蚁flow 阅读(645) 评论(0) 推荐(0) 编辑

tf.expand_dims和tf.squeeze函数

摘要: tf.expand_dims() Function tf.expand_dims(input, axis=None, name=None, dim=None) Inserts a dimension of 1 into a tensor’s shape. 在第axis位置增加一个维度 Given a 阅读全文

posted @ 2017-07-31 16:03 蚂蚁flow 阅读(495) 评论(0) 推荐(0) 编辑

深度学习数据集MNIST ImageNet COCO PASCAL VOC介绍

摘要: 参考文档 深度学习数据集汇总介绍 1. MNIST 深度学习领域的“Hello World!”,入门必备!MNIST是一个手写数字数据库,它有60000个训练样本集和10000个测试样本集,每个样本图像的宽高为28*28。此数据集是以二进制存储的,不能直接以图像格式查看,不过很容易找到将其转换成图像 阅读全文

posted @ 2017-07-31 10:41 蚂蚁flow 阅读(3050) 评论(0) 推荐(0) 编辑

Microsoft COCO 数据集

摘要: 本篇博客主要以介绍MS COCO数据集为目标,分为3个部分:COCO介绍,数据集分类和COCO展示。 本人主要下载了其2014年版本的数据,一共有20G左右的图片和500M左右的标签文件。标签文件标记了每个segmentation+bounding box的精确坐标,其精度均为小数点后两位。一个目标 阅读全文

posted @ 2017-07-31 10:35 蚂蚁flow 阅读(12538) 评论(0) 推荐(0) 编辑

导航