text: # Copyright 2020 Huawei Technologies Co., Ltd
  #
  # 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.
  # ============================================================================
  """centerface unique configs"""
  
  class Config_centerface():
  
      flip_idx = [[0, 1], [3, 4]]
      default_resolution = [512, 512]
      heads = {'hm': 1, 'wh': 2, 'hm_offset': 2, 'landmarks': 5 * 2}
      head_conv = 64
      max_objs = 64
  
      rand_crop = True
      scale = 0.4
      shift = 0.1
      aug_rot = 0
      color_aug = True
      flip = 0.5
      input_res = 512 #768 #800
      output_res = 128 #192 #200
      num_classes = 1
      num_joints = 5
      reg_offset = True
      hm_hp = True
      reg_hp_offset = True
      dense_hp = False
      hm_weight = 1.0
      wh_weight = 0.1
      off_weight = 1.0
      lm_weight = 0.1
      rotate = 0
  
      # for test
      mean = [0.408, 0.447, 0.470]
      std = [0.289, 0.274, 0.278]
      test_scales = [0.999, ] #[1, ]
      nms = 1
      flip_test = 0
      fix_res = True
      input_h = 832 #800
      input_w = 832 #800
      K = 200
      down_ratio = 4
  
      seed = 317
      master_batch_size = 8
      num_workers = 8
      not_rand_crop = False
      no_color_aug = False
  