训练与测试自己的图片

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测试集与自己手写的数字进行测试

使用caffe的cifar10网络模型训练自己的图片数据

跑通caffe-ssd demo代码(训练测试自己数据集)

学习总结win10下yolov5使用GPU训练自己的模型,同时调用网络摄像头识别目标学习指南与踩坑经验

Yolov5训练自己的数据集(详细完整版)

ubuntu16.04 使用tensorflow object detection训练自己的模型