preprocessing.sh

#!/bin/bash

set -e

echo "=========================================="
echo "YCB Dataset Download & Preparation Script"
echo "=========================================="

if ! command -v gdown &> /dev/null; then
    echo "Installing gdown..."
    pip install gdown
fi

if ! command -v unzip &> /dev/null; then
    echo "Installing unzip..."
    apt-get update && apt-get install -y unzip
fi

BASE_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
DATA_DIR="${BASE_DIR}/data"

mkdir -p "${DATA_DIR}"
cd "${DATA_DIR}"

echo ""
echo "Data directory: ${DATA_DIR}"
echo ""

# Download Real Data
echo "Downloading YCBSight-Real-Reorganized.zip..."
REAL_FILE_ID="1GWT7qZBtIc4_5yzX7AajSAbD991e3PjU"
gdown "https://drive.google.com/uc?id=${REAL_FILE_ID}" -O YCBSight-Real-Reorganized.zip

# Download Sim Train
echo "Downloading YCBSight-Sim-Reorganized-train.zip..."
SIM_TRAIN_FILE_ID="1wXsODxrkDzlSMO2p6rqe-SpvYvUmEQTu"
gdown "https://drive.google.com/uc?id=${SIM_TRAIN_FILE_ID}" -O YCBSight-Sim-Reorganized-train.zip

# Download Sim Val
echo "Downloading YCBSight-Sim-Reorganized-val.zip..."
SIM_VAL_FILE_ID="1KPKzjl5acBs-m4MBlLWkO0KxxObwaPaP"
gdown "https://drive.google.com/uc?id=${SIM_VAL_FILE_ID}" -O YCBSight-Sim-Reorganized-val.zip

# Extract all zip files (replace existing files)
echo ""
echo "Extracting all datasets..."
for zip_file in *.zip; do
    if [ -f "$zip_file" ]; then
        echo "  Extracting: $zip_file"
        unzip -o -q "$zip_file"
        echo "  Done: $zip_file"
    fi
done

echo ""
echo "=========================================="
echo "Download and extraction complete!"
echo "Data location: ${DATA_DIR}"
echo "=========================================="