胶囊神经网络
吃了亏还皮
向左向右总要再向前
展开
-
Capsules胶囊神经网络代码测试(1)——测试mnist数据
原地址:https://github.com/Sarasra/models/tree/master/research/capsules 运行必须要有GPU 下载两个数据包,第一个是 MNIST tfrecords,第二个是已经训练好的模型文件。 Download and extract MNIST tfrecords to $DATA_DIR/ from:https://storage.go...原创 2018-08-14 11:42:54 · 3794 阅读 · 1 评论 -
制作自己的TFRecord数据集
一直在做CNN图像分类,原来采用的图像读入方式是放到文件夹下,直接将数据加载进内存,然后再分batch输入网络进行训练,但是后来发现太占用内存了,加了新内存条还是不够用。查阅资料,原来使用TFRecords这种结构能够有效地节省内存空间。 下面就来制作自己的数据集吧!找到10类电商图像,调整和mnist类似,单通道,28x28大小。数据路径为D:\data\TFRecord import...原创 2018-08-14 18:14:14 · 634 阅读 · 0 评论 -
Capsules胶囊神经网络代码测试(2)——测试cifar10数据
原地址:https://github.com/Sarasra/models/tree/master/research/capsules 运行必须要有GPU 下载两个数据包,第一个是 cifar10 tfrecords,第二个是已经训练好的模型文件。 Download and extract cifar10 binary version to $DATA_DIR/ from https://w...原创 2018-08-16 16:59:34 · 1347 阅读 · 1 评论