参考网站:
(linux下)
(windows下)
(最简单明了的例子)
(mnist数据格式与lmdb转换代码分析)
(官网教程python做分类)
Caffe将自己的文件生成lmdb
- 生成txt图片文件列表清单
(标签制作)
- 调用convert_imageset.sh进行转换
创建可执行文件create_imagenet_train.bat
- ..\..\Build\x64\Release\convert_imageset.exe
- --resize_height=256
- --resize_width=256 #缩放
- --shuffle #打乱顺序
- F:\ImageDatabase\GoogleImage\GoogleImage-small\ #图像路径
- D:\Dev\caffe-master\examples\googleimages\train.txt #标签路径
- F:\ImageDatabase\GoogleImage\lmdb\googleimagessmall_train_lmdb #目标路径,不可存在
-
- Pause #暂停
同理create_imagenet_val.bat
- 计算均值文件
创建批处理文件compute_mean.bat
- ..\..\Build\x64\Release\compute_image_mean.exe
- F:\ImageDatabase\GoogleImage\lmdb\googleimagessmall_train_lmdb #train lmdb路径
- D:\Dev\caffe-master\examples\googleimages\mean.binaryproto #输出路径
-
- Pause
四、文件清单
将lmdb文件、mean文件、label文件等放到合适位置
配置网络配置文件中的路径,即可进行训练