训练与测试自己的图片
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了训练与测试自己的图片相关的知识,希望对你有一定的参考价值。
1)数据集准备
家装风格图片共500张,分为5个类,每类各100张图片。80张作为train,20张作为val。
2)图片统一大小
所有图片统一大小为256*256大小。
3)生成train.txt与val.txt文件
类标签最好从0开始。
下面开始使用caffe
1)使用convert_imageset.exe把图片数据库转换为.lmdb或者.leveldb的格式。
使用批处理命令:converttrain和convertval。
2)使用compute_image_mean.exe进行取均值的预处理,生成.binaryproto文件。
使用批处理命令:mean
以上是关于训练与测试自己的图片的主要内容,如果未能解决你的问题,请参考以下文章
利用mnist训练集生成的caffemodel对mnist测试集与自己手写的数字进行测试