#!/bin/bash


MAX_FRAMES=16
IMAGE_SIZE=224
MAX_AUDIO_LEN_S="4"
MODE="vanilla"

GPU=0

DATASET="JavisBench-mini"  # 
INPUT_FILE="data/meta/${DATASET}.csv"
OUTPUT_FILE="data/eval/av_cache/${DATASET}-${MODE}-max${MAX_AUDIO_LEN_S}s.pt"

export CUDA_VISIBLE_DEVICES=${GPU}

python -m eval.javisbench.tools.extract_gt_fvd_fad_feat \
    --input_file ${INPUT_FILE} \
    --output_file ${OUTPUT_FILE} \
    --max_frames ${MAX_FRAMES} \
    --max_audio_len_s ${MAX_AUDIO_LEN_S} \
    --image_size ${IMAGE_SIZE} \
    --fvd_mode ${MODE}