| # Recreate the C-JEPA expected layout (pendulum_train/pendulum_val) for extract_four.py. | |
| # Usage: bash prepare_cjepa_root.sh [dataset_root] | |
| set -e | |
| ROOT=${1:-/home/user/pendulum3k/dataset} | |
| mkdir -p "$ROOT/cjepa_data_root/pendulum_train/videos" "$ROOT/cjepa_data_root/pendulum_val/videos" | |
| ln -sf "$ROOT"/videos/train/*.mp4 "$ROOT/cjepa_data_root/pendulum_train/videos/" | |
| ln -sf "$ROOT"/videos/val/*.mp4 "$ROOT/cjepa_data_root/pendulum_val/videos/" | |
| echo "linked $(ls "$ROOT/cjepa_data_root/pendulum_train/videos" | wc -l) train + \ | |
| $(ls "$ROOT/cjepa_data_root/pendulum_val/videos" | wc -l) val videos into $ROOT/cjepa_data_root" | |