代码先锋网 代码片段及技术文章聚合

caffe中使用crop_size剪裁训练图片

转自:https://blog.csdn.net/u010417185/article/details/52651761


下面以一个简单的例子进行介绍。

  1. layer {  
  2.   name: ”data”  
  3.   type: ”Data”  
  4.   top: ”data”  
  5.   top: ”label”  
  6.   include {  
  7.     phase: TRAIN  
  8.   }  
  9.   transform_param {  
  10.     mirror: true  
  11.     crop_size: 600  
  12.     mean_file: ”examples/images/imagenet_mean.binaryproto”  
  13.   }  
  14.   data_param {  
  15.     source: ”examples/images/train_lmdb”  
  16.     batch_size: 256  
  17.     backend: LMDB  
  18.   }  
  19. }  
  20. layer {  
  21.   name: ”data”  
  22.   type: ”Data”  
  23.   top: ”data”  
  24.   top: ”label”  
  25.   include {  
  26.     phase: TEST  
  27.   }  
  28.   transform_param {  
  29.     mirror: false  
  30.     crop_size: 600  
  31.     mean_file: ”examples/images/imagenet_mean.binaryproto”  
  32.   }  
  33.   data_param {  
  34.     source: ”examples/images/val_lmdb”  
  35.     batch_size: 50  
  36.     backend: LMDB  
  37.   }  
  38. }  
layer {
  name: "data"
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
  transform_param {
    mirror: true
    crop_size: 600
    mean_file: "examples/images/imagenet_mean.binaryproto"
  }
  data_param {
    source: "examples/images/train_lmdb"
    batch_size: 256
    backend: LMDB
  }
}
layer {
  name: "data"
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TEST
  }
  transform_param {
    mirror: false
    crop_size: 600
    mean_file: "examples/images/imagenet_mean.binaryproto"
  }
  data_param {
    source: "examples/images/val_lmdb"
    batch_size: 50
    backend: LMDB
  }
}



从上面的 数据层的定义,看得出用了镜像和crop_size,还定义了 mean_file。

利用crop_size这种方式可以剪裁中心关注点和边角特征,mirror可以产生镜像,弥补小数据集的不足.

这里要重点讲一下crop_size在训练层与测试层的区别:

首先我们需要了解mean_file和crop_size没什么大关系。mean_file是根据训练集图片制作出来的,crop_size是对训练集图像进行裁剪,两个都是对原始的训练集图像进行处理。如果原始训练图像的尺寸大小为800*800,crop_size的图片为600*600,则mean_file与crop_size的图片均为800*800的图像集。

在caffe中,如果定义了crop_size,那么在train时会对大于crop_size的图片进行随机裁剪,而在test时只是截取中间部分(详见/caffe/src/caffe/data_transformer.cpp):


  1. //We only do random crop when we do training.  
  2.     if (phase_ == TRAIN) {  
  3.       h_off = Rand(datum_height - crop_size + 1);  
  4.       w_off = Rand(datum_width - crop_size + 1);  
  5.     } else {  
  6.       h_off = (datum_height - crop_size) / 2;  
  7.       w_off = (datum_width - crop_size) / 2;  
  8.     }  
  9.   }  
//We only do random crop when we do training.
    if (phase_ == TRAIN) {
      h_off = Rand(datum_height - crop_size + 1);
      w_off = Rand(datum_width - crop_size + 1);
    } else {
      h_off = (datum_height - crop_size) / 2;
      w_off = (datum_width - crop_size) / 2;
    }
  }



从上述的代码可以看出,如果我们输入的图片尺寸大于crop_size,那么图片会被裁剪。当 phase 模式为 TRAIN 时,裁剪是随机进行裁剪,而当为TEST 模式时,其裁剪方式则只是裁剪图像的中间区域。


下面是我在网上找到的自己进行图像裁剪的程序:

可对照给出的网址进行详细阅读:http://blog.csdn.net/u011762313/article/details/48343799


