Spaces:
Running
Running
File size: 1,064 Bytes
51ba5d6 9acc552 51ba5d6 9acc552 51ba5d6 0f2d9f6 51ba5d6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
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"
|