tutorial - ¿Cuáles son los valores posibles para las opciones data_augmentation_options en la configuración de la tubería TensorFlow Object Detection?

tensorflow object detection api tutorial

Question

He entrenado con éxito un modelo de detección de objetos con TensorFlow con las configuraciones de muestra que se dan aquí: https://github.com/tensorflow/models/tree/master/object_detection/samples/configs

Ahora quiero afinar mi configuración para obtener mejores resultados. Una de las opciones prometedoras que veo allí es "data_augmentation_options" bajo "train_config". Actualmente, se ve así:

train_config: {
  batch_size: 1
  ...
  data_augmentation_options {
    random_horizontal_flip {
    }
  }
}

¿Hay otras opciones para hacer una escala aleatoria, recortar o ajustar el brillo?

Answer 1

La lista de opciones se proporciona en preprocessor.proto :

NormalizeImage normalize_image = 1;
RandomHorizontalFlip random_horizontal_flip = 2;
RandomPixelValueScale random_pixel_value_scale = 3;
RandomImageScale random_image_scale = 4;
RandomRGBtoGray random_rgb_to_gray = 5;
RandomAdjustBrightness random_adjust_brightness = 6;
RandomAdjustContrast random_adjust_contrast = 7;
RandomAdjustHue random_adjust_hue = 8;
RandomAdjustSaturation random_adjust_saturation = 9;
RandomDistortColor random_distort_color = 10;
RandomJitterBoxes random_jitter_boxes = 11;
RandomCropImage random_crop_image = 12;
RandomPadImage random_pad_image = 13;
RandomCropPadImage random_crop_pad_image = 14;
RandomCropToAspectRatio random_crop_to_aspect_ratio = 15;
RandomBlackPatches random_black_patches = 16;
RandomResizeMethod random_resize_method = 17;
ScaleBoxesToPixelCoordinates scale_boxes_to_pixel_coordinates = 18;
ResizeImage resize_image = 19;
SubtractChannelMean subtract_channel_mean = 20;
SSDRandomCrop ssd_random_crop = 21;
SSDRandomCropPad ssd_random_crop_pad = 22;
SSDRandomCropFixedAspectRatio ssd_random_crop_fixed_aspect_ratio = 23;

Puede ver los detalles de cada opción en preprocessor.py . Los argumentos se pueden proporcionar como pares clave-valor.

  data_augmentation_options {
    ssd_random_crop {
    }
  }
  data_augmentation_options {
    random_pixel_value_scale {
      minval: 0.6
    }
  }