我们可以手动将图片裁剪并导入pycaffe中,这样能够提高识别率(pycaffe利用caffemodel进行分类中:进行分类这一步改为如下):


  1. #记录分类概率分布  
  2. pridects = np.zeros((1, CLASS_NUM))  
  3.   
  4. # 图片维度(高、宽)  
  5. img_shape = np.array(img.shape)  
  6. # 裁剪的大小(高、宽)  
  7. crop_dims = (3296)  
  8. crop_dims = np.array(crop_dims)  
  9. # 这里使用的图片高度全部固定为32,长度可变,最小为96  
  10. # 裁剪起点为0,终点为w_range  
  11. w_range = img_shape[1] - crop_dims[1]  
  12. # 从左往右剪一遍,再从右往左剪一遍,步长为96/4=24  
  13. for k in range(0, w_range + 1, crop_dims[1] / 4) + range(w_range, 1, -crop_dims[1] / 4):  
  14.     # 裁剪图片  
  15.     crop_img = img[:, k:k + crop_dims[1], :]  
  16.     # 数据输入、预处理  
  17.     net.blobs[’data’].data[…] = transformer.preprocess(‘data’, crop_img)  
  18.     # 前向迭代,即分类  
  19.     out = net.forward()  
  20.     # 每一次分类,概率分布叠加  
  21.     pridects += out[’prob’]  
  22.   
  23. # 取最大的概率分布为最终结果  
  24. pridect = pridects.argmax()  
#记录分类概率分布 
pridects = np.zeros((1, CLASS_NUM))

图片维度(高、宽)

img_shape = np.array(img.shape)

裁剪的大小(高、宽)

crop_dims = (32, 96)
crop_dims = np.array(crop_dims)

这里使用的图片高度全部固定为32,长度可变,最小为96

裁剪起点为0,终点为w_range

w_range = img_shape[1] - crop_dims[1]

从左往右剪一遍,再从右往左剪一遍,步长为96/4=24

for k in range(0, w_range + 1, crop_dims[1] / 4) + range(w_range, 1, -crop_dims[1] / 4):
# 裁剪图片
crop_img = img[:, k:k + crop_dims[1], :]
# 数据输入、预处理
net.blobs['data'].data[...] = transformer.preprocess('data', crop_img)
# 前向迭代,即分类
out = net.forward()
# 每一次分类,概率分布叠加
pridects += out['prob']

取最大的概率分布为最终结果

pridect = pridects.argmax()





  • caffe中提供了过采样的方法(oversample),详见/caffe/python/caffe/io.py,裁剪的是图片中央、4个角以及镜像共10张图片。

注:如果图片过大,需要适当缩小batch_size的值,否则使用GPU时可能超出其缓存大小而报错

版权声明:本文为buyaobianlan2017原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/buyaobianlan2017/article/details/80540032

智能推荐

图片剪裁

做项目是经常会遇见需要对图片进行剪裁的情况,下面来点干货 是不是很简单啊,end。。。。...

jQuery图片剪裁插件Image Cropper使用

初始化剪裁区 使用插件 cropper:https://www.jq22.com/jquery-info9322 必须IP+PORT 形式打开页面; 步骤: 找到剪裁区的图片 (img#image) 设置配置项 调用cropper方法,创建剪裁区 上传 选择图片 常规:上传图片都是默认的样式,有自己的按钮,样式不好看 现在: html中加入一个隐藏的文件域!为什么要隐藏?样式不好看; 点击上传按钮...

猜你喜欢

Caffe 中Crop 层详解

在Fully Convolutional Networks(FCN)中,会用到Crop 层,他的主要作用是进行裁切。下面我们举一个例子来说明如何使用Crop 层。   Caffe中的数据是以 blobs形式存在的,blob是四维数据,即 (Batch size, number of Chennels, Height, Width)=(N, C, H, W)(0,1,2,3)。 Crop层的输入(b...

caffe中的crop层

在Fully Convolutional Networks(FCN)中,会用到Crop 层,他的主要作用是进行裁切。下面我们举一个例子来说明如何使用Crop 层。     Caffe中的数据是以 blobs形式存在的,blob是四维数据,即 (Batch size, number of Chennels, Height, Width)=(N, C, H, W)。---(0,1,2,3) ...

关于不同Batch_Size在LSTM训练与预测中使用(Keras框架)

博文内容借鉴: How to use Different Batch Sizes when Training and Predicting with LSTMs 链接:https://machinelearningmastery.com/use-different-batch-sizes-training-predicting-python-keras/ 首先,我们会有一些问题才会产生这样子的疑问...

上拉加载

上拉加载 小程序有内置函数可以实现上拉加载...