caffe-ssd源码解析--数据增强

最新推荐文章于 2022-10-05 18:03:38 发布

原创

最新推荐文章于 2022-10-05 18:03:38 发布 · 992 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#Caffe

train.prototxt注释：

distort_param {
      brightness_prob: 0.5  #调整亮度，im_transforms.cpp:RandomBrightness函数
      brightness_delta: 32.0   #添加像素值，对原图增加-32至32中的随机像素值
      contrast_prob: 0.5    #对比度
      contrast_lower: 0.5   #随机对比因子的下界
      contrast_upper: 1.5   #随机对比因子的上界
      hue_prob: 0.5   #调整色调
      hue_delta: 18.0  #调整色调通道数量
      saturation_prob: 0.5  #调整饱和度
      saturation_lower: 0.5 #调整饱和因子的上界
      saturation_upper: 1.5 #调整饱和因子的下界
      random_order_prob: 0.0 #随机排列图像通道
    }

以第一个数据增强brightness为例：

对应SSD/caffe/src/caffe/util/im_transforms.cpp中的

void RandomBrightness(const cv::Mat& in_img, cv::Mat* out_img,
    const float brightness_prob, const float brightness_delta) {
  float prob;
  caffe_rng_uniform(1, 0.f, 1.f, &prob);
  if (prob < brightness_prob) {
    CHECK_GE(brightness_delta, 0) << "brightness_delta must be non-negative.";