config.yml

seed: 42

feature_extraction_configs:
  blink_detection: true
  upscale: 1
  extraction_library: "mediapipe"
  show_features: ['faces', 'eyes', 'blinks']

model_configs:
  models_path: "pre_trained_models"
  registered_model_names: ["ResNet18", "ResNet50"]
  labels: ["left_eye", "right_eye"]
  targets: ["left_pupil", "right_pupil"]
  num_classes: 1

xai_configs:
  attribution_methods: [
    "IntegratedGradients",
    "Saliency",
    "InputXGradient",
    "GuidedBackprop",
    "Deconvolution",
    # "GuidedGradCam",
    # "LayerGradCam",
    # "LayerGradientXActivation",
  ]
  cam_methods: [
    "CAM",
    "GradCAM",
    "GradCAMpp",
    "SmoothGradCAMpp",
    "ScoreCAM",
    "SSCAM",
    "ISCAM",
    "XGradCAM",
    "LayerCAM",
  ]

use_sr: false

upscale_configs:
  upscale: [1, 2, 3, 4]
  upscale_method_configs:
    size: [16, 32]
    antialias: true
    interpolation: ["bicubic"]
      
sr_methods: ["GFPGAN", "RealESRGAN", "SRResNet", "CodeFormer", "HAT"]
sr_method_configs:
  bg_upsampler_name: "realesrgan"
  prefered_net_in_upsampler: "RRDBNet"