knorth55
commented
4 years ago @k-okada @MiyabiTane configsファイルにオーグメンテーションについては書かれています．
もともとのSSDの学習スクリプトのままですが，random_holizontal_flipをしています．
https://github.com/knorth55/coral_usb_ros/blob/master/training/labelme_voc/scripts/configs/pipeline_mobilenet_v2_ssd_retrain_last_few_layers.config#L135-L142
もとから何が用意されているかは，ここを読むと書いています． https://github.com/tensorflow/models/blob/v1.12.0/research/object_detection/builders/preprocessor_builder.py
試してはいないですが，コードを読む限り，先に定義されているものとしては以下のようなものがあるようです．
'normalize_image':
preprocessor.normalize_image,
'random_pixel_value_scale':
preprocessor.random_pixel_value_scale,
'random_image_scale':
preprocessor.random_image_scale,
'random_rgb_to_gray':
preprocessor.random_rgb_to_gray,
'random_adjust_brightness':
preprocessor.random_adjust_brightness,
'random_adjust_contrast':
preprocessor.random_adjust_contrast,
'random_adjust_hue':
preprocessor.random_adjust_hue,
'random_adjust_saturation':
preprocessor.random_adjust_saturation,
'random_distort_color':
preprocessor.random_distort_color,
'random_jitter_boxes':
preprocessor.random_jitter_boxes,
'random_crop_to_aspect_ratio':
preprocessor.random_crop_to_aspect_ratio,
'random_black_patches':
preprocessor.random_black_patches,
'rgb_to_gray':
preprocessor.rgb_to_gray,
'scale_boxes_to_pixel_coordinates': (
preprocessor.scale_boxes_to_pixel_coordinates),
'subtract_channel_mean':
preprocessor.subtract_channel_mean,
'convert_class_logits_to_softmax':
preprocessor.convert_class_logits_to_softmax, if step_type == 'random_horizontal_flip':
if step_type == 'random_vertical_flip':
if step_type == 'random_rotation90':
if step_type == 'random_crop_image':
if step_type == 'random_pad_image':
if step_type == 'random_crop_pad_image':
if step_type == 'random_resize_method':
if step_type == 'resize_image':
if step_type == 'ssd_random_crop':
if step_type == 'ssd_random_crop_pad':
if step_type == 'ssd_random_crop_fixed_aspect_ratio':
if step_type == 'ssd_random_crop_pad_fixed_aspect_ratio':ちなみにですが，学習スクリプトとしてはtensorflow/models@v1.12.0の以下にあるPythonスクリプトを走らせています．
https://github.com/tensorflow/models/blob/v1.12.0/research/object_detection/model_main.py
k-okada
MiyabiTane
tongtybj
@knorth55 トレーニングデータをサーバに送る際に自動でデータオーギュメンテ―ションできたら便利だな，と思ったんだけど，すでにそういう機能があったりするかな？
Cc: @miyabitane