1 写在前面
使用Caffe2搭建卷积神经网络,按照一般卷积网络流程走一遍没有问题,笔者想分享的是关于Caffe2数据载入问题,Caffe2有专用训练格式数据,如lmdb,leveldb,rocksdb...这三种格式,在读取时,如果发生这样的错误:
db_..../mnist-train-nchw-lmdb)
是因为Caffe2和Pytorch合并后,使用官方预编译安装或源码编译安装的安装方式,默认数据读取参数是关闭的,即USE_LMDB=OFF,因此会出现不能读取或找不到对应数据集的问题....USE_LMDB=ON python setup.py install
详细参考博客:Caffe2问题总结(一)
2 搭建卷积网络及获取数据
2.0 网络结构
手写字体识别使用经典LeNet-5卷积神经网络