首页 > 代码库 > 训练与测试自己的图片

训练与测试自己的图片

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

 

训练与测试自己的图片