Tensorflow(object detection API)训练模型时做数据增强

修改配置文件:


train_config: {
  data_augmentation_options {
    random_horizontal_flip {
    }
  }
  data_augmentation_options {
    ssd_random_crop {
    }
  }
}

可选的类型:

normalize_image #归一化
random_pixel_value_scale #随机像素值比例
random_image_scale #随机图像比例
random_rgb_to_gray#随机将RGB转成Gray
random_adjust_brightness#随机调节亮度
random_adjust_contrast#随机调节对比度
random_adjust_hue#随机调整色调
random_adjust_saturation#随机调节饱和度
random_distort_color#随机扭曲颜色
random_jitter_boxes#随机抖动boxes
random_crop_to_aspect_ratio#随机裁剪长宽比
random_black_patches#随机添加黑色补丁
rgb_to_gray#RGB转成Gray
scale_boxes_to_pixel_coordinates#
subtract_channel_mean#减去通道平均值
random_horizontal_flip#随机水平翻转
random_vertical_flip#随机垂直翻转
random_rotation90#随机旋转90°
random_pad_image#随机填充图片
random_crop_pad_image#随机裁剪填充图片
random_resize_method#随机调整大小
resize_image#调整大小
ssd_random_crop
ssd_random_crop_pad
ssd_random_crop_fixed_aspect_ratio
ssd_random_crop_pad_fixed_aspect_ratio

你可能感兴趣的:(TensorFlow)