fetch_data.sh

#!/bin/bash

urle () { 
    [[ "${1}" ]] || return 1
    local LANG=C i x
    for (( i = 0; i < ${#1}; i++ )); do 
        x="${1:i:1}"
        [[ "${x}" == [a-zA-Z0-9.~-] ]] && echo -n "${x}" || printf '%%%02X' "'${x}"
    done
    echo
}

# Function to download, unzip, and remove the zip file
download_and_unzip() {
    local url=$1
    local output_file=$(basename "$url" | sed 's/.*sfile=//')

    wget --post-data "username=$username&password=$password" "$url" -O "$output_file" --no-check-certificate --continue
    unzip $output_file
    rm $output_file
}

# Function to download, extract tar.gz, and remove the tar.gz file
download_and_extract_targz() {
    local url=$1
    local output_file=$(basename "$url" | sed 's/.*sfile=//')

    wget --post-data "username=$username&password=$password" "$url" -O "$output_file" --no-check-certificate --continue
    tar -xzf $output_file
    rm $output_file
}

# Prompt for credentials
echo -e "\nYou need to register at https://interactvlm.is.tue.mpg.de"
read -p "Username:" username
read -sp "Password:" password
echo

username=$(urle $username)
password=$(urle $password)

mkdir -p ./trained_models

# Define download URLs
DATA_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=data.zip'
HCONTACT_DAMON_MODEL_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=interactvlm-3d-hcontact-damon.zip'
HCONTACT_DAMON_LEMON_RICH_WSCENE_MODEL_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=interactvlm-3d-hcontact-wScene-damon-lemon-rich.zip'
OAFFORD_LEMON_PIAD_MODEL_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=interactvlm-3d-oafford-lemon-piad.zip'
H2DCONTACT_MODEL_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=interactvlm-2d-hcontact.zip'
HCONTACT_OCONTACT_MODEL_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=interactvlm-3d-hcontact-ocontact.zip'
DAMON_DATASET_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=damon.tar.gz'
LEMON_DATASET_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=lemon.tar.gz'
PIAD_OCONTACT_DATASET_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=piad_ocontact_seen.tar.gz'
PICO_DATASET_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=pico.tar.gz'
OPTIM_DEMO_DATA_URL='https://download.is.tue.mpg.de/download.php?domain=interactvlm&sfile=optim_data.zip'

# Check command line arguments
if [ $# -eq 0 ]; then
    # No arguments provided - download all files
    echo "No argument provided. Downloading all files..."
    download_and_unzip "$DATA_URL"
    download_and_unzip "$HCONTACT_DAMON_MODEL_URL"
    download_and_unzip "$OAFFORD_LEMON_PIAD_MODEL_URL"
    download_and_unzip "$H2DCONTACT_MODEL_URL"
elif [ "$1" = "damon-dataset" ]; then
    # damon-dataset argument provided - download only that file
    echo "Downloading DAMON dataset only..."
    download_and_extract_targz "$DAMON_DATASET_URL"
elif [ "$1" = "lemon-dataset" ]; then
    # lemon-dataset argument provided - download only that file
    echo "Downloading LEMON dataset only..."
    download_and_extract_targz "$LEMON_DATASET_URL"
elif [ "$1" = "piad-ocontact-dataset" ]; then
    # piad-ocontact-dataset argument provided - download only that file
    echo "Downloading PIAD Object Contact dataset only..."
    download_and_extract_targz "$PIAD_OCONTACT_DATASET_URL"
elif [ "$1" = "pico-dataset" ]; then
    # pico-dataset argument provided - download only that file
    echo "Downloading PICO dataset only..."
    download_and_extract_targz "$PICO_DATASET_URL"
elif [ "$1" = "hcontact-wScene" ]; then
    # hcontact-wScene argument provided - download the specific model
    echo "Downloading Human Contact with Scene model..."
    download_and_unzip "$HCONTACT_DAMON_LEMON_RICH_WSCENE_MODEL_URL"
elif [ "$1" = "h2dcontact" ]; then
    # h2dcontact argument provided - download the specific model
    echo "Downloading 2D Human Contact model..."
    download_and_unzip "$H2DCONTACT_MODEL_URL"
elif [ "$1" = "joint-reconstruction" ]; then
    # joint-reconstruction argument provided - download the specific model
    echo "Downloading Joint Human-Object Contact model..."
    download_and_unzip "$HCONTACT_OCONTACT_MODEL_URL"
elif [ "$1" = "optim-demo-data" ]; then
    # optim-demo-data argument provided - download the specific demo data
    echo "Downloading Optimization Demo Data..."
    download_and_unzip "$OPTIM_DEMO_DATA_URL"
else
    echo "Unknown argument: $1"
    echo "Usage: $0 [damon-dataset|lemon-dataset|piad-ocontact-dataset|pico-dataset|hcontact-wScene|h2dcontact|joint-reconstruction|optim-demo-data]"
    echo "  No argument: Downloads all model files (data.zip, interactvlm-3d-hcontact-damon.zip, interactvlm-3d-oafford-lemon-piad.zip, interactvlm-2d-hcontact.zip)"
    echo "  damon-dataset: Downloads only damon.tar.gz dataset"
    echo "  lemon-dataset: Downloads only lemon.tar.gz dataset"
    echo "  piad-ocontact-dataset: Downloads only piad_ocontact_seen.tar.gz dataset"
    echo "  pico-dataset: Downloads only pico.tar.gz dataset"
    echo "  hcontact-wScene: Downloads interactvlm-3d-hcontact-wScene-damon-lemon-rich.zip"
    echo "  h2dcontact: Downloads interactvlm-2d-hcontact.zip"
    echo "  joint-reconstruction: Downloads interactvlm-3d-hcontact-ocontact.zip"
    echo "  optim-demo-data: Downloads optim_data.zip for optimization demo"
    exit 1
fi