configs/augmentations.json

{
  "ToGray": [],
  "ToSepia": [],
  "ChannelShuffle": [],
  "InvertImg": [],
  "VerticalFlip": [],
  "HorizontalFlip": [],
  "Flip": [],
  "Transpose": [],
  "RandomGridShuffle": [
    {
      "param_name": "grid",
      "type": "several_ints",
      "subparam_names": [
        "height",
        "width"
      ],
      "limits_list": [
        [
          1,
          10
        ],
        [
          1,
          10
        ]
      ],
      "defaults_list": [
        3,
        3
      ]
    }
  ],
  "Blur": [
    {
      "param_name": "blur_limit",
      "type": "int_interval",
      "limits_list": [
        3,
        100
      ],
      "defaults": [
        3,
        7
      ]
    }
  ],
  "MotionBlur": [
    {
      "param_name": "blur_limit",
      "type": "int_interval",
      "limits_list": [
        3,
        100
      ],
      "defaults": [
        3,
        7
      ]
    }
  ],
  "MedianBlur": [
    {
      "param_name": "blur_limit",
      "type": "int_interval",
      "limits_list": [
        3,
        100
      ],
      "defaults": [
        3,
        7
      ]
    }
  ],
  "GaussNoise": [
    {
      "param_name": "var_limit",
      "type": "int_interval",
      "limits_list": [
        0.0,
        500.0
      ],
      "defaults": [
        10.0,
        50.0
      ]
    },
    {
      "param_name": "mean",
      "type": "int_interval",
      "limits_list": [
        -100.0,
        100.0
      ],
      "defaults": 0.0
    }
  ],
  "RandomBrightness": [
    {
      "param_name": "limit",
      "type": "int_interval",
      "limits_list": [
        -1.0,
        1.0
      ],
      "defaults": [
        -0.2,
        0.2
      ]
    }
  ],
  "RandomContrast": [
    {
      "param_name": "limit",
      "type": "int_interval",
      "limits_list": [
        -1.0,
        1.0
      ],
      "defaults": [
        -0.2,
        0.2
      ]
    }
  ],
  "HueSaturationValue": [
    {
      "param_name": "hue_shift_limit",
      "type": "int_interval",
      "limits_list": [
        -100,
        100
      ],
      "defaults": [
        -20,
        20
      ]
    },
    {
      "param_name": "sat_shift_limit",
      "type": "int_interval",
      "limits_list": [
        -100,
        100
      ],
      "defaults": [
        -30,
        30
      ]
    },
    {
      "param_name": "val_shift_limit",
      "type": "int_interval",
      "limits_list": [
        -100,
        100
      ],
      "defaults": [
        -20,
        20
      ]
    }
  ],
  "RGBShift": [
    {
      "param_name": "r_shift_limit",
      "type": "int_interval",
      "limits_list": [
        -255,
        255
      ],
      "defaults": [
        -20,
        20
      ]
    },
    {
      "param_name": "g_shift_limit",
      "type": "int_interval",
      "limits_list": [
        -255,
        255
      ],
      "defaults": [
        -20,
        20
      ]
    },
    {
      "param_name": "b_shift_limit",
      "type": "int_interval",
      "limits_list": [
        -255,
        255
      ],
      "defaults": [
        -20,
        20
      ]
    }
  ],
  "JpegCompression": [
    {
      "param_name": ["quality_lower","quality_upper"],
      "type": "min_max",
      "limits_list": [
        0,
        100
      ],
      "defaults_list": [80,100]
    }
  ],
  "ImageCompression": [
    {
      "param_name": "quality_lower",
      "type": "int_interval",
      "limits_list": [
        0,
        100
      ],
      "defaults": 99
    },
    {
      "param_name": "quality_upper",
      "type": "int_interval",
      "limits_list": [
        0,
        100
      ],
      "defaults": 100
    },
    {
      "param_name": "compression_type",
      "type": "radio",
      "options_list": [
        0,
        1
      ]
    }
  ],
  "CLAHE": [
    {
      "param_name": "clip_limit",
      "type": "int_interval",
      "limits_list": [
        1,
        100
      ],
      "defaults": [
        1,
        4
      ]
    },
    {
      "param_name": "tile_grid_size",
      "type": "several_ints",
      "subparam_names": [
        "height",
        "width"
      ],
      "limits_list": [
        [
          1,
          100
        ],
        [
          1,
          100
        ]
      ],
      "defaults_list": [
        8,
        8
      ]
    }
  ],
  "Cutout": [
    {
      "param_name": "num_holes",
      "type": "int_interval",
      "limits_list": [
        1,
        100
      ],
      "defaults": 8
    },
    {
      "param_name": "max_h_size",
      "type": "int_interval",
      "limits_list": [
        1,
        100
      ],
      "defaults": 8
    },
    {
      "param_name": "max_w_size",
      "type": "int_interval",
      "limits_list": [
        1,
        100
      ],
      "defaults": 8
    },
    {
      "param_name": "fill_value",
      "type": "rgb"
    }
  ],
  "RandomSnow": [
    {
      "param_name": "snow_point_lower",
      "type": "int_interval",
      "limits_list": [
        0.0,
        1.0
      ],
      "defaults": 0.1
    },
    {
      "param_name": "snow_point_upper",
      "type": "int_interval",
      "limits_list": [
        0.0,
        1.0
      ],
      "defaults": 0.1
    },
    {
      "param_name": "brightness_coeff",
      "type": "int_interval",
      "limits_list": [
        0.0,
        20.0
      ],
      "defaults": 2.5
    }
  ],
  "RandomFog": [
    {
      "param_name": "fog_coef_lower",
      "type": "int_interval",
      "limits_list": [
        0.0,
        1.0
      ],
      "defaults": 0.1
    },
    {
      "param_name": "fog_coef_upper",
      "type": "int_interval",
      "limits_list": [
        0.0,
        1.0
      ],
      "defaults": 0.1
    },
    {
      "param_name": "alpha_coef",
      "type": "int_interval",
      "limits_list": [
        0.0,
        1.0
      ],
      "defaults": 0.08
    }
  ],
  "ChannelDropout": [
    {
      "param_name": "channel_drop_range",
      "type": "int_interval",
      "limits_list": [
        1,
        3
      ],
      "defaults": [
        1,
        1
      ]
    },
    {
      "param_name": "fill_value",
      "type": "int_interval",
      "limits_list": [
        0,
        255
      ],
      "defaults": 0
    }
  ],
  "Solarize": [
    {
      "param_name": "threshold",
      "type": "int_interval",
      "limits_list": [
        0,
        255
      ],
      "defaults": 128
    }
  ],
  "Equalize": [
    {
      "param_name": "mode",
      "type": "radio",
      "options_list": [
        "cv",
        "pil"
      ]
    },
    {
      "param_name": "by_channels",
      "type": "checkbox",
      "defaults": 1
    }],
    "CenterCrop" :
    [
      {
        "param_name": "height",
        "type": "int_interval",
        "placeholder": {
          "limits_list" : [1, "image_height"],
          "defaults" : "image_half_height"
        }
      },
      {
        "param_name": "width",
        "type": "int_interval",
        "placeholder": {
          "limits_list" : [1, "image_width"],
          "defaults" : "image_half_width"
        }
      }
    ]
    }