#!/bin/bash
# Batch generate affordance masks for all four LIBERO subsets sequentially.

SRC_ROOT="/gemini/space/wrz/libero_per_frame"
TGT_ROOT="/gemini/space/wrz/ragnet_results"

for ds in libero_object libero_goal libero_spatial libero_10; do
    echo "========== Processing ${ds} =========="
    CUDA_VISIBLE_DEVICES=0 python batch_generate.py \
        --data_dir "${SRC_ROOT}/${ds}_converted" \
        --save_dir "${TGT_ROOT}/${ds}"
    echo "========== ${ds} done =========="
    echo
done