// Copyright 2021 The Deeplab2 Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

syntax = "proto2";

package deeplab2;

// Configure the dataset options.
message DatasetOptions {
  // Set the dataset. See dataset.py for supported datasets.
  optional string dataset = 1;
  // Set the dataset file pattern to be used with glob.
  repeated string file_pattern = 2;
  // Set the number of samples per batch. This must be a multiple of replicas.
  // E.g. batch_size = 8 on 4 GPUs equals a batch size of 2 on each GPU.
  optional int32 batch_size = 3 [default = 32];
  // Set the crop size as a list of [crop_height, crop_width].
  repeated int32 crop_size = 4;
  // Minimum value for resize. Can be 1) empty; or 2) an integer, indicating
  // the desired size of the shorter image side (either height or width); or
  // 3) a 2-tuple of (height, width), indicating the desired minimum value for
  // height and width after resize. Setting values to non-positive indicate
  // no minimum value would be used.
  repeated int32 min_resize_value = 5;
  // Maximum value for resize. Can be 1) empty; or 2) an integer, indicating
  // the maximum allowed size of the longer image side (either height or width);
  // or 3) a 2-tuple of (height, width), indicating the maximum allowed size
  // after resize. Setting values to non-positive indicates no maximum value
  // would be used.
  repeated int32 max_resize_value = 6;
  // Set the resizing factor.
  optional int32 resize_factor = 7;

  /* Augmentation options.*/
  message AugmentationOptions {
    // Set the minimum scale factor for augmentation. Default not to use.
    optional float min_scale_factor = 1 [default = 1.0];
    // Set the maximum scale factor for augmentation. Default not to use.
    optional float max_scale_factor = 2 [default = 1.0];
    // Set the scale factor step size for data augmentation.
    optional float scale_factor_step_size = 3 [default = 0.25];
    // The name of the AutoAugment policy to use.
    optional string autoaugment_policy_name = 4;
  }
  optional AugmentationOptions augmentations = 8;
  // Set the standard deviation used to generate Gaussian center ground-truth.
  optional float sigma = 9 [default = 8.0];
  // Set whether to use increased weights on small instances.
  optional bool increase_small_instance_weights = 10 [default = false];
  // Set the pixel threshold for small instances.
  optional int32 small_instance_threshold = 11 [default = 4096];
  // Set the small instance weight.
  optional float small_instance_weight = 12 [default = 3.0];
  // Set whether to use two frames togetehr (current frame + previous frame) as
  // input for video panoptic segmentation.
  optional bool use_two_frames = 13 [default = false];
  // Whether to decode the groundtruth label. Some dataset splits (e.g., test
  // set) may not contain any groundtruth label. In that case, set this field
  // to false to avoid decoding non-existing groundtruth label.
  optional bool decode_groundtruth_label = 14 [default = true];
  // Whether the model needs thing_id_mask annotations. When True, we will
  // additionally return mask annotation for each `thing` instance, encoded with
  // a unique thing_id. This ground-truth annotation could be used to learn a
  // better segmentation mask for each instance. `thing_id` indicates the number
  // of unique thing-ID to each instance in an image, starting the counting from
  // 0 (default: False).
  optional bool thing_id_mask_annotations = 15 [default = false];
  // Set the maximum number of possible thing instances per image. It is used
  // together when enabling generation of thing_id_mask_annotations (= True),
  // representing the maximum thing ID encoded in the thing_id_mask.
  optional int32 max_thing_id = 16 [default = 128];
  // Set whether to use the next frame together with the current frame for video
  // panoptic segmentation (VPS). This field also controls using two-frame as
  // input for VPS. Note that `use_two_frames` is adopted in Motion-DeepLab,
  // while `use_next_frame` is used in ViP-DeepLab.
  optional bool use_next_frame = 17 [default = false];
}