frcnn_train_data_param的distort_param实现

frcnn_train_data_param

frcnn_train_data_param {
    source: "./data/train_list.txt"
    root_folder: "./data/train_images"
    num_classes:
    ignore_label:
    distort_param {
      brightness_prob: 0.5
      brightness_delta:
      contrast_prob: 0.5
      contrast_lower: 0.5
      contrast_upper: 1.5
      hue_prob: 0.5
      hue_delta:
      saturation_prob: 0.5
      saturation_lower: 0.5
      saturation_upper: 1.5
      random_order_prob: 0.0
    }
  }

下面是frcnn_train_data_layer使用distort的代码,通过调用ApplyDistort函数来实现,这个函数是在im_transforms.cpp里实现的

DistortionParameter是caffe.proto中的message,就是一个类

  if (param.has_distort_param()) {
    DistortionParameter distort_param = param.distort_param();
    img = ApplyDistort(img, distort_param);
  }