storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-21 07:55:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

better command

Browse Source
This commit is contained in:
sayakpaul
2024-11-29 13:34:05 +05:30
parent 95775ba5d1
commit 0011fa1406
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/mochi-1/prepare_dataset.sh
+5 -1
View File
@@ -6,6 +6,10 @@ OUTPUT_DIR=videos_prepared
NUM_FRAMES=37 NUM_FRAMES=37
RESOLUTION=480x848 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 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 CUDA_VISIBLE_DEVICES=$GPU_ID python embed.py $OUTPUT_DIR --shape=${NUM_FRAMES}x${WIDTH}x${HEIGHT}