From 0011fa140644dbd45d4931386b02c97822647f7b Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Fri, 29 Nov 2024 13:34:05 +0530 Subject: [PATCH] better command --- training/mochi-1/prepare_dataset.sh | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/training/mochi-1/prepare_dataset.sh b/training/mochi-1/prepare_dataset.sh index 03a9655..c424b4e 100644 --- a/training/mochi-1/prepare_dataset.sh +++ b/training/mochi-1/prepare_dataset.sh @@ -6,6 +6,10 @@ OUTPUT_DIR=videos_prepared NUM_FRAMES=37 RESOLUTION=480x848 +# Extract width and height from RESOLUTION +WIDTH=$(echo $RESOLUTION | cut -dx -f1) +HEIGHT=$(echo $RESOLUTION | cut -dx -f2) + python trim_and_crop_videos.py $VIDEO_DIR $OUTPUT_DIR --num_frames=$NUM_FRAMES --resolution=$RESOLUTION --force_upsample -CUDA_VISIBLE_DEVICES=$GPU_ID python embed.py $OUTPUT_DIR --shape=37x480x848 \ No newline at end of file +CUDA_VISIBLE_DEVICES=$GPU_ID python embed.py $OUTPUT_DIR --shape=${NUM_FRAMES}x${WIDTH}x${HEIGHT}