用caffe对自己的图片进行分类,包含详细代码

这篇博客记录了使用深度学习框架Caffe对自定义图片数据集进行分类的完整步骤,包括数据预处理、转换为lmdb格式、计算均值、创建模型配置文件以及训练和测试过程。
摘要由CSDN通过智能技术生成

今年开始接触深度学习,之前很多工作都没有写博客觉得很可惜,决定从这次实验开始记录过程。第一次写,缺点比较多,欢迎大家提建议,以后改正。

一、数据
数据是我从网上下载的,后期打算自己采集图片再进行一次实验。数据集由三类图片构成,分别是bike,person和both,每类200张,一共600张。需要的同学可以从我的网盘下载,也可以自己找图片进行训练。网盘链接:链接:http://pan.baidu.com/s/1mihxBnU 密码:61t6然后分成数据集和测试集,测试集450张,每类150张,训练集150张,每类50张。建两个文件夹分别存放,路径是caffe-master/data/mydata/train 和caffe-master/data/mydata/val。因为接下来的程序要求图片大小一致,所以第一步是图片预处理,把图片都处理成256*256的,具体程序如下:(这步也可以省略,用后面的脚本进行处理更加方便。)

/**********************************************  
*           批量更改图片大小
*          Win10 + VS2015 + Opencv 3.1.0
*           Sun Ruiyun
* 2017 / 07 / 06
***********************************************/
#include "stdafx.h"  
#include <opencv2/opencv.hpp>  
#include <opencv2/imgproc/imgproc.hpp>  
#include <iostream>  
#include <string>  
#include <direct.h>  
using namespace cv;
using namespace std;


int main()
{
    int k;
    for (k = 1; k <= 9; k++)
    {
   

        char imageName[100];
        char imageSave[100];
        sprintf_s(imageName, "D:\\documents\\visual studio 2015\\Projects\\resize\\resize\\resize\\bikes\\both_00%d.bmp", k);
        sprintf_s(imageSave, "D:\\documents\\visual studio 2015\\Projects\\resize\\resize\\resize\\aft_bikes\\both00%d.bmp", k);
        Mat src_img = imread(imageName);//读入图片数据  
        Mat dst_img1;
        resize(src_img, dst_img1, Size(256, 256), 0, 0, CV_INTER_LINEAR);   // 双线性插值  
        imwrite(imageSave, dst_img1);
        waitKey(0);
    }


}

用程序处理完的图片如下图所示,都是256*256大小的图片。

  • 1
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值