JAV-Gen / scripts /bash /misc /extract_eval_avfeat.sh
kaiw7's picture
Upload folder using huggingface_hub
e490e7e verified
#!/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}