#!/usr/bin/env bash
# Laedt die OpenCV-dnn_superres-Modelle (echte SR-CNNs) nach $1 (default: models/).
set -euo pipefail
DEST="${1:-models}"
mkdir -p "$DEST"

declare -A URLS=(
  ["EDSR_x2.pb"]="https://raw.githubusercontent.com/Saafke/EDSR_Tensorflow/master/models/EDSR_x2.pb"
  ["EDSR_x4.pb"]="https://raw.githubusercontent.com/Saafke/EDSR_Tensorflow/master/models/EDSR_x4.pb"
  ["FSRCNN_x2.pb"]="https://raw.githubusercontent.com/Saafke/FSRCNN_Tensorflow/master/models/FSRCNN_x2.pb"
  ["FSRCNN_x4.pb"]="https://raw.githubusercontent.com/Saafke/FSRCNN_Tensorflow/master/models/FSRCNN_x4.pb"
)

for name in "${!URLS[@]}"; do
  if [[ -s "$DEST/$name" ]]; then
    echo "vorhanden: $name"; continue
  fi
  echo "lade $name ..."
  curl -fL --retry 3 -o "$DEST/$name" "${URLS[$name]}"
done
echo "Modelle in $DEST:"
ls -lh "$DEST"